Parallelism and Generation Order in Masked Diffusion Language Models: Limits Today, Potential Tomorrow
Yangyang Zhong, Yanmei Gu, Zhengqing Zang, Xiaomeng Li, Yuqi Ding, Xibei Jia, Yuting Shen, Zhenzhong Lan, Liwang Zhu, Weiping Liu, Junlin Zhou, Haisheng Liu, Zhong Xin Yu, Pengxin Luo, Donglian Qi, Yunfeng Yan, Junbo Zhao
TL;DR
This paper systematically analyzes Masked Diffusion Language Models (MDLMs) along two degrees of freedom—parallelism and generation order—using Average Finalization Parallelism (AFP) and Kendall’s $\tau$, across 58 benchmarks and eight 100B-scale models. It demonstrates that current MDLMs, despite potential for parallelism, lag autoregressive models due to a fundamental parallelization-induced dependency loss, quantified via a conditional total correlation bound that grows with block size. The authors show adaptive decoding dynamics across domains, with higher parallelism correlating with correct outputs and domain-dependent order patterns, and they reveal non-monotonic behaviors such as order disruptions at semantic pivots and improvements on Sudoku-like tasks that leverage non-sequential reasoning. The work proposes a Generate-then-Edit paradigm to mitigate dependency loss while preserving parallel decoding efficiency, providing both theoretical proofs and practical insights for bridging the accuracy gap. Overall, the study offers a comprehensive framework for understanding and unlocking the latent potential of MDLMs in non-linear, efficient language generation and reasoning.
Abstract
Masked Diffusion Language Models (MDLMs) promise parallel token generation and arbitrary-order decoding, yet it remains unclear to what extent current models truly realize these capabilities. We characterize MDLM behavior along two dimensions -- parallelism strength and generation order -- using Average Finalization Parallelism (AFP) and Kendall's tau. We evaluate eight mainstream MDLMs (up to 100B parameters) on 58 benchmarks spanning knowledge, reasoning, and programming. The results show that MDLMs still lag behind comparably sized autoregressive models, mainly because parallel probabilistic modeling weakens inter-token dependencies. Meanwhile, MDLMs exhibit adaptive decoding behavior: their parallelism and generation order vary significantly with the task domain, the stage of reasoning, and whether the output is correct. On tasks that require "backward information" (e.g., Sudoku), MDLMs adopt a solution order that tends to fill easier Sudoku blanks first, highlighting their advantages. Finally, we provide theoretical motivation and design insights supporting a Generate-then-Edit paradigm, which mitigates dependency loss while retaining the efficiency of parallel decoding.
