Skip to content
The AI Research Digest is included with Super Saver. See Super Saver
Tech AI Magazine Research
Digest

Research Digest » October 2026 research » Breakthrough 2 of 5

Breakthrough 2 of 5 · October 2026

Soft-MTP: a one-line change strips noise out of multi-token prediction training

A new preprint finds that the standard way of training multi-token prediction heads hides a noisy gradient term. Removing it improved perplexity, speculative decoding and reasoning at no measurable extra cost.

ML systems
Soft-MTP’s illustration, pseudocode, and loss performance compared to MTP. Blue line is the one-line code change . We refer readers to Algorithm 1 for the full code implementation of Soft-MTP, including the MTP loss weighting.
Soft-MTP’s illustration, pseudocode, and loss performance compared to MTP. Blue line is the one-line code change . We refer readers to Algorithm 1 for the full code implementation of Soft-MTP, including the MTP loss weighting. Figure from “Multi-Token Prediction Done Right” (arXiv)
RD
AI Research Digest research desk
October 2026 research · 4 min read · fact-checked against the source

Many recent language models use multi-token prediction (MTP): extra heads that guess several tokens ahead and can speed up inference through speculative decoding. A new preprint shows that the standard MTP training gradient splits exactly into two parts. One is a useful self-distillation term. The other is a noisy residual that works against next-token learning. The proposed method, Soft-MTP, keeps only the useful term with a one-line code change. It adds no parameters and no measurable overhead. In models from 167M to 1.2B parameters, it beat standard MTP on perplexity, zero-shot accuracy and speculative decoding.

Super Saver

Keep reading, and get the Explain it Like I'm 12 version

The full article and its Explain it Like I'm 12 version are for Super Saver members. Sign in if you're a member.