Large language models (LLMs) face critical limitations in dynamically integrating external knowledge, mitigating hallucinations, and addressing domain-specific expertise. While retrieval-augmented generation (RAG) mitigates these issues, traditional chunk-based methods suffer from retrieval noise, multi-hop reasoning failures, and poor interpretability. Graph-based RAG (GraphRAG) has improved RAG in terms of context retrieval and relational reasoning, but there are still shortcomings in jointly processing these two types of tasks. To bridge this gap, we propose TriG-RAG, a unified framework that synergizes Triple-Granularity graph indexing with an adaptive retrieval strategy. TriG-RAG constructs a heterogeneous graph with a document-chunk-entity hierarchical structure and a weight allocation strategy, enabling flexible handling of both context-dependent and relation-dependent queries. Extensive experiments on six benchmarks demonstrate the superiority of TriG-RAG in terms of retrieval capability, flexibility and interpretability.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

TriG-RAG: Triple-Granularity Fusion for Retrieval-Augmented Generation with Adaptive Context-Relation Balance

  • Jingrui Zhang,
  • Yufeng Chen,
  • Jinan Xu

摘要

Large language models (LLMs) face critical limitations in dynamically integrating external knowledge, mitigating hallucinations, and addressing domain-specific expertise. While retrieval-augmented generation (RAG) mitigates these issues, traditional chunk-based methods suffer from retrieval noise, multi-hop reasoning failures, and poor interpretability. Graph-based RAG (GraphRAG) has improved RAG in terms of context retrieval and relational reasoning, but there are still shortcomings in jointly processing these two types of tasks. To bridge this gap, we propose TriG-RAG, a unified framework that synergizes Triple-Granularity graph indexing with an adaptive retrieval strategy. TriG-RAG constructs a heterogeneous graph with a document-chunk-entity hierarchical structure and a weight allocation strategy, enabling flexible handling of both context-dependent and relation-dependent queries. Extensive experiments on six benchmarks demonstrate the superiority of TriG-RAG in terms of retrieval capability, flexibility and interpretability.