Self-Distillation for Multi-Token Prediction

Guoliang Zhao, Ruobing Xie, An Wang, Kai Zhang, Shuaipeng Li, Huaibing Xie, Samm Sun


Abstract
As Large Language Models (LLMs) continue to scale, inference efficiency has emerged as a critical bottleneck. Multi-Token Prediction (MTP) offers a promising avenue to accelerate LLM inference by predicting multiple future tokens simultaneously. Wherein, DeepSeek MTP has gradually become an indispensable component in industrial LLMs. However, it still faces two challenges: limited accept rates of MTP heads, and difficulties in jointly training multiple MTP heads. To address these challenges, we propose MTP-D, a simple yet effective self-distillation method that substantially boosts MTP head accept rates while maximally preserving main-head performance. We further introduce a looped extension strategy for MTP-D, enabling effective and economical extension to multiple MTP heads and yielding significant additional inference speedup (up to 4.55×). Beyond these contributions, we conduct systematic experiments across seven benchmarks to explore and validate key insights into distillation strategies and the scalability potential of DeepSeek MTP. Results demonstrate that MTP-D and the grouped extension strategy consistently enhance MTP head performance and inference efficiency, facilitating the practical usage of MTP in LLMs.
Anthology ID:
2026.aacl-long.189
Volume:
Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
November
Year:
2026
Address:
Hengqin, China
Editors:
Yue Zhang, Ruifeng Xu, Ayu Purwarianti, Tong Rong, Jun Suzuki
Venues:
AACL | IJCNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
4225–4243
Language:
URL:
https://aclanthology.org/2026.aacl-long.189/
DOI:
Bibkey:
Cite (ACL):
Guoliang Zhao, Ruobing Xie, An Wang, Kai Zhang, Shuaipeng Li, Huaibing Xie, and Samm Sun. 2026. Self-Distillation for Multi-Token Prediction. In Proceedings of the 15th International Joint Conference on Natural Language Processing and the 5th Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics (Volume 1: Long Papers), pages 4225–4243, Hengqin, China. Association for Computational Linguistics.
Cite (Informal):
Self-Distillation for Multi-Token Prediction (Zhao et al., AACL-IJCNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.aacl-long.189.pdf