Cluster-Level Attention-Guided Parallel Decoding for Masked Diffusion Language Models

Heqiang Qi, Wei Huang, Mingyuan Bai, Xiangming Meng


Abstract
Masked diffusion language models (MDLMs) enable parallel decoding by predicting all masked positions at each denoising step, yet existing training-free samplers usually decide which positions to commit at token-level granularity. We revisit this granularity and observe that reliable predictions often emerge as contiguous high-confidence spans, suggesting that the unit of parallel commitment can be larger than a single token. We first group adjacent high-confidence candidates into confidence-induced clusters (CICs) as span-level update units. We then use self-attention maps from the same forward pass to estimate inter-cluster dependencies, enabling conflict-aware selection of mutually compatible CICs for parallel commitment. This yields CLAD (**C**luster-**L**evel **A**ttention-Guided **D**ecoding), a training-free cluster-level decoder for MDLMs. Experiments on LLaDA and Dream model families across four reasoning and code-generation benchmarks show that CLAD achieves 1.77×–8.47× speedups over Vanilla decoding while maintaining broadly comparable task accuracy in most settings.
Anthology ID:
2026.findings-emnlp.275
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
5785–5805
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.275/
DOI:
Bibkey:
Cite (ACL):
Heqiang Qi, Wei Huang, Mingyuan Bai, and Xiangming Meng. 2026. Cluster-Level Attention-Guided Parallel Decoding for Masked Diffusion Language Models. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 5785–5805, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Cluster-Level Attention-Guided Parallel Decoding for Masked Diffusion Language Models (Qi et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.275.pdf