Efficient Diffusion LLMs via Temporal-Spatial Parallel Decoding and Confidence Extrapolation
On the map: Efficient LLM inference and serving → Diffusion LLM and MoE inference
Abstract (arXiv)
Diffusion-based large language models (dLLMs) support parallel text generation via iterative denoising, yet inference remains latency-heavy because many steps are spent on redundant refinement and repeated remasking of tokens whose final values are already determined. Prior acceleration methods mainly depend on step-local confidence heuristics or fixed schedules, which are sensitive to prompt and task variation and ignore strong positional effects within a sequence. We cast diffusion decoding as a dynamic control problem and show that token-wise denoising trajectories provide the key signal for reliable control. We propose a trace-aware decoding framework with two components. First, Temporal-Spatial Parallel Decoding (TSPD) uses a lightweight temporalspatial controller that consumes per-token trajectory features, including confidence, entropy, and momentum, together with token position, to decide when a token has converged and can be safely fixed. Second, we introduce Confidence Extrapolation (CE), a training-free state-space module that forecasts future logit trends with uncertainty to support proactive decisions, including safe look-ahead and targeted stabilization when trajectories are oscillatory or underconfident. Together, TSPD and CE reduce unnecessary denoising iterations while preserving output quality, and they compose cleanly with system optimizations such as KV caching.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion LLM and MoE inference, one of 12 directions in Efficient LLM inference and serving. Builds on 45 of its 51 references that are in the map. First posted on arXiv 29 May 2026.
Builds on
45 of 51 references.
Closest ancestors · 7
Same direction, newest first
- 2025Accelerating Diffusion LLM Inference via Local Determinism Propagation8 Oct
- 2025CreditDecoding: Accelerating Parallel Decoding in Diffusion Large Language Models with Trace Credit7 Oct
- 2025dParallel: Learnable Parallel Decoding for dLLMs30 Sep
- 2025Learning to Parallel: Accelerating Diffusion Large Language Models via Learnable Parallel Decoding29 Sep
- 2025Diffusion LLMs Can Do Faster-Than-AR Inference via Discrete Diffusion Forcing8 Aug
- 2025Accelerating Diffusion LLMs via Adaptive Parallel Decoding31 May
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding28 May
Other references · 38
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
26 more
- 2025Large Language Diffusion Modelscited by 872
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
- 2016Bidirectional Attention Flow for Machine Comprehensioncited by 278
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Modelscited by 268
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 178
- 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 168
- 2025Fast-dLLM v2: Efficient Block-Diffusion LLMcited by 128
- 2024MiniCache: KV Cache Compression in Depth Dimension for Large Language Modelscited by 122
- 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusioncited by 77
- 2025Diffusion Language Models Know the Answer Before Decodingcited by 68
- 2025LLaDA-MoE: A Sparse MoE Diffusion Language Modelcited by 54
- 2025Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Principlescited by 43
- 2025Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Modelscited by 33
- 2025dInfer: An Efficient Inference Framework for Diffusion Language Modelscited by 33
- 2024Autoregressive Large Language Models are Computationally Universalcited by 14
- 2025Beyond Confidence: Adaptive and Coherent Decoding for Diffusion Language Modelscited by 9
- 2025Decoding Large Language Diffusion Models with Foreseeing Movementcited by 6
- 2023Bidirectional Attention as a Mixture of Continuous Word Expertscited by 1
- 2025Learning to Look at the Other Side: A Semantic Probing Study of Word Embeddings in LLMs with Enabled Bidirectional Attentioncited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Trajectory-Level Speculative Decoding for Diffusion Language Models
- 2026Roll Out and Roll Back: Diffusion LLMs are Their Own Efficiency Teachers
- 2026PSD: Pushing the Pareto Frontier of Diffusion LLMs via Parallel Speculative Decoding
- 2026Early-Bird Decoding: Accelerating Diffusion LLMs with Learnable Block Sizes and Parallel Sampling
- 2026Stability-Weighted Decoding for Diffusion Language Models
- 2026Dynamic-dLLM: Dynamic Cache-Budget and Adaptive Parallel Decoding for Training-Free Acceleration of Diffusion LLM
- 2025From Bits to Rounds: Parallel Decoding with Exploration for Diffusion Language Models
- 2026Mean-Field Parallel Decoding for Discrete Diffusion Language Models