ES-dLLM: Efficient Inference for Diffusion Large Language Models by Early-Skipping
Not on the map. Why?
Abstract (arXiv)
Diffusion large language models (dLLMs) are emerging as a promising alternative to autoregressive models (ARMs) due to their ability to capture bidirectional context and the potential for parallel generation. Despite the advantages, dLLM inference remains computationally expensive as the full input context is processed at every iteration. In this work, we analyze the generation dynamics of dLLMs and find that intermediate representations, including key, value, and hidden states, change only subtly across successive iterations. Leveraging this insight, we propose ES-dLLM, a training-free inference acceleration framework for dLLM that reduces computation by skipping tokens in early layers based on the estimated importance. Token importance is computed with intermediate tensor variation and confidence scores of previous iterations. Experiments on LLaDA-8B and Dream-7B demonstrate that ES-dLLM achieves throughput of up to 226.57 and 308.51 tokens per second (TPS), respectively, on an NVIDIA H200 GPU, delivering 5.6× to 16.8× speedup over the vanilla implementation and up to 1.85× over the state-of-the-art caching method, while preserving generation quality.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 20 of its 21 references that are in the map. First posted on arXiv 10 Mar 2026.
Builds on
20 of 21 references.
References · 20
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
- 2025Large Language Diffusion Modelscited by 872
- 2023Splitwise: Efficient generative LLM inference using phase splittingcited by 536
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2024Taming Throughput-Latency Tradeoff in LLM Inference with Sarathi-Servecited by 507
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
8 more
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
- 2023A Comprehensive Survey of AI-Generated Content (AIGC): A History of Generative AI from GAN to ChatGPTcited by 221
- 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 168
- 2025dKV-Cache: The Cache for Diffusion Language Modelscited by 140
- 2023EE-LLM: Large-Scale Training and Inference of Early-Exit Large Language Models with 3D Parallelismcited by 69
- 2025Sparse-dLLM: Accelerating Diffusion LLMs with Dynamic Cache Evictioncited by 64
- 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
- 2025DPad: Efficient Diffusion Language Models with Suffix Dropoutcited by 38
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Elastic-dLLM: Position Preserving Context Compression and Augmentation of Diffusion LLMscited by 1
- 2026Enabling KV Caching of Shared Prefix for Diffusion Language Modelscited by 1
- 2026Efficient On-Device Diffusion LLM Inference with Mobile NPUcited by 1
- 2026Understanding Evaluation Illusion in Diffusion Large Language Modelscited by 0
- 2026CAI-DLLM: Convergence Aware Inference for Diffusion Language Modelscited by 0
- 2026Affix Cache for Diffusion Large Language Modelscited by 0
- 2026CAI-DLLM: Convergence Aware Inference for Diffusion Language Models23 Aug 2026
- 2026Understanding Evaluation Illusion in Diffusion Large Language Models28 Jun 2026
- 2026Affix Cache for Diffusion Large Language Models26 Jun 2026
- 2026Efficient On-Device Diffusion LLM Inference with Mobile NPU11 Jun 2026
- 2026Enabling KV Caching of Shared Prefix for Diffusion Language Models26 May 2026
- 2026Elastic-dLLM: Position Preserving Context Compression and Augmentation of Diffusion LLMs18 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniques
- 2025dInfer: An Efficient Inference Framework for Diffusion Language Models
- 2026Dynamic-dLLM: Dynamic Cache-Budget and Adaptive Parallel Decoding for Training-Free Acceleration of Diffusion LLM
- 2025Fast-dLLM v2: Efficient Block-Diffusion LLM
- 2026Streaming-dLLM: Accelerating Diffusion LLMs via Suffix Pruning and Dynamic Decoding
- 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion
- 2025A Survey on Diffusion Language Models
- 2026dLLM: Simple Diffusion Language Modeling