Can distillation work on the student's own generated sequences?
Supervised distillation trains on fixed teacher outputs, but students must generate at inference. Does training on self-generated sequences scored by the teacher close this distribution mismatch and improve learning?
Supervised knowledge distillation trains an auto-regressive student to imitate a fixed set of teacher output sequences, which creates a distribution mismatch: at training time the student sees teacher-authored sequences, but at inference it must continue from its own generations, and the two distributions diverge. Generalized Knowledge Distillation (GKD) fixes this by making distillation on-policy — the student generates sequences and the teacher provides feedback on those sequences. Because the training distribution is now the student's own output distribution, the exposure-bias gap closes: the student is corrected precisely on the mistakes it will actually make at inference. This is the same failure that RL solves for policy learning, imported into distillation.
Two consequences follow. First, GKD offers flexibility in the divergence used between student and teacher, which matters when the student lacks the expressivity to match the teacher's distribution exactly — forward-KL mode-covering can be swapped for objectives that degrade more gracefully under capacity limits. Second, learning from self-generated sequences with a reward signal means GKD composes with RL fine-tuning, so a single loop can distill teacher knowledge and optimize a sequence-level reward — a natural upgrade path for the RLHF phase. This connects to the risk surfaced in Does self-distillation harm mathematical reasoning performance?: on-policy distillation improves what distribution the student learns from, but it does not by itself protect the uncertainty tokens that distillation tends to strip, so the two concerns are orthogonal and both must be managed.
Inquiring lines that read this note 1
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?Related concepts in this collection 2
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 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?
a distillation failure GKD's on-policy fix does not address; orthogonal risk
-
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?
style transfer is part of distillation; GKD changes which distribution is imitated, not the style-inheritance dynamic
Related papers in this collection 8
Papers most semantically related to this note, ranked by cosine similarity in the embedding space.
- On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes
- Self-distillation Enables Continual Learning
- Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?
- Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
- Trust Region Policy Distillation
- Reinforcement Learning via Self-Distillation
- Supervised Reinforcement Learning: From Expert Trajectories to Step-wise Reasoning
- Reverse Thinking Makes LLMs Stronger Reasoners
Original note title
on-policy distillation trains the student on its own generated sequences scored by the teacher fixing the train-inference mismatch that plagues supervised knowledge distillation