about
From Local to Global: A Graph RAG Approach to Query-Focused Summarization (arxiv.org)
2 points by Anon84 114 days ago | hide | past | pdf | discuss on HN

In plain words: It turns a document collection into a map of connected entities, prewrites summaries for each cluster, then combines them to answer big-picture questions like "what are the main themes." On collections around 1 million tokens, its answers beat standard retrieval on comprehensiveness and diversity.

Abstract

The use of retrieval-augmented generation (RAG) to retrieve relevant information from an external knowledge source enables large language models (LLMs) to answer questions over private and/or previously unseen document collections. However, RAG fails on global questions directed at an entire text corpus, such as "What are the main themes in the dataset?", since this is inherently a query-focused summarization (QFS) task, rather than an explicit retrieval task. Prior QFS methods, meanwhile, do not scale to the quantities of text indexed by typical RAG systems. To combine the strengths of these contrasting methods, we propose GraphRAG, a graph-based approach to question answering over private text corpora that scales with both the generality of user questions and the quantity of source text. Our approach uses an LLM to build a graph index in two stages: first, to derive an entity knowledge graph from the source documents, then to pregenerate community summaries for all groups of closely related entities. Given a question, each community summary is used to generate a partial response, before all partial responses are again summarized in a final response to the user. For a class of global sensemaking questions over datasets in the 1 million token range, we show that GraphRAG leads to substantial improvements over a conventional RAG baseline for both the comprehensiveness and diversity of generated answers.

Darren Edge, Ha Trinh, Newman Cheng, Joshua Bradley, Alex Chao, Apurva Mody, Steven Truitt, Dasha Metropolitansky, Robert Osazuwa Ness, Jonathan Larson
arXiv:2404.16130 · cs.CL, cs.AI, cs.IR · submitted Apr 24, 2024 · updated Feb 19, 2025
abstract · pdf · html

add comment on HN