arXiv:2601.11144cs.IRcs.AI2026-01被引 1

提出兼顾效率与全面性的图结构检索生成框架,提升问答准确性。

Deep GraphRAG: A Balanced Approach to Hierarchical Retrieval and Adaptive Integration

  • 分层检索策略:从全局到局部逐步缩小搜索范围,结合社区内外上下文。
  • 三阶段检索:过滤、精炼、细粒度搜索,动态重排保证效率与覆盖。
  • 小模型达大模型效果:1.5B模型性能接近70B大模型,适合资源受限场景。

基于图的检索增强生成(GraphRAG)框架在全局搜索全面性与局部搜索效率之间存在权衡。现有方法常面临大规模分层图导航困难、检索路径优化不足及探索-利用动态失衡问题,且缺乏稳健的多阶段重排机制。为此,本文提出 Deep GraphRAG,一种兼顾层次化检索与自适应融合的平衡框架。该框架采用三级分层检索策略:(1) 基于局部上下文的社区间过滤,剪枝搜索空间;(2) 通过实体交互分析进行社区级精炼,优先选取相关子图;(3) 在目标社区内进行实体级细粒度搜索。一个基于束搜索优化的动态重排模块持续筛选候选结果,平衡效率与全局覆盖。此外,该框架还引入知识融合模块,采用紧凑型大模型(1.5B)并结合动态权重奖励的GRPO训练方法(DW-GRPO),动态调整相关性、忠实性与简洁性三项目标权重。实验表明,在 Natural Questions 与 HotpotQA 数据集上,Deep GraphRAG 显著优于基线图检索方法,在准确率和效率上均有提升。

原文摘要 · Abstract (English)

Graph-based Retrieval-Augmented Generation (GraphRAG) frameworks face a trade-off between the comprehensiveness of global search and the efficiency of local search. Existing methods are often challenged by navigating large-scale hierarchical graphs, optimizing retrieval paths, and balancing exploration-exploitation dynamics, frequently lacking robust multi-stage re-ranking. To overcome these deficits, we propose Deep GraphRAG, a framework designed for a balanced approach to hierarchical retrieval and adaptive integration. It introduces a hierarchical global-to-local retrieval strategy that integrates macroscopic inter-community and microscopic intra-community contextual relations. This strategy employs a three-stage process: (1) inter-community filtering, which prunes the search space using local context; (2) community-level refinement, which prioritizes relevant subgraphs via entity-interaction analysis; and (3) entity-level fine-grained search within target communities. A beam search-optimized dynamic re-ranking module guides this process, continuously filtering candidates to balance efficiency and global comprehensiveness. Deep GraphRAG also features a Knowledge Integration Module leveraging a compact LLM, trained with Dynamic Weighting Reward GRPO (DW-GRPO). This novel reinforcement learning approach dynamically adjusts reward weights to balance three key objectives: relevance, faithfulness, and conciseness. This training enables compact models (1.5B) to approach the performance of large models (70B) in the integration task. Evaluations on Natural Questions and HotpotQA demonstrate that Deep GraphRAG significantly outperforms baseline graph retrieval methods in both accuracy and efficiency.

图神经网络检索增强高效生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。