arXiv:2601.19225cs.CLcs.AI2026-01中稿 · The Web Conference

让小模型更准地回答知识图谱问题,通过智能选路径和优化推理过程。

RPO-RAG: Aligning Small LLMs with Relation-aware Preference Optimization for Knowledge Graph Question Answering

  • 用语义感知路径采样+关系感知偏好优化,让小模型更好理解知识图谱
  • 在WebQSP上提升8.8%的准确率,在CWQ上刷新3B以下模型最佳表现
  • 专为小模型设计,适合资源受限场景下的本地化知识问答应用

大语言模型在知识密集型任务中常出现幻觉。检索增强生成(RAG)通过外部知识源(如知识图谱)缓解此问题。但现有基于知识图谱的RAG方法依赖语义无关的路径采样,且与知识图谱推理目标对齐不足,难以提升小模型性能。此外,它们直接输入检索路径,未组织成以答案为中心的可解释推理路径。本文提出RPO-RAG,首个专为小模型设计的基于知识图谱的RAG框架。创新包括:(1)查询-路径语义采样策略,提供有效监督信号;(2)关系感知偏好优化,对齐中间推理阶段的图谱关系信号;(3)答案中心提示设计,将实体与推理路径结构化呈现。在WebQSP和CWQ两个基准数据集上的实验表明,RPO-RAG显著缩小了小模型与大模型间的性能差距。在WebQSP上F1提升最高达8.8%,在CWQ上实现了参数低于80亿模型中的新最优表现,尤其在30亿参数以下模型中表现突出。该方法大幅提升了小模型的推理能力,凸显其在轻量化、本地部署的知识图谱问答中的潜力。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have recently demonstrated remarkable reasoning abilities, yet hallucinate on knowledge-intensive tasks. Retrieval-augmented generation (RAG) mitigates this issue by grounding answers in external sources, e.g., knowledge graphs (KGs). However, existing KG-based RAG approaches rely on semantics-unaware path sampling and are weakly aligned with KG reasoning objectives, which limits further accuracy gains. They also feed retrieved paths directly into the reasoner without organizing them into answer-centered reasoning paths, hindering small LLMs' ability to leverage the retrieved knowledge. Furthermore, prior works predominantly rely on large LLMs (e.g., ChatGPT/GPT-4) or assume backbones above 7B parameters, leaving sub-7B models underexplored. We address this gap with RPO-RAG, the first KG-based RAG framework specifically designed for small LLMs, to the best of our knowledge. RPO-RAG introduces three key innovations: (1) a query-path semantic sampling strategy that provides informative supervisory signals; (2) a relation-aware preference optimization that aligns training with intermediate KG reasoning signals (e.g., relation); and (3) an answer-centered prompt design that organizes entities and reasoning paths in an interpretable format. Extensive experiments on two benchmark Knowledge Graph Question Answering (KGQA) datasets, WebQSP and CWQ, demonstrate that RPO-RAG effectively bridges the performance gap between small and large language models. On WebQSP, it improves F1 by up to 8.8%, reflecting enhanced answer precision, while on CWQ it achieves new state-of-the-art results among models under 8B parameters in both Hit and F1. Overall, RPO-RAG substantially improves the reasoning capability of small LLMs, even under 3B parameters-highlighting their potential for resource-efficient and practical on-device KGQA applications.

知识图谱小模型RAG推理优化

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