Efficient Training-Free Multi-Token Prediction via Embedding-Space Probing
Not on the map. Why?
Abstract (arXiv)
Large Language Models (LLMs) possess latent multi-token prediction (MTP) abilities despite being trained only for next-token generation. We introduce ESP (Embedding-Space Probing), a simple and training-free MTP method that probes an LLM using on-the-fly mask tokens drawn from its embedding space, enabling parallel future-token prediction without modifying weights or relying on draft models. ESP constructs a speculative token tree by sampling Top-K candidates from mask-token logits and applies a lightweight pruning rule to retain high-probability continuations. During generation, predictions are verified in parallel, yielding lossless decoding while significantly reducing model calls and increasing token throughput. ESP consistently outperforms existing training-free baselines, improving acceptance length by 7-11% over LADE on LLaMA3 and 7-8% on Qwen3, and increasing throughput by up to 15-19% over the strongest baseline. Finally, we provide theoretical insight and empirical evidence showing that decoder layers naturally align mask-token representations with next-token states, enabling accurate multi-step prediction without retraining or auxiliary models.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 36 of its 43 references that are in the map. First posted on arXiv 18 Mar 2026.
Builds on
36 of 43 references.
References · 36
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 843
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 370
24 more
- 2024Better & Faster Large Language Models via Multi-token Predictioncited by 360
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 331
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 307
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decodingcited by 298
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decodingcited by 269
- 2020ProphetNet: Predicting Future N-gram for Sequence-to-Sequence Pre-trainingcited by 222
- 2023SpecTr: Fast Speculative Decoding via Optimal Transportcited by 129
- 2023Accelerating Transformer Inference for Translation via Parallel Decodingcited by 124
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 121
- 2025Accelerating Diffusion LLMs via Adaptive Parallel Decodingcited by 112
- 2024The rising costs of training frontier AI modelscited by 95
- 2024SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Accelerationcited by 72
- 2023PaSS: Parallel Speculative Samplingcited by 49
- 2024Do language models plan ahead for future tokens?cited by 49
- 2025Your LLM Knows the Future: Uncovering Its Multi-Token Prediction Potentialcited by 47
- 2024Speculative Streaming: Fast LLM Inference without Auxiliary Modelscited by 38
- 2024Multi-Candidate Speculative Decodingcited by 31
- 2024Recursive Speculative Decoding: Accelerating LLM Inference via Sampling Without Replacementcited by 27
- 2024Direct Alignment of Draft Model for Speculative Decoding with Chat-Fine-Tuned LLMscited by 22
- 2024Hardware-Aware Parallel Prompt Decoding for Memory-Efficient Acceleration of LLM Inferencecited by 16
- 2025Multi-Token Prediction Needs Registerscited by 16
- 2024BiTA: Bi-Directional Tuning for Lossless Acceleration in Large Language Modelscited by 13
- 2025On multi-token prediction for efficient LLM inferencecited by 7
- 2025Accelerated Test-Time Scaling with Model-Free Speculative Samplingcited by 2
Built on by
Not cited yet.