Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GRAG: Graph Retrieval-Augmented Generation

About

Naive Retrieval-Augmented Generation (RAG) focuses on individual documents during retrieval and, as a result, falls short in handling networked documents which are very popular in many applications such as citation graphs, social media, and knowledge graphs. To overcome this limitation, we introduce Graph Retrieval-Augmented Generation (GRAG), which tackles the fundamental challenges in retrieving textual subgraphs and integrating the joint textual and topological information into Large Language Models (LLMs) to enhance its generation. To enable efficient textual subgraph retrieval, we propose a novel divide-and-conquer strategy that retrieves the optimal subgraph structure in linear time. To achieve graph context-aware generation, incorporate textual graphs into LLMs through two complementary views-the text view and the graph view-enabling LLMs to more effectively comprehend and utilize the graph context. Extensive experiments on graph reasoning benchmarks demonstrate that in scenarios requiring multi-hop reasoning on textual graphs, our GRAG approach significantly outperforms current state-of-the-art RAG methods. Our datasets as well as codes of GRAG are available at https://github.com/HuieL/GRAG.

Yuntong Hu, Zhihan Lei, Zheng Zhang, Bo Pan, Chen Ling, Liang Zhao• 2024

Related benchmarks

TaskDatasetResultRank
Graph Question AnsweringExplaGraphs
Accuracy88.05
38
Question AnsweringHotpotQA
Exact Match36.1
10
Question AnsweringMedQA
ROUGE-135.33
6
Question AnsweringAALCR
ROUGE-113.07
6
Showing 4 of 4 rows

Other info

Follow for update