Advanced RAG Models with Graph Structures: Optimizing Complex Knowledge Reasoning and Text Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Dong, Yuxin, Wang, Shuo, Zheng, Hongye, Chen, Jiajing, Zhang, Zhenhong, Wang, Chihang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915007318458368
author Dong, Yuxin
Wang, Shuo
Zheng, Hongye
Chen, Jiajing
Zhang, Zhenhong
Wang, Chihang
author_facet Dong, Yuxin
Wang, Shuo
Zheng, Hongye
Chen, Jiajing
Zhang, Zhenhong
Wang, Chihang
contents This study aims to optimize the existing retrieval-augmented generation model (RAG) by introducing a graph structure to improve the performance of the model in dealing with complex knowledge reasoning tasks. The traditional RAG model has the problem of insufficient processing efficiency when facing complex graph structure information (such as knowledge graphs, hierarchical relationships, etc.), which affects the quality and consistency of the generated results. This study proposes a scheme to process graph structure data by combining graph neural network (GNN), so that the model can capture the complex relationship between entities, thereby improving the knowledge consistency and reasoning ability of the generated text. The experiment used the Natural Questions (NQ) dataset and compared it with multiple existing generation models. The results show that the graph-based RAG model proposed in this paper is superior to the traditional generation model in terms of quality, knowledge consistency, and reasoning ability, especially when dealing with tasks that require multi-dimensional reasoning. Through the combination of the enhancement of the retrieval module and the graph neural network, the model in this study can better handle complex knowledge background information and has broad potential value in multiple practical application scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2411_03572
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Advanced RAG Models with Graph Structures: Optimizing Complex Knowledge Reasoning and Text Generation
Dong, Yuxin
Wang, Shuo
Zheng, Hongye
Chen, Jiajing
Zhang, Zhenhong
Wang, Chihang
Information Retrieval
This study aims to optimize the existing retrieval-augmented generation model (RAG) by introducing a graph structure to improve the performance of the model in dealing with complex knowledge reasoning tasks. The traditional RAG model has the problem of insufficient processing efficiency when facing complex graph structure information (such as knowledge graphs, hierarchical relationships, etc.), which affects the quality and consistency of the generated results. This study proposes a scheme to process graph structure data by combining graph neural network (GNN), so that the model can capture the complex relationship between entities, thereby improving the knowledge consistency and reasoning ability of the generated text. The experiment used the Natural Questions (NQ) dataset and compared it with multiple existing generation models. The results show that the graph-based RAG model proposed in this paper is superior to the traditional generation model in terms of quality, knowledge consistency, and reasoning ability, especially when dealing with tasks that require multi-dimensional reasoning. Through the combination of the enhancement of the retrieval module and the graph neural network, the model in this study can better handle complex knowledge background information and has broad potential value in multiple practical application scenarios.
title Advanced RAG Models with Graph Structures: Optimizing Complex Knowledge Reasoning and Text Generation
topic Information Retrieval
url https://arxiv.org/abs/2411.03572