CG-RAG: Research Question Answering by Citation Graph Retrieval-Augmented LLMs
Yuntong Hu, Zhihan Lei, Zhongjie Dai, Allen Zhang, Abhinav Angirekula, Zheng Zhang, Liang Zhao
TL;DR
The paper tackles research question answering over scientific literature by leveraging citation graphs to capture inter-paper relationships. It introduces Lexical-Semantic Graph Retrieval (LeSeGR), a graph-encoder-based fusion of sparse lexical and dense semantic signals, and Contextualized Graph Retrieval-Augmented Generation (CG-RAG) that uses top-N contextual subgraphs to guide generation. By formalizing a contextual, chunk-level citation graph and propagating information through an entangled retrieval framework, the authors demonstrate state-of-the-art retrieval and generation performance on PubMedQA-1k and PapersWithCodeQA, along with meaningful efficiency gains. The work highlights the practical impact of graph-contextualized representations for scalable, accurate scholarly question answering, and generalizes beyond existing retrieval paradigms to integrate structure and semantics in literature graphs.
Abstract
Research question answering requires accurate retrieval and contextual understanding of scientific literature. However, current Retrieval-Augmented Generation (RAG) methods often struggle to balance complex document relationships with precise information retrieval. In this paper, we introduce Contextualized Graph Retrieval-Augmented Generation (CG-RAG), a novel framework that integrates sparse and dense retrieval signals within graph structures to enhance retrieval efficiency and subsequently improve generation quality for research question answering. First, we propose a contextual graph representation for citation graphs, effectively capturing both explicit and implicit connections within and across documents. Next, we introduce Lexical-Semantic Graph Retrieval (LeSeGR), which seamlessly integrates sparse and dense retrieval signals with graph encoding. It bridges the gap between lexical precision and semantic understanding in citation graph retrieval, demonstrating generalizability to existing graph retrieval and hybrid retrieval methods. Finally, we present a context-aware generation strategy that utilizes the retrieved graph-structured information to generate precise and contextually enriched responses using large language models (LLMs). Extensive experiments on research question answering benchmarks across multiple domains demonstrate that our CG-RAG framework significantly outperforms RAG methods combined with various state-of-the-art retrieval approaches, delivering superior retrieval accuracy and generation quality.
