atlas

Research map / Graph learning / Knowledge and temporal graphs

G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering

NeurIPS 2024 · Graph learning › Knowledge and temporal graphs · approach: Graph-based dialog systems · 74 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces G-Retriever, a retrieval-augmented, soft-prompted LLM framework that answers questions over real-world textual graphs by retrieving a PCST-optimized subgraph, along with the GraphQA benchmark built from ExplaGraphs, SceneGraphs, and WebQSP.

Problem

Answering natural-language questions over large, real-world textual graphs is hard: existing graph-LLM work targets classification or toy graph queries, and textualizing whole graphs overflows LLM context windows and induces hallucinated nodes and edges.

Approach

Formulate subgraph retrieval for a query as a Prize-Collecting Steiner Tree problem over node and edge embeddings, then feed the retrieved subgraph to a frozen LLM via a GNN-encoded soft graph token plus a textualized subgraph, with optional LoRA tuning.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add graph encoder (ablation)G-Retriever · WebQSPHit@1: 54.62 ± 0.78 -> 70.49 ± 1.21Helped
Add projection layer (ablation)G-Retriever · WebQSPHit@1: 69.70 ± 0.68 -> 70.49 ± 1.21Helped
Include the textualized graph alongside the question (ablation)G-Retriever · WebQSPHit@1: 56.96 ± 1.83 -> 70.49 ± 1.21Helped
Add graph retrieval (ablation)G-Retriever · WebQSPHit@1: 63.84 ± 0.41 -> 70.49 ± 1.21Helped
Increase retrieval k from 3 to 5 (ablation)G-Retriever · WebQSPHit@1: 0.6977 -> 0.7063Helped
Increase retrieval k from 10 to 20 (ablation)G-Retriever · WebQSPHit@1: 0.7248 -> 0.7039Hurt
Replace Graph Transformer with GCN (ablation)G-Retriever · WebQSPHit@1: 70.49 -> 70.70No clear effect
Replace Graph Transformer with GAT (ablation)G-Retriever · WebQSPHit@1: 70.49 -> 70.27No clear effect
Increase LLM size from Llama2-7b to Llama2-13b (ablation)G-Retriever · WebQSPHit@1: 70.49 -> 75.58Helped

Surprising findings

Limitations

Experimental setup

Datasets: ExplaGraphs SceneGraphs WebQSP GQA

Benchmarks: GraphQA (ExplaGraphs, SceneGraphs, WebQSP)

Metrics: accuracy Hit@1 valid node/edge rate tokens and training time

Baselines: Zero-shot Zero-CoT CoT-BAG KAPING Prompt tuning GraphToken LoRA

Builds on: KAPING GraphToken MiniGPT-4 LoRA Sentence-BERT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy