arXiv:2601.15160cs.AIcs.CL2026-01被引 6

用知识图谱路径生成奖励信号,让模型学会分步推理。

Knowledge Graphs are Implicit Reward Models: Path-Derived Signals Enable Compositional Reasoning

  • 用知识图谱路径构造隐式奖励,引导模型逐步推理解题。
  • 140亿参数模型在4-5跳任务上超越大得多的GPT-5.2和Gemini 3 Pro。
  • 适合需要可解释、可验证推理的医学等专业领域研究者。

大型语言模型在数学与编程等结构化推理任务中已接近专家水平,但在特定科学领域进行组合式多跳推理的能力仍受限。本文提出一种自下而上的学习范式,将模型基于公理化领域事实,并通过组合这些事实解决复杂未知任务。为此,我们设计了一种后训练流程,结合监督微调与强化学习(RL),以知识图谱作为隐式奖励模型。通过从知识图谱路径中衍生出新的奖励信号,提供可验证、可扩展且具根基的监督,促使模型在强化学习中更注重中间公理的组合,而非仅优化最终答案。我们在医学领域验证该方法:使用140亿参数模型在短跳推理路径(1-3跳)上训练,并评估其零样本泛化至复杂多跳查询(4-5跳)的表现。实验表明,路径衍生奖励充当‘组合桥梁’,使模型显著优于更大模型及前沿系统如GPT-5.2和Gemini 3 Pro,尤其在最困难的任务上。此外,我们还证明该方法对选项打乱等对抗性扰动具有鲁棒性。结果表明,将推理过程扎根于结构化知识是实现智能推理的高效可扩展路径。代码已公开:https://github.com/jha-lab/kg-implicit-reward-compositional-rl/。

原文摘要 · Abstract (English)

Large language models have achieved near-expert performance in structured reasoning domains like mathematics and programming, yet their ability to perform compositional multi-hop reasoning in specialized scientific fields remains limited. We propose a bottom-up learning paradigm in which models are grounded in axiomatic domain facts and compose them to solve complex, unseen tasks. To this end, we present a post-training pipeline, based on a combination of supervised fine-tuning and reinforcement learning (RL), in which knowledge graphs act as implicit reward models. By deriving novel reward signals from knowledge graph paths, we provide verifiable, scalable, and grounded supervision that encourages models to compose intermediate axioms rather than optimize only final answers during RL. We validate this approach in the medical domain, training a 14B model on short-hop reasoning paths (1-3 hops) and evaluating its zero-shot generalization to complex multi-hop queries (4-5 hops). Our experiments show that path-derived rewards act as a "compositional bridge", enabling our model to significantly outperform much larger models and frontier systems like GPT-5.2 and Gemini 3 Pro, on the most difficult reasoning tasks. Furthermore, we demonstrate the robustness of our approach to adversarial perturbations against option-shuffling stress tests. This work suggests that grounding the reasoning process in structured knowledge is a scalable and efficient path toward intelligent reasoning. Our code is publicly available at: https://github.com/jha-lab/kg-implicit-reward-compositional-rl/.

知识图谱强化学习多跳推理医学AI

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