Better and Faster Large Language Models via Multi-Token Prediction

Multi-token prediction and grammaticality

  • Clarification that during training the model predicts several future tokens, but at inference it usually uses only the standard next-token head.
  • Optional use of extra heads for speculative decoding: predict multiple tokens, then verify and discard mismatches so quality matches ordinary decoding.
  • Concern that independently predicting token n+1 and n+2 could break grammar; response is that verification/rejection and standard sampling keep outputs consistent.

Sentence-level / semantic vectors vs token prediction

  • Several commenters wonder why we don’t predict sentence- or “concept”-level vectors and then decode to text.
  • Issues raised: need a second model to map back to text; embeddings are lossy; inverse mapping is many-to-one and may produce invalid language.
  • Some link this to encoder–decoder or autoencoder ideas and note that similar architectures have been tried; main difficulty is focusing on semantics when supervision is in token space.
  • Related idea: hierarchical prediction (paragraph style → sentence form → words), but interactions across levels are hard.

Planning vs pure next-token prediction

  • Debate over whether LLMs “plan” ahead or just locally optimize next tokens.
  • Some argue that if transformers did not implicitly encode likely future structure, they’d behave like Markov models and perform far worse.
  • Others insist that mathematically the model only computes P(next token | past tokens); any appearance of long-term planning is emergent from sequential conditioning and training data statistics.
  • Examples discussed: grammar like “a/an”, rhyming, and the ability to end a paragraph on a specified word.

Speculative decoding and self-speculation

  • Multi-token prediction here is viewed as a built-in draft model for self-speculative decoding, reducing latency by up to ~3× without degrading quality.
  • Emphasis that speculative decoding, when implemented correctly, samples from the same distribution as standard autoregressive decoding; differences are due only to randomness.

Joint vs single-token probabilities

  • One subthread analyzes how predicting only the next token can diverge from the most likely multi-token sequence.
  • Suggestions include training to predict further-ahead tokens (nth token) to better approximate joint distributions, though practical methods remain unclear.

Meta: scaling and trade-offs

  • Reports that multi-token prediction helps at some scales but can hurt at larger scales or higher numbers of predicted tokens.
  • Noted general confusion in the ecosystem (terminology like “heads,” rapidly changing practices) and recommendations for educational resources to understand the broader pipeline.