Knowledge Graphs are Implicit Reward Models: Path-Derived Signals Enable Compositional Reasoning
On the map: LLM reasoning → Knowledge- and tool-augmented reasoning
Abstract (arXiv)
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/.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 27 of its 43 references that are in the map. First posted on arXiv 21 Jan 2026.
Builds on
27 of 43 references.
References · 27
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2025Qwen3 Technical Reportcited by 8,820
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2023Let's Verify Step by Stepcited by 3,789
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
- 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
- 2025Process Reinforcement through Implicit Rewardscited by 504
- 2025The Lessons of Developing Process Reward Models in Mathematical Reasoningcited by 415
- 2021QA-GNN: Reasoning with Language Models and Knowledge Graphs for Question Answeringcited by 388
- 2025Rubrics as Rewards: Reinforcement Learning Beyond Verifiable Domainscited by 366
- 2017DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoningcited by 266
- 2025Logic-RL: Unleashing LLM Reasoning with Rule-Based Reinforcement Learningcited by 262
15 more
- 2025Hierarchical Reasoning Modelcited by 177
- 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoningcited by 162
- 2017Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learningcited by 161
- 2025Beyond Binary Rewards: Training LMs to Reason About Their Uncertaintycited by 117
- 2018Multi-Hop Knowledge Graph Reasoning with Reward Shapingcited by 102
- 2025Medical Hallucinations in Foundation Models and Their Impact on Healthcarecited by 93
- 2025Sample More to Think Less: Group Filtered Policy Optimization for Concise Reasoningcited by 75
- 2025From f(x) and g(x) to f(g(x)): LLMs Learn New Skills in RL by Composing Old Onescited by 42
- 2025RL Squeezes, SFT Expands: A Comparative Study of Reasoning LLMscited by 34
- 2025RL Fine-Tuning Heals OOD Forgetting in SFTcited by 31
- 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 28
- 2025Bottom-up Domain-specific Superintelligence: A Reliable Knowledge Graph is What We Needcited by 13
- 2025The Illusion of Readiness in Health AIcited by 12
- 2025Toward Scientific Reasoning in LLMs: Training from Expert Discussions via Reinforcement Learningcited by 7
- 2025Brittleness and Promise: Knowledge Graph Based Reward Modeling for Diagnostic Reasoningcited by 4
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026SHAPE: Unifying Safety, Helpfulness and Pedagogy for Educational LLMscited by 3
- 2026An Alternative Trajectory for Generative AIcited by 1
- 2026Knowledge Graph-Driven Expert-Level Reasoning for Neurosciencecited by 1
- 2026Peak-Then-Collapse and the Four Interface Channels of Knowledge-Graph Tool Usecited by 0
- 2026Co-Evolving Structured Knowledge and Reasoning in Language Modelscited by 0
- 2026Surgical Alignment in Knowledge Graph Training for Clinical Diagnosis with Large Language Modelscited by 0
- 2026Repair Before Reinforce: Context-Augmented Knowledge Graph Reasoning for Multi-Hop Question Answeringcited by 0
- 2026Repair Before Reinforce: Context-Augmented Knowledge Graph Reasoning for Multi-Hop Question Answering10 Sep 2026
- 2026Surgical Alignment in Knowledge Graph Training for Clinical Diagnosis with Large Language Models27 Aug 2026
- 2026Co-Evolving Structured Knowledge and Reasoning in Language Models26 Aug 2026
- 2026Peak-Then-Collapse and the Four Interface Channels of Knowledge-Graph Tool Use25 May 2026
- 2026Knowledge Graph-Driven Expert-Level Reasoning for Neuroscience24 May 2026
- 2026SHAPE: Unifying Safety, Helpfulness and Pedagogy for Educational LLMs24 Apr 2026
- 2026An Alternative Trajectory for Generative AI14 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2024Enhancing Multi-Hop Knowledge Graph Reasoning through Reward Shaping Techniques
- 2026Compositional Reasoning in Language Models under Reinforcement Learning Post-Training
- 2026Explore-on-Graph: Incentivizing Autonomous Exploration of Large Language Models on Knowledge Graphs with Path-refined Reward Modeling
- 2024Walk Wisely on Graph: Knowledge Graph Reasoning with Dual Agents via Efficient Guidance-Exploration
- 2026Graph Reasoning Paradigm: Structured and Symbolic Reasoning with Topology-Aware Reinforcement Learning for Large Language Models
- 2025Rewarding Structural Conformance of Reasoning using Process Mining
- 2025NPG-Muse: Scaling Long Chain-of-Thought Reasoning with NP-Hard Graph Problems
- 2026Outcome-Based RL Provably Leads Transformers to Reason, but Only With the Right Data