When Clipping Reverses Correction: Failure Dynamics of Pointwise Forward-KL On-Policy Self-Distillation
cs.CL
Submitted: 2026-09-30
Updated: 2026-09-30
Comments: 21 pages, 5 figures
Code: https://github.com/huggingface/math-verify
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy self-distillation (OPSD) trains a student on its own generated responses using feedback from the same model conditioned on privileged information.
Terminology
Abstract
On-policy self-distillation (OPSD) trains a student on its own generated responses using feedback from the same model conditioned on privileged information. On mathematical reasoning, the original OPSD study finds that stylistic tokens can dominate the training signal over math-related tokens, and that pointwise clipping of the forward KL objective stabilizes training. Pointwise clipping caps each vocabulary-wise forward KL term at a fixed threshold before summing over the vocabulary. Follow-up studies have adopted this clipping, but its effect on training has not been directly examined. In matched training runs differing only in whether clipping is applied, we observe that clipped runs produce substantially more repetitions that persist to the end of the response than their unclipped counterparts. We trace this failure to the clipped objective. We prove that the clipped objective can fail to correct the student toward the teacher and can instead push clipped and unclipped token probabilities away from its teacher. Our training runs agree with this analysis: inside repetitions, the clipped student places less probability than its teacher on leaving the repetition, and more on continuing it, whereas the unclipped runs stay close to their teachers.
Sources
- SCOPE-OPSD: Fisher-Conditioned Privileged Subspaces for On-Policy Self-Distillation
- DualOPSD: Closing the Feedback Loop in On-Policy Self-Distillation
- PAST: Privileged Adaptation from Complete Student Trajectories for On-Policy Self-Distillation
- Rethinking Reward Supervision: Rubric-Conditioned Self-Distillation
- DASH: Divergence-Adaptive Supervision Horizons for On-Policy Self-Distillation of Reasoning Models
- Reinforcement Learning via Self-Distillation
- Privileged Solutions or Context-Induced Teacher Behavior? Dissecting On-Policy Self-Distillation
- On-Policy Self-Distillation without Any Supervision
- Search-E1: Self-Distillation Drives Self-Evolution in Search-Augmented Reasoning
- When Are Teacher Tokens Reliable? Position-Weighted On-Policy Self-Distillation for Reasoning
- RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation
- HybridFlow: A Flexible and Efficient RLHF Framework
- PAINT: Partial-Solution Adaptive Interpolated Training for Self-Distilled Reasoners
- Self-Supervised On-Policy Distillation for Reasoning Language Models
- STEPS: Selective On-Policy Self-Distillation for Reasoning
- Qwen3 Technical Report
- OGLS-SD: On-Policy Self-Distillation with Outcome-Guided Logit Steering for LLM Reasoning
- What Does Privileged Information Add to On-Policy Self-Distillation?
- Is More Privileged Information Better? From Solution Traces to Problem-Solving Structure in Self-Distilled Reasoning
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering