Privileged Likelihood Is Not Automatically Value: Three Checks for Token Credit in On-Policy Self-Distillation
Not on the map. Why?
Abstract (arXiv)
On-policy self-distillation aims to improve upon reinforcement learning from verifiable rewards (RLVR) by providing token-level scores derived from privileged information, such as reference solutions or critic feedback. These scores are treated as estimates of token-level action values, yet they answer a fundamentally different question: how the model's prediction changes when its input context is enriched, rather than how the expected outcome changes when a token is changed. We examine this gap along three dimensions: (i) whether the token-level score tracks task success; (ii) whether feedback generated from the same rollout causes the score to reflect agreement with its own description, and whether using feedback from other rollouts in the group mitigates this self-referential effect; and (iii) what behavior the resulting training objective actually reinforces. In experiments on AIME 2025, the implemented score distinguishes correct from incorrect rollouts at approximately chance level (AUC=0.505); using feedback from a different rollout does not consistently improve this discrimination; and all training configurations achieve only 24.2-33.9% Avg@4, compared with 64.2% for outcome-only GRPO. Moreover, the highest-entropy token decile accounts for 57-71% of the total absolute token-advantage mass, despite the score being least informative about reasoning quality in this regime. By contrast, similar experiments on SciKnowEval Biology improves held-out Avg@8 by 28.0%, while its corresponding trajectory scores achieve AUCs of 0.81-0.92. Together, these results suggest that dense credit assignment through distillation can be effective when its likelihood-based scores are empirically validated as meaningful proxies for outcome-relevant credit. When this alignment does not hold, however, the resulting supervision can fail to generalize and may substantially underperform outcome-based RL.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 43 of its 60 references that are in the map. First posted on arXiv 10 Aug 2026.
Builds on
43 of 60 references.
References · 43
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Large Language Models Cannot Self-Correct Reasoning Yetcited by 965
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 386
- 2026Reinforcement Learning via Self-Distillationcited by 285
- 2019A Theory of Regularized Markov Decision Processescited by 279
- 2015Unifying distillation and privileged informationcited by 218
- 2021Does Knowledge Distillation Really Work?cited by 187
- 2022Defining and Characterizing Reward Hackingcited by 184
- 2020Self-Distillation Amplifies Regularization in Hilbert Spacecited by 171
- 2018RUDDER: Return Decomposition for Delayed Rewardscited by 162
31 more
- 2023MiniLLM: On-Policy Distillation of Large Language Modelscited by 129
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
- 2015Gradient Estimation Using Stochastic Computation Graphscited by 120
- 2026Self-Distilled RLVRcited by 109
- 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
- 2026Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?cited by 87
- 2019Hindsight Credit Assignmentcited by 74
- 2022Invariance in Policy Optimisation and Partial Identifiability in Reward Learningcited by 46
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
- 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillationcited by 33
- 2026Rethinking On-Policy Self-Distillation for Thinking Modelscited by 28
- 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Modelscited by 24
- 2026When Are Teacher Tokens Reliable? Position-Weighted On-Policy Self-Distillation for Reasoningcited by 24
- 2026Purified OPSD: On-Policy Self-Distillation Without Losing How to Thinkcited by 23
- 2026ROSD: Reflective On-Policy Self-Distillation for Language Model Reasoning across Domainscited by 17
- 2026Privileged, but Biased: How PI-Conditioned Teachers Break Self-Distillationcited by 12
- 2021Even your Teacher Needs Guidance: Ground-Truth Targets Dampen Regularization Imposed by Self-Distillationcited by 11
- 2023Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysiscited by 10
- 2026Self-Supervised On-Policy Distillation for Reasoning Language Modelscited by 10
- 2026Tailoring Teaching to Aptitude: Direction-Adaptive Self-Distillation for LLM Reasoningcited by 7
- 2026OPSD Compresses What RLVR Teaches: A Post-RL Compaction Stage for Reasoning Modelscited by 6
- 2026Learning from Language Feedback via Variational Policy Distillationcited by 6
- 2026On-Policy Self-Distillation with Sampled Demonstrations Reduces Output Diversitycited by 5
- 2026Diagnosing and Mitigating Thinking Collapse in On-Policy Self-Distillationcited by 5
- 2026OPPO: Bayesian Value Recursion for Token-Level Credit Assignment in LLM Reasoningcited by 4
- 2026Learning from Own Solutions: Self-Conditioned Credit Assignment for Reinforcement Learning with Verifiable Rewardscited by 4
- 2026CriPO: Enhancing Rubric-based RL via Self-Distillationcited by 4
- 2026CAST: Non-Privileged Clipped Asymmetric Self-Teaching with Advantage Flipping for GRPOcited by 3
- 2026Procedural Memory Distillation: Online Reflection for Self-Improving Language Modelscited by 3
- 2026Why Does Feedback-Augmented Self-Distillation Fail to Improve Retrieval-Interleaved Search Agents?cited by 2
- 2026β-OPSD: Deriving with Policy Optimization, Training with Self-Distillationcited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026From Generic Correlation to Input-Specific Credit in On-Policy Self Distillation
- 2026STEPS: Selective On-Policy Self-Distillation for Reasoning
- 2026PR-OPD: Privileged Representation On-policy Self-Distillation for Agentic Reinforcement Learning
- 2026SIPO: Unifying Reinforcement Learning with On-Policy Self-Distillation
- 2026Preference-Based Self-Distillation: Beyond KL Matching via Reward Regularization
- 2026Outcome-Guided On-Policy Self-Distillation
- 2026Localizing Credit at the Divergence: Path-Conditioned Self-Distillation for LLM Reasoning
- 2026I-SDPO: Instance-Level Adaptive Self-Distillation Policy Optimization