Decoding Diffusion Language Models with Foreseeing Movement

Yichuan Mo, Quan Chen, Mingjie Li, Zeming Wei, Yisen Wang


Abstract
Diffusion Language Models (DLMs) benefit from a flexible decoding mechanism that enables parallelized inference and controllable generations over autoregressive models. Yet such flexibility introduces a critical challenge: inference performance becomes highly sensitive to the decoding order of tokens. Existing heuristic methods, however, focus mainly on local effects while overlooking long-term impacts. To address this limitation, we propose the Foreseeing Decoding Method (FDM), a novel approach that integrates both local and global considerations to unlock the full potential, employing a search-based strategy to enable effective optimization in discrete spaces. Furthermore, by analyzing the consistency of chosen tokens in the full decoding process, we develop a variant, FDM with Acceleration (FDM-A), which restricts deep exploration to critical steps identified as the exploration and balance circumstances. Extensive experiments across diverse benchmarks and model architectures validate the scalability of FDM and demonstrate the superior efficiency-performance trade-off achieved by FDM-A. Our work might potentially provide a principled step toward more powerful decoding methods for DLMs. Our code is available at https://github.com/PKU-ML/FDM.
Anthology ID:
2026.emnlp-main.909
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
20943–20958
Language:
URL:
https://aclanthology.org/2026.emnlp-main.909/
DOI:
Bibkey:
Cite (ACL):
Yichuan Mo, Quan Chen, Mingjie Li, Zeming Wei, and Yisen Wang. 2026. Decoding Diffusion Language Models with Foreseeing Movement. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 20943–20958, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Decoding Diffusion Language Models with Foreseeing Movement (Mo et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.909.pdf