K² · Artificial intelligence
An AI that teaches itself gets sharper but stops exploring
The paper studies on-policy self-distillation, where a single model acts as both teacher and student, with the teacher conditioned on a correct demonstration to give dense token-level feedback.
AN
MP
Andrei Liviu Nicolicioiu, Mohammad Pezeshki, Aaron Courville
3 authors · cs.LG, cs.AI
The 30-second scan
Let an AI grade its own homework, and it stops trying new things.
- The authors show that on-policy self-distillation's strong pass@1 accuracy can come at a hidden cost: rollout diversity decreases and pass@k curves flatten so that generating more rollouts fails to improve accuracy.
- They trace this diversity loss to compounding biases in the design of self-distillation with sampled demonstrations, where the teacher scores rollouts through the model's own biases.
- The authors theoretically analyze the optimal self-distillation policy and show it tilts the base distribution by a pointwise conditional mutual information score between the student's rollout and the correct rollout used as context.