Abstract:Large Language Models (LLMs) demonstrate strong reasoning abilities but face limitations such as hallucinations and outdated knowledge. Knowledge Graph (KG)-based Retrieval-Augmented Generation (RAG) addresses these issues by grounding LLM outputs in structured external knowledge from KGs. However, current KG-based RAG frameworks still struggle to optimize the trade-off between retrieval effectiveness and efficiency in identifying a suitable amount of relevant graph information for the LLM to digest. We introduce SubgraphRAG, extending the KG-based RAG framework that retrieves subgraphs and leverages LLMs for reasoning and answer prediction. Our approach innovatively integrates a lightweight multilayer perceptron with a parallel triple-scoring mechanism for efficient and flexible subgraph retrieval while encoding directional structural distances to enhance retrieval effectiveness. The size of retrieved subgraphs can be flexibly adjusted to match the query's need and the downstream LLM's capabilities. This design strikes a balance between model complexity and reasoning power, enabling scalable and generalizable retrieval processes. Notably, based on our retrieved subgraphs, smaller LLMs like Llama3.1-8B-Instruct deliver competitive results with explainable reasoning, while larger models like GPT-4o achieve state-of-the-art accuracy compared with previous baselines -- all without fine-tuning. Extensive evaluations on the WebQSP and CWQ benchmarks highlight SubgraphRAG's strengths in efficiency, accuracy, and reliability by reducing hallucinations and improving response grounding.

Enhancing Retrieval Augmented Generation Systems with Knowledge Graphs

Retrieval-Augmented Generation with Knowledge Graphs for Customer Service Question Answering

Document Knowledge Graph to Enhance Question Answering with Retrieval Augmented Generation

SKETCH: Structured Knowledge Enhanced Text Comprehension for Holistic Retrieval

Graph Retrieval-Augmented Generation for Large Language Models: A Survey

SimGRAG: Leveraging Similar Subgraphs for Knowledge Graphs Driven Retrieval-Augmented Generation

Generative Knowledge Graph Construction: A Review

KG-Retriever: Efficient Knowledge Indexing for Retrieval-Augmented Large Language Models

Select and Augment: Enhanced Dense Retrieval Knowledge Graph Augmentation

Adapting to Non-Stationary Environments: Multi-Armed Bandit Enhanced Retrieval-Augmented Generation on Knowledge Graphs

Simple is Effective: The Roles of Graphs and Large Language Models in Knowledge-Graph-Based Retrieval-Augmented Generation

A Systematic Investigation of Knowledge Retrieval and Selection for Retrieval Augmented Generation

WeKnow-RAG: An Adaptive Approach for Retrieval-Augmented Generation Integrating Web Search and Knowledge Graphs

Empowering Large Language Models to Set up a Knowledge Retrieval Indexer via Self-Learning

Toward Subgraph-Guided Knowledge Graph Question Generation With Graph Neural Networks

HybridRAG: Integrating Knowledge Graphs and Vector Retrieval Augmented Generation for Efficient Information Extraction

Knowledge Graph-Augmented Language Models for Knowledge-Grounded Dialogue Generation

Graph-augmented Learning to Rank for Querying Large-scale Knowledge Graph

Knowledge Graph Retrieval-Augmented Generation for LLM-based Recommendation

Multi-Level Querying using A Knowledge Pyramid