SMOPD: Selective Token-Entropy Masking for Dirty-History Multi-Turn On-Policy Self-Distillation
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
Dirty-history rollouts make multi-turn on-policy self-distillation (OPSD) brittle: once a student emits an erroneous intermediate reply, later turns are conditioned on that reply, and uniform distillation can spend loss on tokens that carry little corrective signal. We introduce SMOPD (Selective Masking for On-Policy Distillation), a loss-only stabilization method for multi-turn OPSD. For each generated middle-turn reply, SMOPD ranks token positions by student entropy and removes the lowest-entropy 20% from the clipped generalized Jensen-Shannon distillation loss; final-answer and FULL-preservation losses are unchanged. This design targets token-level uncertainty rather than coarse trajectory outcomes, adds no parameters, and has zero inference-time overhead. We compare SMOPD with a correctness-scaling variant that multiplies a common detached reliability proxy using final-answer correctness. On LiC with Qwen3 models, SMOPD improves SHARDED-view accuracy by 1.0-2.5 percentage points in single-seed 1.7B, 4B, and 8B comparisons, and a small 4B multi-seed check shows a +1.7pp mean SHARDED gain over baseline (two-tailed p = 0.022). Adding the outcome scalar is harmful without masking at 1.7B (-4.0pp) and remains scale-dependent when combined with masking (+1.3pp at 4B, neutral at 1.7B, and -0.5pp at 8B). These archived aggregate results suggest that token-level uncertainty is a more reliable stabilization signal than scalar final-answer correctness in this evaluated dirty-history OPSD setting, while leaving causal mechanism tests and broader benchmark validation to future work.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 32 of its 33 references that are in the map. First posted on arXiv 30 Jul 2026.
Builds on
32 of 33 references.
Closest ancestors · 15
Same direction, newest first
- 2026ROSD: Reflective On-Policy Self-Distillation for Language Model Reasoning across Domains27 May
- 2026MAIGO: Mitigating Lost-in-Conversation with History-Cleaned On-Policy Self-Distillation26 May
- 2026OGLS-SD: On-Policy Self-Distillation with Outcome-Guided Logit Steering for LLM Reasoning12 May
- 2026From Generic Correlation to Input-Specific Credit in On-Policy Self Distillation12 May
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May
- 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level7 May
- 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weighting12 Apr
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes26 Mar
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
- 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
- 2026On-Policy Context Distillation for Language Models12 Feb
3 more
Other references · 17
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2023Let's Verify Step by Stepcited by 3,789
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
- 2024Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluatorscited by 897
- 2024Self-Rewarding Language Modelscited by 681
- 2024Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Modelscited by 557
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 485
5 more
- 2025LLMs Get Lost In Multi-Turn Conversationcited by 429
- 2025Reasoning with Exploration: An Entropy Perspectivecited by 281
- 2024KTO: Model Alignment as Prospect Theoretic Optimizationcited by 264
- 2026UniSD: Towards a Unified Self-Distillation Framework for Large Language Modelscited by 7
- 2026Diversity-Aware Reverse Kullback-Leibler Divergence for Large Language Model Distillationcited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026E2-OPSD: Taming Entropy Overshoot in On-Policy Self-Distillation
- 2026A Unified Per-Token Gating Family for On-Policy Distillation: FKL/RKL Mixing with Multi-Channel and Bias Coefficients
- 2026On-Policy Self-Distillation without Any Supervision
- 2026Influence-Directed Distillation: Solving the Diversity Bottleneck in Sampled-Token On-Policy Distillation
- 2026Escaping the KL Agreement Trap in On-Policy Distillation
- 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation
- 2026ReNIO: Reweighting Negative Trajectory Importance for LLM On-Policy Distillation
- 2026ReTaCo: Residual-Target Control for On-Policy Distillation