arXiv:2510.16582cs.AI2025-10NeurIPS被引 8

用流匹配优化知识图谱检索,提升复杂查询的准确与多样。

Can Knowledge-Graph-based Retrieval Augmented Generation Really Retrieve What You Need?

  • 基于状态转移的流匹配机制,联合优化检索策略与奖励估计。
  • 在STaRK基准上平均提升10%命中率与召回率,超越GPT-4o。
  • 适用于多领域真实查询,对未见知识图谱有强泛化能力。

基于知识图谱的检索增强生成(KG-RAG)通过提供结构化且可解释的外部知识来增强大语言模型。然而,现有方法在从富含文本的知识图谱中为复杂现实查询检索准确且多样信息时表现不佳。过程奖励模型(PRMs)虽能对齐检索过程与查询需求,但依赖昂贵且难以获取的过程级监督信号。为此,我们提出GraphFlow框架,高效地从富含文本的知识图谱中检索真实查询所需的知识。GraphFlow采用基于转移的流匹配目标,联合优化检索策略与流估计器。流估计器将检索结果的奖励分解至中间检索状态,引导检索策略按奖励比例选取候选节点,从而探索高质量知识区域,实现多样化与相关性兼备的结果。我们在涵盖多领域真实查询的STaRK基准上评估GraphFlow,其在命中率和召回率上平均比强基线(包括GPT-4o)高出10%,并在未见知识图谱上表现出良好泛化能力,验证了其有效性与鲁棒性。

原文摘要 · Abstract (English)

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.

知识图谱检索增强流匹配大模型

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