mlsanityData to 7 Oct 2026

Understanding On-Policy Distillation: A Mechanistic Interpretability Perspective via Sparse Crosscoders

arXiv 2609.35210 ↗First posted 28 Sep 2026cs.CLcited by 0

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

Abstract (arXiv)

On-policy distillation (OPD) is a widely adopted post-training technique for LLM reasoning. It is commonly believed to transfer knowledge from a stronger teacher, yet what OPD actually distills into the student's internal representations remains unclear. We study this question with sparse crosscoders, which learn one feature dictionary shared by the student before and after OPD and the teacher. Standard crosscoder analyses, however, identify model-specific features but cannot tell how a model's use of its features changes, since all models are encoded into one set of feature activations. We therefore propose the swap readout, which reads each student checkpoint's feature activations on its own, measuring how training changes the student's use of each feature, even for checkpoints unseen by the crosscoder. Across three OPD settings, we find that OPD neither creates features nor passes on the teacher's own, and leaves the firing rates of over 98% of the student's frequently used features within 20%. We further examine the SFT warm-up on the teacher's rollouts that commonly precedes OPD and makes it more effective. Rather than adding features, the warm-up reweights the shared ones in two ways. First, it already raises and lowers many of the features that OPD later raises and lowers, doing part of OPD's work in advance. Second, it changes features that OPD alone would not, notably those for conversation format, reasoning style, and mathematical notation, and these changes persist through OPD. Imposing this reweighting on a directly distilled student's features, without changing its weights, brings its accuracy close to that of the warmed-up student, whereas the same change on shuffled features does not. Together, these findings suggest that OPD reweights existing features rather than acquiring new ones: the student learns from the teacher how to use the features they already share.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 28 of its 33 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

28 of 33 references.

Closest ancestors · 5

Same direction, newest first

  1. 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling12 Aug
  2. 2026Rethinking On-Policy Self-Distillation for Thinking Models6 Jul
  3. 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation29 May
  4. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
  5. 2025Towards Understanding Distilled Reasoning Models: A Representational Approach5 Mar

Other references · 23

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2025Qwen3 Technical Reportcited by 8,195
  3. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
  4. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  5. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  6. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  7. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
  8. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 502
  9. 2024RedPajama: an Open Dataset for Training Large Language Modelscited by 178
  10. 2026MiMo-V2-Flash Technical Reportcited by 163
  11. 2025Skywork Open Reasoner 1 Technical Reportcited by 161
  12. 2024BatchTopK Sparse Autoencoderscited by 126
11 more
  1. 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 57
  2. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
  3. 2025Overcoming Sparsity Artifacts in Crosscoders to Interpret Chat-Tuningcited by 31
  4. 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Modelscited by 24
  5. 2026Cross-Architecture Model Diffing with Crosscoders: Unsupervised Discovery of Differences Between LLMscited by 12
  6. 2026Escaping the KL Agreement Trap in On-Policy Distillationcited by 8
  7. 2026Does On-Policy Distillation Really Distill? From Noisy Teacher to Self-Improvementcited by 8
  8. 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreementcited by 7
  9. 2025Internal states before wait modulate reasoning patternscited by 6
  10. 2026Simple-OPD: Demystifying Warm-up for On-policy Distillationcited by 3
  11. 2026Why Does Reinforcement Learning Generalize? A Feature-Level Mechanistic Study of Post-Training in Large Language Modelscited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations
  2. 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation
  3. 2026Contrastive On-Policy Distillation
  4. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
  5. 2026Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe
  6. 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
  7. 2026Reward-Gated On-Policy Distillation
  8. 2026Calibrating Teacher--Student Discrepancy for On-Policy Distillation