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
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Code Generation | HumanEval 0-shot | Accuracy51.83 | 69 | |
| Mathematical Reasoning | GSM8k 5-shot | Accuracy81.2 | 54 | |
| Question Answering | TriviaQA | Accuracy52.08 | 41 | |
| Code Generation | MBPP 3-shot | Accuracy54.4 | 33 | |
| Math Reasoning | MATH 4-shot | Accuracy33.04 | 33 | |
| Reasoning | GSM8k 5-shot | Accuracy74.07 | 12 |