Can proximity between teacher and student fix distillation instability?
On-policy distillation works well in theory but fails in practice due to capacity gaps. Does dynamically constructing a proximal teacher within a trust region resolve this fragility?
On-policy distillation (OPD) has become the default LLM post-training paradigm because it occupies a sweet spot: it is mathematically equivalent to RL where the immediate reward is the log-probability ratio between teacher and student policies, so its on-policy nature avoids the catastrophic forgetting of SFT while its dense rewards escape the sample inefficiency and instability of RLVR (which gives only a sparse verifiable signal at the end). But standard OPD is optimization-fragile in practice, and TOP-D locates the bottleneck precisely: the capacity gap between a strong target teacher and a weaker student produces high-variance, unstable gradients.
The fix is to stop distilling directly from the distant target teacher and instead dynamically construct a proximal teacher — a teacher close to the current student — and iterate within a trust region. Theoretically this inherently controls gradient variance and, with safe off-policy data reuse inside the trust-region iterations, yields a formal global-convergence result and a monotonic-improvement bound. The framing is deliberately the classic RL move (TRPO's trust region) transplanted onto distillation: bound how far each update moves the policy, and stability follows for free — TOP-D adds zero computational overhead, decisively outperforming standard OPD and competitive RLVR baselines on mathematical reasoning across scales.
This connects to a growing recognition that the teacher/student gap is the load-bearing variable in distillation, not raw teacher strength. Since Does richer teacher context hurt student generalization?, the intuition that a maximally strong or maximally informed teacher is best keeps failing; proximity, not power, governs whether the transferred signal is learnable.
Inquiring lines that read this note 10
This note is a source for these research framings, grouped by the broader line of inquiry each explores. Scan the bold lines of inquiry; follow any specific question forward.
What makes weaker teacher models effective for stronger student training?- Why does teacher-student proximity matter more than absolute teacher strength?
- What causes on-policy distillation to become unstable at scale despite dense rewards?
- What makes student-teacher distributional mismatch derail on-policy distillation?
- How does student capacity limit what it can learn from teachers?
- Can signal quality regulations help smaller teachers outperform larger ones?
- Why does style transfer happen during knowledge distillation?
Related concepts in this collection 3
This note in its neighbourhood — explore the map, then jump to a related concept in the list below.
Click a node to walk · click center to open · click Open in graph to see this note in the full knowledge graph
-
Does richer teacher context hurt student generalization?
When teachers are given more information during distillation, they produce confident but brittle students. Does this trade-off between in-domain wins and out-of-distribution robustness hold across different task distributions?
reinforces that the teacher-student relationship, not teacher strength, drives distillation outcomes
-
Does self-distillation harm mathematical reasoning performance?
Self-distillation usually improves models while shortening outputs, but mathematical reasoning shows a puzzling exception: performance drops up to 40%. What mechanism explains this counter-intuitive degradation?
another failure mode of distillation dynamics that naive OPD does not anticipate
-
Does RLVR actually expand what models can reason about?
Explores whether reinforcement learning from verifiable rewards teaches models genuinely new reasoning skills or simply makes existing capabilities more reliable. Pass@k analysis suggests the latter.
grounds the contrast with RLVR that OPD's dense signal is designed to overcome
Related papers in this collection 8
Papers most semantically related to this note, ranked by cosine similarity in the embedding space.
- Trust Region Policy Distillation
- Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations
- Reinforcement Learning via Self-Distillation
- Self-distillation Enables Continual Learning
- Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?
- Consistency Models Made Easy
- Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?
- Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
Original note title
a dynamically constructed proximal teacher with a trust region turns unstable on-policy distillation into a stable monotonically improving paradigm