MMODELYST
Papers/Cluster-Level Attention-Guided Parallel Decoding for Masked Diffusion Language Models
PAP

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

May 28, 2026

arXiv
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 (Cluster-Level Attention-Guided Decoding), 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.77x--8.47x speedups over Vanilla decoding while maintaining broadly comparable task accuracy in most settings.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Heqiang Qi, Wei Huang, Mingyuan Bai, Xiangming Meng
Your notes (browser-local)
saved
arXiv:2605.29607