Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DAPD: Dependency-Aware Parallel Decoding via Attention for Diffusion LLMs

About

Parallel decoding for Diffusion LLMs (dLLMs) is difficult because each denoising step provides only token-wise marginal distributions, while unmasking multiple tokens simultaneously requires accounting for inter-token dependencies. We propose Dependency-Aware Parallel Decoding (DAPD), a simple, training-free decoding method that uses self-attention to induce a conditional dependency graph over masked tokens. At each iteration, edges in this graph capture strong token interactions, while non-edges indicate weak dependence. Parallel decoding is then reduced to selecting an independent set on the graph and unmasking the selected tokens in parallel. This avoids co-updating strongly coupled tokens without auxiliary models or retraining. Experiments on LLaDA and Dream show that DAPD improves the accuracy-steps trade-off over existing methods and enables more globally distributed parallel updates that better exploit the any-order generation capability of dLLMs. The project is available at https://ai-isl.github.io/dapd

Bumjun Kim, Dongjae Jeon, Moongyu Jeon, Albert No• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval 0-shot
Accuracy51.83
69
Mathematical ReasoningGSM8k 5-shot
Accuracy81.2
54
Question AnsweringTriviaQA
Accuracy52.08
41
Code GenerationMBPP 3-shot
Accuracy54.4
33
Math ReasoningMATH 4-shot
Accuracy33.04
33
ReasoningGSM8k 5-shot
Accuracy74.07
12
Showing 6 of 6 rows

Other info

Follow for update