Localizing Credit at the Divergence: Path-Conditioned Self-Distillation for LLM Reasoning
On the map: LLM reasoning → Reinforcement learning for reasoning
also in 1 more
Abstract (arXiv)
Reinforcement learning from verifiable rewards assigns a single scalar to each rollout, leaving token-level credit assignment underspecified in long reasoning traces. On-policy self-distillation addresses this by letting the same model act as a teacher conditioned on privileged information, producing a dense per-token signal. But the common choice of a ground-truth answer is only an endpoint cue: on terse-answer tasks, the teacher falls silent at the intermediate positions where path-level guidance matters most. We propose Hindsight Self-Distillation (HSD), which conditions the teacher on a successful peer rollout drawn from the current training group. Such a peer is an exact sample from the success-conditioned policy, requiring no additional sampled rollouts. By providing a full successful continuation rather than only the final answer, the resulting credit signal concentrates at the divergence position between a failed rollout and a successful peer. Across Qwen3-8B and Qwen3-32B on math and code benchmarks, HSD obtains the best result against GRPO variants and on-policy distillation baselines, with the largest gains on terse-answer tasks such as AIME.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 22 of its 23 references that are in the map. First posted on arXiv 14 Jun 2026.
Builds on
22 of 23 references.
Closest ancestors · 4
Same direction, newest first
- 2026OPPO: Bayesian Value Recursion for Token-Level Credit Assignment in LLM Reasoning21 May
- 2026When Right Meets Wrong: Bilateral Context Conditioning with Reward-Confidence Correction for GRPO13 Mar
- 2026RC-GRPO: Reward-Conditioned Group Relative Policy Optimization for Multi-Turn Tool Calling Agents3 Feb
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
Other references · 18
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2021Measuring Coding Challenge Competence With APPScited by 902
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 482
- 2026Reinforcement Learning via Self-Distillationcited by 332
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 299
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
- 2025What Makes a Reward Model a Good Teacher? An Optimization Perspectivecited by 73
- 2015Nested Sequential Monte Carlo Methodscited by 19
- 2026MuonEq: Balancing Before Orthogonalization with Lightweight Equilibrationcited by 17
6 more
- 2025InSPO: Unlocking Intrinsic Self-Reflection for LLM Preference Optimizationcited by 11
- 2025Learning to Substitute Words with Model-based Score Rankingcited by 10
- 2025MiCoTA: Bridging the Learnability Gap with Intermediate CoT and Teacher Assistantscited by 9
- 2023Towards A Unified View of Answer Calibration for Multi-Step Reasoningcited by 8
- 2025Calibrating and Rotating: A Unified Framework for Weight Conditioning in PEFTcited by 4
- 2026Calibrating Model-Based Evaluation Metrics for Summarizationcited by 3
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026LoongReflect: Boosting Long-Horizon Reflection in Search Agents via Global Perspective Distillationcited by 1
- 2026One Symptom, Three Levers: A Critical Review of On-Policy Self-Distillationcited by 0
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teacherscited by 0
- 2026Where the Model Changes Its Mind: Hindsight-Divergence Localization for Efficient Reinforcement Learning with Verifiable Rewardscited by 0
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Modelscited by 0
- 2026Privileged Context as Drift in On-Policy Self-Distillationcited by 0
- 2026A Privileged Teacher That Only Scores Cannot Search: Critique-Conditioned On-Policy Self-Distillation for Reasoningcited by 0
- 2026A Privileged Teacher That Only Scores Cannot Search: Critique-Conditioned On-Policy Self-Distillation for Reasoning9 Oct 2026
- 2026Privileged Context as Drift in On-Policy Self-Distillation6 Oct 2026
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models30 Sep 2026
- 2026Where the Model Changes Its Mind: Hindsight-Divergence Localization for Efficient Reinforcement Learning with Verifiable Rewards29 Sep 2026
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teachers26 Sep 2026
- 2026One Symptom, Three Levers: A Critical Review of On-Policy Self-Distillation26 Aug 2026
- 2026LoongReflect: Boosting Long-Horizon Reflection in Search Agents via Global Perspective Distillation12 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Learning from Own Solutions: Self-Conditioned Credit Assignment for Reinforcement Learning with Verifiable Rewards
- 2026H2SD: Hybrid Hindsight Self-Distillation
- 2026AMR-SD: Asymmetric Meta-Reflective Self-Distillation for Token-Level Credit Assignment
- 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
- 2026Skill-Conditioned Gated Self-Distillation for LLM Reasoning
- 2026SIPO: Unifying Reinforcement Learning with On-Policy Self-Distillation
- 2026Restoring the Sweet Spot: Pass-Rate Weighted Self-Distillation for LLM Reasoning
- 2026Matching Supervision to the Student's Learning Capacity: A Unified Framework for On-Policy Self-Distillation