Fast-Decoding Diffusion Language Models via Progress-Aware Confidence Schedules
On the map: Diffusion models → Discrete and text diffusion
also in 1 more
Abstract (arXiv)
Diffusion large language models (dLLMs) offer a promising alternative to autoregressive models, but their practical utility is severely hampered by slow, iterative sampling. We present SchED, a training-free, model-agnostic early-exit algorithm that aggregates full-span logit margins and halts decoding once a smooth, progress-dependent confidence threshold is met. We evaluated SchED on two dLLM families (Dream and LLaDA), in base and instruction-tuned variants across ten benchmarks spanning downstream tasks including multiple-choice question answering (MCQ), math, long-form QA/summarization, and translation. SchED delivers large, stable accelerations: on instruction-tuned models, it achieves 3.8-4.0× speedups while retaining 99.8-100% of the baseline score on average. On base models, SchED yields consistent speedup gains with 99.1-100% performance retention, with up to 2.34× under more aggressive settings. Using a conservative speed metric that heavily penalizes quality loss (QPS, γ=4), we show that SchED is robust and clearly outperforms prior confidence-based early-exit methods, which break down on long-form generation. An entropy analysis of the model's token predictions reveals that instruction tuning speeds up the decay of predictive entropy. By turning genuine confidence stabilization into computational savings, SchED makes dLLM decoding substantially more efficient.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 35 of its 39 references that are in the map. First posted on arXiv 2 Dec 2025.
Builds on
35 of 39 references.
Closest ancestors · 15
Same direction, newest first
- 2025LLaDA-MoE: A Sparse MoE Diffusion Language Model29 Sep
- 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Models28 Sep
- 2025FS-DFM: Fast and Accurate Long Text Generation with Few-Step Diffusion Language Models24 Sep
- 2025Diffusion Language Models Know the Answer Before Decoding27 Aug
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Models12 Aug
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inference4 Aug
- 2025The Diffusion Duality12 Jun
- 2025DLM-One: Diffusion Language Models for One-Step Sequence Generation30 May
- 2025dKV-Cache: The Cache for Diffusion Language Models21 May
- 2025Large Language Diffusion Models14 Feb
- 2024Scaling up Masked Diffusion Models on Text24 Oct
3 more
Other references · 20
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2023A Survey on Multimodal Large Language Modelscited by 898
- 2019Multi-News: a Large-Scale Multi-Document Summarization Dataset and Abstractive Hierarchical Modelcited by 431
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 178
8 more
- 2025Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Principlescited by 43
- 2025Self Speculative Decoding for Diffusion Large Language Modelscited by 29
- 2025d2Cache: Accelerating Diffusion-Based LLMs via Dual Adaptive Cachingcited by 24
- 2025Structuring The Future: Diffusion LLM Speculative Decoding via Calibrated Draft Graphscited by 21
- 2025Accelerating Diffusion LLM Inference via Local Determinism Propagationcited by 21
- 2025UltraLLaDA: Scaling the Context Length to 128K for Diffusion Large Language Modelscited by 13
- 2025CreditDecoding: Accelerating Parallel Decoding in Diffusion Large Language Models with Trace Creditcited by 11
- 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Modelscited by 5
Built on by
Citing papers per year: 2026: 10.
10 papers.
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLMcited by 2
- 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniquescited by 1
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decodingcited by 0
- 2026Multi-Token Residual Predictioncited by 0
- 2026Forward-Free Diffusion Language Models with BPTT-Free Looped Refinementcited by 0
- 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Modelscited by 0
- 2026Commit Locally, Exit Globally: Coordinating Adaptive Sampling and Early Exit in Diffusion Language Modelscited by 0
- 2026CARVE: Verified Expansion for Variable-Length Generation in Diffusion Language Modelscited by 0
- 2026LOCKR: A Hidden-State Trajectory-Guided Planner for Detecting and Repairing Stable-but-Wrong Lock-In in Diffusion Language Modelscited by 0
- 2026HyperZip: Efficient Data Compression through Personalized Diffusion LLMs with Hypernetworkscited by 0
- 2026HyperZip: Efficient Data Compression through Personalized Diffusion LLMs with Hypernetworks28 Sep 2026
- 2026LOCKR: A Hidden-State Trajectory-Guided Planner for Detecting and Repairing Stable-but-Wrong Lock-In in Diffusion Language Models23 Sep 2026
- 2026CARVE: Verified Expansion for Variable-Length Generation in Diffusion Language Models31 Aug 2026
- 2026Commit Locally, Exit Globally: Coordinating Adaptive Sampling and Early Exit in Diffusion Language Models30 Jul 2026
- 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniques14 Jul 2026
- 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Models15 Jun 2026
- 2026Forward-Free Diffusion Language Models with BPTT-Free Looped Refinement6 Jun 2026
- 2026Multi-Token Residual Prediction12 May 2026
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLM10 May 2026
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decoding4 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2025From Bits to Rounds: Parallel Decoding with Exploration for Diffusion Language Models
- 2025Beyond Static Cutoffs: One-Shot Dynamic Thresholding for Diffusion Language Models
- 2026CAI-DLLM: Convergence Aware Inference for Diffusion Language Models
- 2025Sequential Diffusion Language Models
- 2025Fast and Fluent Diffusion Language Models via Convolutional Decoding and Rejective Fine-tuning
- 2025Fast-dLLM v2: Efficient Block-Diffusion LLM
- 2025Learning to Parallel: Accelerating Diffusion Large Language Models via Learnable Parallel Decoding
- 2025A Survey on Diffusion Language Models