Speculative Self-Distillation enables Efficient Knowledge Internalization
Shayan Talaei ⋅ Agam Bhatia ⋅ Arshia Soltani Moakhar ⋅ Jonas Hübotter ⋅ Amin Saberi ⋅ Azalia Mirhoseini
Abstract
Many practical deployments require language models to internalize knowledge that was absent from pretraining, to incorporate continually updated facts, or to learn new domain skills. Self-distillation has emerged as a natural approach for this task. A model conditioned on the source material acts as the teacher, and an unconditioned student is trained to reproduce the teacher's behavior closed-book. We propose **Speculative Self-Distillation (SSD)**, an efficient per-token mixed-policy method for self-distillation. The student generates each rollout by default, while the teacher intervenes only at positions where its next-token distribution substantially diverges from the student's. This design keeps training rollouts close to the student's test-time behavior, unlike off-policy distillation with teacher-generated prefixes, but avoids spending teacher compute on the many low-signal tokens encountered by fully on-policy methods. In effect, SSD concentrates supervision precisely where privileged context changes the prediction. On three closed-book QA benchmarks measuring fact recall, knowledge updates, and skill acquisition, SSD matches on-policy distillation with $45\%$ fewer supervised tokens on average and consistently outperforms off-policy distillation on the held-out test set. On a forgetting stress test, SSD retains the most general capability while also achieving the best in-domain accuracy and using less supervision than the on-policy baseline. SSD therefore establishes a new accuracy--efficiency--forgetting Pareto frontier for self-distillation. Code is available at [anonymous.4open.science/r/Speculative-Self-Distillation](https://anonymous.4open.science/r/Speculative-Self-Distillation).
Chat is not available.
Successful Page Load