Why Diffusion Language Models Struggle with Truly Parallel (Non-Autoregressive) Decoding?
Authors: Pengxiang Li, Dilxat Muhtar, Tianlong Chen, Lu Yin, Shiwei Liu
Organizations: The Hong Kong Polytechnic University, Hong Kong, China · ELLIS Institute Tübingen, Tübingen, Germany · Max Planck Institute for Intelligent Systems, Tübingen, Germany · Tübingen AI Center, Tübingen, Germany · The University of North Carolina at Chapel Hill, Chapel Hill, NC, USA · University of Surrey, Guildford, United Kingdom
Diffusion Language Models (DLMs) are often advertised as enabling parallel token generation, yet practical fast DLMs frequently converge to left-to-right, autoregressive (AR)-like decoding dynamics. In contrast, genuinely non-AR generation is promising because it removes AR's sequential bottleneck, better exploiting parallel hardware to reduce synchronization/communication overhead and improve latency scaling with output length. We argue that a primary driver of AR-like decoding is a mismatch between DLM objectives and the highly sequential structure of widely used training data, including standard pretraining corpora and long chain-of-thought (CoT) supervision. Motivated by this diagnosis, we propose NAP (Non-Autoregressive Parallel DLMs), a proof-of-concept, data-centric approach that better aligns supervision with non-AR parallel decoding. NAP curates examples as multiple independent reasoning trajectories and couples them with a parallel-forced decoding strategy that encourages multi-token parallel updates. Across math reasoning benchmarks, NAP yields stronger performance under parallel decoding than DLMs trained on standard long CoT data, with gains growing as parallelism increases. Our results suggest that revisiting data and supervision is a principled direction for mitigating AR-like behavior and moving toward genuinely non-autoregressive parallel generation in DLMs. Our code is available at https://github.com/pixeli99/NAP.
Figures & tables
Figure 1: Visualization of decoding dynamics. We plot the token position being unmasked (y-axis) against the decoding step (x-axis). (a, b) Despite using confidence-based Arbitrary Order (AO) decoding, standard DLMs (LLaDA and Dream) exhibit a strict linear diagonal pattern, revealing that their behavior collapses into autoregressive (left-to-right) generation. (c) Random decoding eliminates AR bias but lacks structure. (d) Our method (NAP) breaks the single-stream bottleneck, generating multiple reasoning trajectories simultaneously.
Figure 2: Performance on GSM8K (left) and MATH-500 (right). Forcing low-ARness behavior (Random decoding) generally causes reasoning performance to collapse. Notably, for LLaDA, we employ a constrained block-wise decoding strategy to ensure generation validity. This preserves local structural integrity, resulting in the Arbitrary Order (AO) decoding maintaining comparable performance, unlike the sharp drop observed in fully unstructured random decoding.
Figure 3: Sequential Dependence (SeqDep) Analysis on (a) OpenR1-Math and (b) FineWeb Datasets. The consistently high and rising SeqDep scores indicate that standard training corpora possess strong intrinsic sequentiality, driving models to internalize AR-like dependencies.
Model
AR Order
AO (Conf)
ARness
Acc
ARness
Acc
LLaDA-8B ( Nie et al., 2025c )
1.00
71.9
0.73
51.9
+ Fast-dLLM ( Wu et al., 2025 )
1.00
71.9
0.87
51.6
Dream-7B ( Ye et al., 2025 )
1.00
78.2
0.92
78.2
+ Fast-dLLM ( Wu et al., 2025 )
1.00
78.3
0.94
78.1
Table 1: Quantifying Autoregressive Bias (ARness) and Accuracy. Comparison of sequential bias and performance across different decoding strategies. While AR Order implies strict sequentiality (1.00), AO (Conf) maintains high ARness and competitive accuracy.
Model
Base (Pretrained)
Long-CoT (SFT)
Change
LLaDA-8B
0.73
0.81
↑ 0.08
Dream-7B
0.92
0.93
↑ 0.01
Table 2: Long-CoT Supervision Increases ARness. Comparison of Global ARness@1 scores (using AO decoding) before and after fine-tuning.
Figure 4: Long-CoT Supervision Increases ARness. The positive deltas show models converging toward strict left-to-right generation (1.0), confirming that current supervision methods actively discourage non-autoregressive parallel decoding.
Figure 6: Overview of the parallel-forced decoding framework . The model concurrently generates multiple independent reasoning paths within structured thinking blocks. These parallel trajectories are then synthesized into a result within a designated summary block.
Benchmark
Steps
Tok/Step
LLaDA 8B
LLaDA 8B (Long-CoT)
NAP-LLaDA 8B
Dream-7B
Dream-7B (Long-CoT)
NAP-Dream-7B
Mathematics & Scientific
GSM8K
256
4
46.4
54.1
56.1 (+2.0)
35.0
46.5
60.9 (+14.4)
336
3
54.4
60.9
63.3 (+2.4)
49.4
56.9
70.9 (+14.0)
512
2
62.0
82.0
82.6 (+0.4)
58.5
66.8
79.2 (+12.4)
1024
1
66.5
83.5
84.1 (+0.6)
68.9
78.0
83.6 (+5.6)
MATH-500
256
4
17.8
21.4
26.6 (+5.2)
8.8
16.2
23.8 (+7.6)
Table 3: Benchmark results on LLaDA-8B-Instruct and Dream-7B-Instruct under different step budgets. Tok/Step denotes the number of tokens decoded per decoding step; larger Tok/Step corresponds to higher decoding parallelism.
Training Data
Decoding
256
512
1024
Base (Pretrained)
AO
35.0
58.5
68.9
Base (Pretrained)
Parallel-Forced
31.0
52.6
60.2
NAP (Ours)
AO
57.4
78.9
85.1
NAP (Ours)
Parallel-Forced
60.9
79.2
83.6
Table 4: GSM8K accuracy using Dream-7B. Simply applying parallel decoding to a base model hurts performance; gains require aligned supervision.
Method
1 Path
2 Paths
3 Paths
NAP-Dream
75.4
78.9
83.6
NAP-LLaDA
79.4
82.6
84.1
Table 5: Accuracy on GSM8K with varying m . Total token budget is fixed.
Figure 7: SeqDep Analysis on Dparallel . We visualize the Sequential Dependence (SeqDep) of our curated parallel reasoning data against token length. The green curve (binned mean) shows that SeqDep remains stable and relatively low across varying lengths.
Diffusion language models (DLMs) have emerged as a promising alternative to autoregressive (AR) models, offering sub-linear generation latency and bidirectional capabilities that are particularly appealing for code generation and editing. Achieving sub-linear latency in discrete DLMs requires predicting multiple tokens in parallel. However, standard DLMs sample tokens independently from conditional marginal distributions, failing to capture the joint dependencies among concurrently generated tokens. As a result, they often lead to syntactic inconsistencies and break multi-token structures. In this work, we introduce CoDiLA (Coherent Diffusion with Local Autoregression), a method that reconciles parallel sampling with local dependency modeling. Rather than forcing the DLM to resolve fine-grained syntax, CoDiLA delegates local decoding to a small, auxiliary AR model operating on the diffusion latents. This design allows for parallel generation while ensuring sequential validity within a block and maintaining core DLM capabilities, including bidirectional modeling across blocks. We demonstrate that using a highly compact auxiliary AR model (e.g., 0.6B parameters) effectively eliminates coherence artifacts, establishing a new Pareto frontier for accuracy and speed in code generation benchmarks.
Michael Hersche, Nicolas Menet, Ronan Tanios +1
IBM Research - Zurich · Department of Computer Science, ETH Zürich
While most autoregressive LLMs are constrained to one-by-one decoding, diffusion LLMs (dLLMs) have attracted growing interest for their potential to dramatically accelerate inference through parallel decoding. Despite this promise, the conditional independence assumption in dLLMs causes parallel decoding to ignore token dependencies, inevitably degrading generation quality when these dependencies are strong. However, existing works largely overlook these inherent challenges, and evaluations on standard benchmarks (e.g., math and coding) are not sufficient to capture the quality degradation caused by parallel decoding. To address this gap, we first provide an information-theoretic analysis of parallel decoding. We then conduct case studies on analytically tractable synthetic list operations from both data distribution and decoding strategy perspectives, offering quantitative insights that highlight the fundamental limitations of parallel decoding. Building on these insights, we propose ParallelBench, the first benchmark specifically designed for dLLMs, featuring realistic tasks that are trivial for humans and autoregressive LLMs yet exceptionally challenging for dLLMs under parallel decoding. Using ParallelBench, we systematically analyze both dLLMs and autoregressive LLMs, revealing that: (i) dLLMs under parallel decoding can suffer dramatic quality degradation in real-world scenarios, and (ii) current parallel decoding strategies struggle to adapt their degree of parallelism based on task difficulty, thus failing to achieve meaningful speedup without compromising quality. Our findings underscore the pressing need for innovative decoding methods that can overcome the current speed-quality trade-off. We release our benchmark to help accelerate the development of truly efficient dLLMs.
Wonjun Kang, Kevin Galim, Seunghyuk Oh +8
FuriosaAI · Seoul National University · UW-Madison +4
Diffusion language models promise parallel generation, yet still lag behind autoregressive (AR) models in quality. We stem this gap to a failure of introspective consistency: AR models agree with their own generations, while DLMs often do not. We define the introspective acceptance rate, which measures whether a model accepts its previously generated tokens. This reveals why AR training has a structural advantage: causal masking and logit shifting implicitly enforce introspective consistency. Motivated by this observation, we introduce Introspective Diffusion Language Model (I-DLM), a paradigm that retains diffusion-style parallel decoding while inheriting the introspective consistency of AR training. I-DLM uses a novel introspective strided decoding (ISD) algorithm, which enables the model to verify previously generated tokens while advancing new ones in the same forward pass. From a systems standpoint, we build I-DLM inference engine on AR-inherited optimizations and further customize it with a stationary-batch scheduler. To the best of our knowledge, I-DLM is the first DLM to match the quality of its same-scale AR counterpart while outperforming prior DLMs in both model quality and practical serving efficiency across 15 benchmarks. It reaches 69.6 on AIME-24 and 45.7 on LiveCodeBench-v6, exceeding LLaDA-2.1-mini (16B) by more than 26 and 15 points, respectively. Beyond quality, I-DLM is designed for the growing demand of large-concurrency serving, delivering about 3x higher throughput than prior state-of-the-art DLMs.
Yifan Yu, Yuqing Jian, Junxiong Wang +12
Together AI · University of Illinois Urbana-Champaign · Princeton University +2