Predict, Don’t Iterate: Efficient Adaptive-Length Infilling for Diffusion Language Models

Haobo Xu, Sirui Chen, Yuanchen Bei, Lingjie Chen, Yuchen Yan, Dongqi Fu, Jingrui He, Hanghang Tong


Abstract
Diffusion language models (DLMs) have emerged as a promising alternative to the auto-regressive paradigm. With bidirectional attention and any-order generation, DLMs naturally fit infilling tasks, which require generating a middle span conditioned on both the prefix and the suffix. However, infilling is sensitive to the length of the span, while DLMs require the length to be fixed before generation. Although prior studies extend DLMs to dynamic lengths, they still suffer from two limitations. (i) *Sensitivity to initial length.* These methods require a preset length to initialize the search and are highly sensitive to this initial length, often yielding suboptimal results. (ii) *Inference inefficiency.* They either insert length-changing operations during generation or repeatedly search for an appropriate length using multi-step denoising confidence, both of which introduce substantial additional forward passes and computational cost. Therefore, we propose PILL (**P**robing-based **I**nfi**L**ing with preset-**L**ength-free decoding), an efficient infilling method for DLMs that requires no preset initial length and drastically reduces the number of forward passes compared to baselines. Experiments show that, across five DLMs spanning different families, architectures, and training recipes on eight infilling benchmarks, PILL improves over the strongest baseline by +4.8 average pass rate on code and +6.0 BLEU-2 on text, while running 1.82× faster than that baseline.
Anthology ID:
2026.emnlp-main.1053
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:
24171–24190
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1053/
DOI:
Bibkey:
Cite (ACL):
Haobo Xu, Sirui Chen, Yuanchen Bei, Lingjie Chen, Yuchen Yan, Dongqi Fu, Jingrui He, and Hanghang Tong. 2026. Predict, Don’t Iterate: Efficient Adaptive-Length Infilling for Diffusion Language Models. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 24171–24190, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Predict, Don’t Iterate: Efficient Adaptive-Length Infilling for Diffusion Language Models (Xu et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1053.pdf