mlsanityData to 14 Oct 2026

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

arXiv 2601.15160 ↗First posted 21 Jan 2026cs.AI cs.CLcited by 7

On the map: LLM reasoning → Knowledge- and tool-augmented reasoning

also in 1 more
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

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  4. 2023Let's Verify Step by Stepcited by 3,789
  5. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
  6. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
  7. 2025Process Reinforcement through Implicit Rewardscited by 504
  8. 2025The Lessons of Developing Process Reward Models in Mathematical Reasoningcited by 415
  9. 2021QA-GNN: Reasoning with Language Models and Knowledge Graphs for Question Answeringcited by 388
  10. 2025Rubrics as Rewards: Reinforcement Learning Beyond Verifiable Domainscited by 366
  11. 2017DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoningcited by 266
  12. 2025Logic-RL: Unleashing LLM Reasoning with Rule-Based Reinforcement Learningcited by 262
15 more
  1. 2025Hierarchical Reasoning Modelcited by 177
  2. 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoningcited by 162
  3. 2017Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learningcited by 161
  4. 2025Beyond Binary Rewards: Training LMs to Reason About Their Uncertaintycited by 117
  5. 2018Multi-Hop Knowledge Graph Reasoning with Reward Shapingcited by 102
  6. 2025Medical Hallucinations in Foundation Models and Their Impact on Healthcarecited by 93
  7. 2025Sample More to Think Less: Group Filtered Policy Optimization for Concise Reasoningcited by 75
  8. 2025From f(x) and g(x) to f(g(x)): LLMs Learn New Skills in RL by Composing Old Onescited by 42
  9. 2025RL Squeezes, SFT Expands: A Comparative Study of Reasoning LLMscited by 34
  10. 2025RL Fine-Tuning Heals OOD Forgetting in SFTcited by 31
  11. 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 28
  12. 2025Bottom-up Domain-specific Superintelligence: A Reliable Knowledge Graph is What We Needcited by 13
  13. 2025The Illusion of Readiness in Health AIcited by 12
  14. 2025Toward Scientific Reasoning in LLMs: Training from Expert Discussions via Reinforcement Learningcited by 7
  15. 2025Brittleness and Promise: Knowledge Graph Based Reward Modeling for Diagnostic Reasoningcited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 7.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Enhancing Multi-Hop Knowledge Graph Reasoning through Reward Shaping Techniques
  2. 2026Compositional Reasoning in Language Models under Reinforcement Learning Post-Training
  3. 2026Explore-on-Graph: Incentivizing Autonomous Exploration of Large Language Models on Knowledge Graphs with Path-refined Reward Modeling
  4. 2024Walk Wisely on Graph: Knowledge Graph Reasoning with Dual Agents via Efficient Guidance-Exploration
  5. 2026Graph Reasoning Paradigm: Structured and Symbolic Reasoning with Topology-Aware Reinforcement Learning for Large Language Models
  6. 2025Rewarding Structural Conformance of Reasoning using Process Mining
  7. 2025NPG-Muse: Scaling Long Chain-of-Thought Reasoning with NP-Hard Graph Problems
  8. 2026Outcome-Based RL Provably Leads Transformers to Reason, but Only With the Right Data