Fast-Decoding Diffusion Language Models via Progress-Aware Confidence Schedules
Abstract
SchED, a training-free early-exit algorithm, accelerates diffusion large language model decoding with minimal performance loss across various tasks.
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.0times 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.34times 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.
Community
SchED introduces a training-free, early-exit decoding criterion for diffusion LLMs, halting sampling once a smooth, progress-adaptive confidence threshold is satisfied. SchED achieves up to ~4× decoding speedups on average with ≥99–100% performance retention across benchmarks.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- CDLM: Consistency Diffusion Language Models For Faster Sampling (2025)
- Latent Refinement Decoding: Enhancing Diffusion-Based Language Models by Refining Belief States (2025)
- Beyond Static Cutoffs: One-Shot Dynamic Thresholding for Diffusion Language Models (2025)
- From Bits to Rounds: Parallel Decoding with Exploration for Diffusion Language Models (2025)
- Beyond Confidence: Adaptive and Coherent Decoding for Diffusion Language Models (2025)
- Breaking the Bottleneck with DiffuApriel: High-Throughput Diffusion LMs with Mamba Backbone (2025)
- KLASS: KL-Guided Fast Inference in Masked Diffusion Models (2025)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment:
@librarian-bot
recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper