[2602.06019] Multi-Token Prediction via Self-Distillation
Abstract:Existing techniques for accelerating language model inference, such as speculative decoding, require training auxiliary speculator models and building and deploying complex inference pipelines. We consider a new approach for converting a pretrained autoregressive language model from a slow single next token prediction model into a fast standalone multi-token prediction model using a simple online distillation objective. The final model retains the exact same implementation as the pretrained initial checkpoint and is deployable without the addition of any auxiliary verifier or other specialized inference code. On GSM8K, our method produces models that can decode more than $3\times$ faster on average at $<5\%$ drop in accuracy relative to single token decoding performance.
[2602.06019] Multi-Token Prediction via Self-Distillation Skip to main content arXiv is now an independent nonprofit! Learn more × Search arXiv Press Enter to search · Advanced search --> Computer Science > Computation and Language arXiv:2602.06019 (cs) [Submitted on 5 Feb 2026 ( v1 ), last revised 23 Apr 2026 (this version, v2)] Title: Multi-Token Prediction via Self-Distillation Authors: John Kirchenbauer , Abhimanyu Hans , Brian Bartoldson , Micah Goldblum , Ashwinee Panda , Tom Goldstein View a PDF of the paper titled Multi-Token Prediction via Self-Distillation, by John Kirch
Explore this link on the map →related reading
- On-Policy Distillation - Thinking Machines Labthinkingmachines.ai
- Speculative Decoding - philkravphilkrav.com
- Fast Inference from Transformers via Speculative Decodingarxiv.org
- Composer2.pdfcursor.com
- [2601.18734] Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelsarxiv.org
- Large Language Diffusion Modelsarxiv.org
- Transformer Explainer: LLM Transformer Model Visually Explainedpoloclub.github.io
- frontier model training methodologies | Alex Wa's Blogdjdumpling.github.io
- Orthrus: Memory-Efficient Parallel Token Generation via Dual-View Diffusionarxiv.org
- [2605.06546] Efficient Pre-Training with Token Superpositionarxiv.org
- The Bitter Lesson is coming for Tokenization – ⛰️ lucalplucalp.dev
- Nemotron_Diffusion_Tech_Report_v1.pdfd1qx31qr3h6wln.cloudfront.net