Can Knowledge-Graph-based Retrieval Augmented Generation Really Retrieve What You Need?
On the map: Retrieval-augmented generation → Knowledge-graph and structured-knowledge RAG
Abstract (arXiv)
Retrieval-Augmented Generation (RAG) based on knowledge graphs (KGs) enhances large language models (LLMs) by providing structured and interpretable external knowledge. However, existing KG-based RAG methods struggle to retrieve accurate and diverse information from text-rich KGs for complex real-world queries. Process Reward Models (PRMs) offer a way to align the retrieval process of KG-based RAG with query-specific knowledge requirements, but they heavily rely on process-level supervision signals that are expensive and hard to obtain on KGs. To address this challenge, we propose GraphFlow, a framework that efficiently retrieves accurate and diverse knowledge required for real-world queries from text-rich KGs. GraphFlow employs a transition-based flow matching objective to jointly optimize a retrieval policy and a flow estimator. The flow estimator factorizes the reward of the retrieval outcome into the intermediate retrieval states. Such reward factorization guides the retrieval policy to retrieve candidates from KGs in proportion to their reward. This allows GraphFlow to explore high-quality regions of KGs that yield diverse and relevant results. We evaluate GraphFlow on the STaRK benchmark, which includes real-world queries from multiple domains over text-rich KGs. GraphFlow outperforms strong KG-RAG baselines, including GPT-4o, by 10% on average in hit rate and recall. It also shows strong generalization to unseen KGs, demonstrating its effectiveness and robustness.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Placed in Knowledge-graph and structured-knowledge RAG, one of 15 directions in Retrieval-augmented generation. Builds on 77 of its 91 references that are in the map. First posted on arXiv 18 Oct 2025.
Builds on
77 of 91 references.
Closest ancestors · 11
Same direction, newest first
- 2025Weak-to-Strong GraphRAG: Aligning Weak Retrievers with Large Language Models for Graph-based Retrieval Augmented Generation26 Jun
- 2025HopRAG: Multi-Hop Reasoning for Logic-Aware Retrieval-Augmented Generation18 Feb
- 2025RAG vs. GraphRAG: A Systematic Evaluation and Key Insights17 Feb
- 2025GFM-RAG: Graph Foundation Model for Retrieval Augmented Generation3 Feb
- 2024Retrieval-Augmented Generation with Graphs (GraphRAG)31 Dec
- 2024KG-Retriever: Efficient Knowledge Indexing for Retrieval-Augmented Large Language Models7 Dec
- 2024Advanced RAG Models with Graph Structures: Optimizing Complex Knowledge Reasoning and Text Generation6 Nov
- 2024Think-on-Graph 2.0: Deep and Faithful Large Language Model Reasoning with Knowledge-guided Retrieval Augmented Generation15 Jul
- 2024GNN-RAG: Graph Neural Retrieval for Large Language Model Reasoning30 May
- 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarization24 Apr
- 2024G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering12 Feb
Other references · 66
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
- 2023Let's Verify Step by Stepcited by 3,789
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2023A Survey on Hallucination in Large Language Models: Principles, Taxonomy, Challenges, and Open Questionscited by 1,601
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,444
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
38 more
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 963
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 879
- 2023Siren's Song in the AI Ocean: A Survey on Hallucination in Large Language Modelscited by 698
- 2023Unifying Large Language Models and Knowledge Graphs: A Roadmapcited by 668
- 2025Process Reinforcement through Implicit Rewardscited by 504
- 2024RewardBench: Evaluating Reward Models for Language Modelingcited by 499
- 2018Behavioral Cloning from Observationcited by 492
- 2018Scalable agent alignment via reward modeling: a research directioncited by 447
- 2025Search-o1: Agentic Search-Enhanced Large Reasoning Modelscited by 433
- 2025The Lessons of Developing Process Reward Models in Mathematical Reasoningcited by 415
- 2021QA-GNN: Reasoning with Language Models and Knowledge Graphs for Question Answeringcited by 388
- 2023Reasoning on Graphs: Faithful and Interpretable Large Language Model Reasoningcited by 328
- 2021GFlowNet Foundationscited by 290
- 2023Challenges and Applications of Large Language Modelscited by 254
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
- 2022Bayesian Structure Learning with Generative Flow Networkscited by 153
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2025VisualPRM: An Effective Process Reward Model for Multimodal Reasoningcited by 139
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 132
- 2020Graph Information Bottleneck for Subgraph Recognitioncited by 119
- 2022Generative Flow Networks for Discrete Probabilistic Modelingcited by 114
- 2024Rule Based Rewards for Language Model Safetycited by 108
- 2023Amortizing intractable inference in large language modelscited by 97
- 2025Process Reward Models for LLM Agents: Practical Framework and Directionscited by 85
- 2023Towards Understanding and Improving GFlowNet Trainingcited by 75
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 72
- 2024MOOSE-Chem: Large Language Models for Rediscovering Unseen Chemistry Scientific Hypothesescited by 72
- 2024KG-RAG: Bridging the Gap Between Knowledge and Creativitycited by 56
- 2024STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Basescited by 55
- 2023Joint Bayesian Inference of Graphical Structure and Parameters with a Single Generative Flow Networkcited by 50
- 2021Improving Subgraph Recognition with Variational Graph Information Bottleneckcited by 45
- 2025VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Datacited by 40
- 2023Mind the Label Shift of Augmentation-based Graph OOD Generalizationcited by 34
- 2023KnowGPT: Knowledge Graph based Prompting for Large Language Modelscited by 31
- 2025RAG-RL: Advancing Retrieval-Augmented Generation via RL and Curriculum Learningcited by 30
- 2024HybGRAG: Hybrid Retrieval-Augmented Generation on Textual and Relational Knowledge Basescited by 29
- 2024Understand What LLM Needs: Dual Preference Alignment for Retrieval-Augmented Generationcited by 27
- 2024Knowledge-Aware Query Expansion with Large Language Models for Textual and Relational Retrievalcited by 26
50 most cited of 66 All 66 (JSON)
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026Autonomous Knowledge Graph Exploration with Adaptive Breadth-Depth Retrievalcited by 3
- 2026Calibrated Fusion for Heterogeneous Graph-Vector Retrieval in Multi-Hop QAcited by 2
- 2026MemGround: Long-Term Memory Evaluation Kit for Large Language Models in Gamified Scenarioscited by 2
- 2026MG2-RAG: Multi-Granularity Graph for Multimodal Retrieval-Augmented Generationcited by 1
- 2026HVM-GraphRAG: Holistic-View Multimodal Graph Retrieval-Augmented Generation on Complex Documentcited by 1
- 2026SeedER: Seed-Expand-Retrieve for Efficient Knowledge Graph Retrievalcited by 0
- 2026TRACE: Trustworthy Retrieval-Augmented Conversational Enginecited by 0
- 2026LivingRAG: Augmenting Graph RAG with Experiencecited by 0
- 2026LivingRAG: Augmenting Graph RAG with Experience26 Aug 2026
- 2026TRACE: Trustworthy Retrieval-Augmented Conversational Engine10 Aug 2026
- 2026HVM-GraphRAG: Holistic-View Multimodal Graph Retrieval-Augmented Generation on Complex Document26 Jul 2026
- 2026SeedER: Seed-Expand-Retrieve for Efficient Knowledge Graph Retrieval22 May 2026
- 2026MG2-RAG: Multi-Granularity Graph for Multimodal Retrieval-Augmented Generation4 Apr 2026
- 2026Calibrated Fusion for Heterogeneous Graph-Vector Retrieval in Multi-Hop QA30 Mar 2026
- 2026MemGround: Long-Term Memory Evaluation Kit for Large Language Models in Gamified Scenarios23 Mar 2026
- 2026Autonomous Knowledge Graph Exploration with Adaptive Breadth-Depth Retrieval20 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2025Graph-based Approaches and Functionalities in Retrieval-Augmented Generation: A Comprehensive Survey
- 2024Graph Retrieval-Augmented Generation: A Survey
- 2025A Survey of Graph Retrieval-Augmented Generation for Customized Large Language Models
- 2025NodeRAG: Structuring Graph-based RAG with Heterogeneous Nodes
- 2025A Query-Aware Multi-Path Knowledge Graph Fusion Approach for Enhancing Retrieval-Augmented Generation in Large Language Models
- 2025Pseudo-Knowledge Graph: Meta-Path Guided Retrieval and In-Graph Text for RAG-Equipped LLM
- 2026Use Graph When It Needs: Efficiently and Adaptively Integrating Retrieval-Augmented Generation with Graphs
- 2024Simple Is Effective: The Roles of Graphs and Large Language Models in Knowledge-Graph-Based Retrieval-Augmented Generation