Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Fast-Decoding Diffusion Language Models via Progress-Aware Confidence Schedules

About

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\times$ 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\times$ under more aggressive settings. Using a conservative speed metric that heavily penalizes quality loss (QPS, $\gamma{=}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.

Amr Mohamed, Yang Zhang, Michalis Vazirgiannis, Guokan Shang• 2025

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy86.16
1460
Physical ReasoningPIQA
Accuracy73.25
34
Language UnderstandingMMLU
Accuracy80.81
31
Question AnsweringGPQA
Accuracy30.8
24
Commonsense ReasoningWinoGrande
Accuracy0.8624
19
Machine TranslationWMT EN-DE 16
chrF49.86
10
SummarizationMultiNews
ROUGE Score24.6
10
Machine TranslationWMT en-fr 14
chrF55.82
10
Mathematical ReasoningGSM8K
Accuracy74.51
10
Question AnsweringHotpotQA
F1 Score27.48
10
Showing 10 of 18 rows

Other info

Follow for update