mlsanityData to 14 Oct 2026

A Shared Learning Rate Is Not a Neutral Control in Selective On-Policy Distillation

arXiv 2609.22109 ↗First posted 17 Aug 2026cs.LGcited by 0

On the map: Knowledge distillation → On-policy and self-distillation for LLMs

Abstract (arXiv)

Selective on-policy distillation trains a student only at the token positions a selector scores highest, and the literature compares selectors under a single shared learning rate--a control chosen to be neutral. We show it is not. Under LoRA on GSM8K (Qwen2.5-1.5B student, 7B teacher), across an 8x learning-rate grid, dense supervision is statistically flat (swing 1.8 pp, p=0.26) while every selective arm moves with the rate: 5.4 pp for a random 5% subset, 6.7 pp for a total-variation selector, up to 17.7 pp for a teachability selector. Consequently the dense-versus-selective verdict reads 10.1 pp at lr=1e-4 but 5.1 pp at 5e-5--a 2.0x difference decided by a parameter the protocol treats as scenery--and two of six pairwise significance calls between selectors flip between adjacent rates without any rank inversion. We call this selector-rate entanglement and trace it to selection itself rather than step size: AdamW update magnitudes track the rate to within 2.2% despite 15.5x gradient-norm differences across arms. A preregistered frozen-scoring ablation (selection scored by the initial student; criterion, budget, and on-policy rollouts unchanged; 12 seeds per cell) shows live scoring adds 3.79+/-1.69 pp of rate sensitivity (p=0.035) while the frozen arm remains significantly entangled (p=0.015): the feedback loop aggravates the phenomenon rather than causing it. Under full fine-tuning at the rates this literature actually uses (1e-6 to 1e-5) the pattern grows: dense itself swings 19.8 pp, the selective arm 49.5 pp, and the verdict ranges from a non-significant +3.6 pp at the published operating point to +34 pp (p=0.005) one notch hotter. On MATH-500 the rate dependence does not reproduce under LoRA, scoping that result, while the ~10 pp cost of selective training does. We prescribe reporting the arm x rate matrix, not a shared-rate column, as a precondition for selector comparisons.

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 17 of its 20 references that are in the map. First posted on arXiv 17 Aug 2026.

Builds on

17 of 20 references.

Closest ancestors · 7

Same direction, newest first

  1. 2026Outcome-Confounded Local Supervision in On-Policy Distillation26 Jul
  2. 2026DOPD: Dual On-policy Distillation29 Jun
  3. 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Training29 Jun
  4. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation26 May
  5. 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation10 May
  6. 2026TIP: Token Importance in On-Policy Distillation15 Apr
  7. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr

Other references · 10

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  3. 2024Qwen2.5 Technical Reportcited by 4,942
  4. 2023Let's Verify Step by Stepcited by 3,789
  5. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
  6. 2023Self-Consuming Generative Models Go MADcited by 250
  7. 2025Understanding and Mitigating Numerical Sources of Nondeterminism in LLM Inferencecited by 53
  8. 2025SelecTKD: Selective Token-Weighted Knowledge Distillation for LLMscited by 22
  9. 2024SWITCH: Studying with Teacher for Knowledge Distillation of Large Language Modelscited by 9
  10. 2026Rethinking Selective Knowledge Distillationcited by 9

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Not Every Token Is Worth Distilling: Selective Supervision for Direct-OPD
  2. 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
  3. 2026Does On-Policy Distillation Really Distill? From Noisy Teacher to Self-Improvement
  4. 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation
  5. 2026Token Utility for On-Policy Distillation via Local Trust-Region Optimization
  6. 2026Matching Supervision to the Student's Learning Capacity: A Unified Framework for On-Policy Self-Distillation
  7. 2026OPRD: On-Policy Representation Distillation
  8. 2026On the Position Bias of On-Policy Distillation