mlsanityData to 14 Oct 2026

Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teachers

arXiv 2609.32674 ↗First posted 26 Sep 2026cs.AIcited by 0

On the map: LLM reasoning → Reinforcement learning for reasoning also fits: Prompting and chain-of-thought reasoning

Abstract (arXiv)

On-policy reasoning models can learn from task rewards or teacher signals, but these sources differ in form and can favor conflicting updates, leaving unclear which should guide a given reasoning action. We introduce Expected Reasoning-Step Return (ERSR), which treats semantic reasoning steps as macro-actions and uses Monte Carlo student-policy rollouts to estimate the expected final task reward of student-generated and teacher-proposed actions in a common return space for step-level comparison. ERSR analysis reveals an outcome-dependent asymmetry: student actions are more beneficial than teacher replacements on successful trajectories, whereas teacher replacements become more beneficial on failed trajectories. We further show that student answer-probe gains track student-step ERSR utility and distinguish beneficial from harmful reasoning steps. Based on these findings, we propose Return-Referenced On-Policy Learning (R2OPL), which reinforces student reasoning on successful trajectories and distills teacher signals on failed ones, while using group success rate for difficulty scaling and student-probe gains for step-level modulation. Experiments across reasoning benchmarks and teacher--student configurations show that R2OPL consistently outperforms strong baselines. ERSR training dynamics further show that R2OPL jointly exploits substantial utility from both reward- and teacher-side signals, whereas existing hybrids often leave substantial residual utility in one branch.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 52 of its 73 references that are in the map. First posted on arXiv 26 Sep 2026.

Builds on

52 of 73 references.

Closest ancestors · 14

Same direction, newest first

  1. 2026Sequential Beats Joint: On the Interplay between On-Policy Distillation and RLVR3 Sep
  2. 2026Localizing Credit at the Divergence: Path-Conditioned Self-Distillation for LLM Reasoning14 Jun
  3. 2026The Easy, the Hard, and the Learnable: Confidence and Difficulty-Adaptive Policy Optimization for LLM Reasoning6 Jun
  4. 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing2 Apr
  5. 2025Soft Adaptive Policy Optimization25 Nov
  6. 2025Pinpointing crucial steps: Attribution-based Credit Assignment for Verifiable Reinforcement Learning10 Oct
  7. 2025Part I: Tricks or Traps? A Deep Dive into RL for LLM Reasoning11 Aug
  8. 2025On-Policy RL with Optimal Reward Baseline29 May
  9. 2025SEED-GRPO: Semantic Entropy Enhanced GRPO for Uncertainty-Aware Policy Optimization18 May
  10. 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce15 Apr
  11. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  12. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
2 more
  1. 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization4 Jan
  2. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 38

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  3. 2025Qwen3 Technical Reportcited by 8,820
  4. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
  5. 2025Group Sequence Policy Optimizationcited by 835
  6. 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 455
  7. 2026Reinforcement Learning via Self-Distillationcited by 332
  8. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 299
  9. 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 263
  10. 2026Gemma 4 Technical Reportcited by 259
  11. 2023SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Modelscited by 256
  12. 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 187
26 more
  1. 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolationcited by 184
  2. 2024VinePPO: Refining Credit Assignment in RL Training of LLMscited by 183
  3. 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
  4. 2024DistiLLM: Towards Streamlined Distillation for Large Language Modelscited by 163
  5. 2026Self-Distilled RLVRcited by 131
  6. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillationcited by 82
  7. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillationcited by 51
  8. 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillationcited by 45
  9. 2026Reinforcement-aware Knowledge Distillation for LLM Reasoningcited by 35
  10. 2026Trust Region On-Policy Distillationcited by 30
  11. 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Examplecited by 24
  12. 2026Pass the Baton: Trajectory-Relayed On-Policy Distillationcited by 18
  13. 2026On-Policy Self-Distillation without Any Supervisioncited by 17
  14. 2026On-Policy Delta Distillationcited by 15
  15. 2026On-policy Distillation with Verifiable Rewardcited by 15
  16. 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergencecited by 13
  17. 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidancecited by 13
  18. 2026Reward-Gated On-Policy Distillationcited by 13
  19. 2026Beyond GRPO and On-Policy Distillation: An Empirical Sparse-to-Dense Reward Principle for Language-Model Post-Trainingcited by 11
  20. 2026Does On-Policy Distillation Really Distill? From Noisy Teacher to Self-Improvementcited by 11
  21. 2026On-Policy Self-Distillation with Sampled Demonstrations Reduces Output Diversitycited by 10
  22. 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreementcited by 10
  23. 2026Lightning OPD 2.0: Mitigating Style Bias in Cross-Teacher On-Policy Distillation for Large Reasoning Modelscited by 5
  24. 2026Beyond Teacher Likelihood: Group-Calibrated On-Policy Distillation for Long-Context Reasoningcited by 4
  25. 2026Strong Teacher Not Needed? On Distillation in LLM Pretrainingcited by 3
  26. 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progresscited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Reward-Aligned Reweighting for On-Policy Distillation
  2. 2026Outcome-Confounded Local Supervision in On-Policy Distillation
  3. 2026ReflectRL: Learning from Golden Negative Trajectories via Reflective-to-Direct Reasoning
  4. 2026DIAL-OPD: Learning More from Fewer Tokens in On-Policy Distillation
  5. 2026CADENCE: Closing the Reasoning Gap via Coverage-Adaptive On-Policy Distillation
  6. 2026Transfer-Stratified On-Policy Distillation for RL-Improved Reasoning Teachers
  7. 2026Fast and Effective On-policy Distillation from Reasoning Prefixes
  8. 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation