Up nextNext guide
Speculative Streaming and Multi-Token Prediction
Technical
Language AI GUIDE
Multi-token prediction trains a language model with targets beyond the immediately following token.
Listen to this guideAI-narrated · ~3 min
Implementations differ: the cited Meta study uses parallel prediction heads, while DeepSeek-V3 describes sequential auxiliary modules.
Ordinary next-token training predicts the token following a supplied prefix. Better and Faster Large Language Models via Multi-token Prediction extends the target horizon: several future tokens are predicted from each training position. Its architecture uses a shared transformer trunk and independent output heads for different offsets. These heads include transformer layers and use a shared unembedding matrix, so calling every additional head merely a lightweight linear layer is not the paper’s exact architecture. The training loss combines future-token cross-entropy terms. For a four-target design, the first head predicts the token at t+1 and the fourth predicts t+4 from the available prefix representation. Those targets do not mean that inference can accept four arbitrary guesses without checking them. The paper’s basic inference mode retains next-token prediction and discards the other heads. It also investigates greedy self-speculative decoding that proposes and verifies additional tokens. Its measured gains and speedups belong to particular code and language experiments, model sizes, batch sizes, and decoding procedures. DeepSeek-V3’s report presents another MTP implementation. It sequentially predicts additional tokens through auxiliary modules, preserving a causal chain across prediction depths rather than using the same independent parallel-head design. This distinction matters when describing how each module receives its inputs. Both works study additional prediction targets, but sharing an objective label does not make their architectures identical. Neither report establishes that MTP is a default ingredient in every frontier model or guarantees a fixed speedup for every deployment.
Comparing next-token and multi-token training under matched evaluation conditions.
Inspecting future-token targets at explicitly stated sequence offsets.
Using only the next-token head in the Meta study’s basic inference setup.
Studying verified self-speculative decoding with auxiliary predictions.
Experiments can vary prediction horizon, auxiliary-loss weight, architecture, and decoding strategy while measuring quality and resource cost. A deeper prediction horizon is a hypothesis to evaluate, not proof that a model plans correctly over longer tasks. Reports should separate improvements from additional supervision from gains due to a particular inference implementation. Coding benchmarks, text prediction, and algorithmic tasks can reveal different behavior. The useful question for a chosen system is whether its tested quality, memory, and latency justify the extra training or inference components under the intended workload.
Free newsletter
Three verified AI stories every weekday morning, written in plain English. Free forever, no ads.
One email each weekday. Unsubscribe in one click. We never sell or share your address.
Test yourself
Instant feedback on every answer, and a shareable certificate with a verifiable ID once you pass a course.
Support free AI education. AI Understanding is a 501(c)(3) nonprofit — no ads, no paywall, ever. Make a donation
Multi-token prediction trains a language model with targets beyond the immediately following token. Implementations differ: the cited Meta study uses parallel prediction heads, while DeepSeek-V3 describes sequential auxiliary modules.
The heads target different offsets from a shared prefix representation; this is the specified parallel-head design.
DeepSeek-V3 adopted an MTP training objective to improve data efficiency and enable speculative decoding.
Several targets contribute prediction losses under the stated objective.
The reported experiment uses verified greedy self-speculation; arbitrary decoding policies need their own guarantees.
The extra heads are optional at deployment; discarding them keeps the model the same size as a standard next-token model.
Keep learning
More guides picked for this topic
Up nextNext guide
Speculative Streaming and Multi-Token Prediction
Technical