arXiv:2508.05498cs.AI2025-08被引 8

让大模型学会在知识图谱中精准探索,提升问答准确率。

GRAIL:Learning to Interact with Large Knowledge Graphs for Retrieval Augmented Reasoning

  • 用大模型引导随机探索+路径过滤生成细粒度推理轨迹
  • 在三个数据集上平均准确率提升21.01%,F1提升22.43%
  • 适合需要结构化知识推理的AI系统开发者

将大语言模型(LLMs)与检索增强生成(RAG)结合已在多个领域展现优异性能。然而,现有RAG方法主要处理非结构化数据,在应对知识图谱等结构化知识时能力有限。当前图谱检索方法难以兼顾全局结构捕捉与精度控制,常导致关键信息缺失或冗余连接过多,影响推理效果。为此,我们提出GRAIL:图谱检索增强交互学习框架,通过大模型引导的随机探索与路径过滤构建数据合成管道,为每项任务自动生成细粒度推理轨迹。基于合成数据,采用两阶段训练学习动态决策策略,在每个推理步骤选择最优动作。将检索精度与简洁性的总体目标分解为细粒度过程监督奖励,提升数据效率与训练稳定性。实际部署中,GRAIL采用交互式检索范式,使模型自主探索图谱路径,动态平衡检索广度与精度。大量实验表明,GRAIL在三个知识图谱问答数据集上平均准确率提升21.01%,F1值提升22.43%。代码与数据集已开源。

原文摘要 · Abstract (English)

Large Language Models (LLMs) integrated with Retrieval-Augmented Generation (RAG) techniques have exhibited remarkable performance across a wide range of domains. However, existing RAG approaches primarily operate on unstructured data and demonstrate limited capability in handling structured knowledge such as knowledge graphs. Meanwhile, current graph retrieval methods fundamentally struggle to capture holistic graph structures while simultaneously facing precision control challenges that manifest as either critical information gaps or excessive redundant connections, collectively undermining reasoning performance. To address this challenge, we propose GRAIL: Graph-Retrieval Augmented Interactive Learning, a framework designed to interact with large-scale graphs for retrieval-augmented reasoning. Specifically, GRAIL integrates LLM-guided random exploration with path filtering to establish a data synthesis pipeline, where a fine-grained reasoning trajectory is automatically generated for each task. Based on the synthesized data, we then employ a two-stage training process to learn a policy that dynamically decides the optimal actions at each reasoning step. The overall objective of precision-conciseness balance in graph retrieval is decoupled into fine-grained process-supervised rewards to enhance data efficiency and training stability. In practical deployment, GRAIL adopts an interactive retrieval paradigm, enabling the model to autonomously explore graph paths while dynamically balancing retrieval breadth and precision. Extensive experiments have shown that GRAIL achieves an average accuracy improvement of 21.01% and F1 improvement of 22.43% on three knowledge graph question-answering datasets. Our source code and datasets is available at https://github.com/Changgeww/GRAIL.

知识图谱推理增强大模型RAG

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