Beyond Prefix Answering: Parallel Infilling Reading for Long Contexts with Diffusion Language Models

Han Peng, Zican Dong, Peiyu Liu, Jingyuan Zou, Pengfei Li, Xin Zhao


Abstract
Diffusion Language Models (DLMs) have emerged as a promising alternative to autoregressive language models, showing strong non-autoregressive generation abilities. Although the masked denoising objective trains DLMs to recover tokens at arbitrary positions, current long-context usage still mostly follows a prefix-based paradigm: the full context is given as a prefix, and the answer is generated only at the end, leaving the model’s infilling ability underutilized. To investigate whether DLMs can read long contexts by infilling, we introduce Parallel Infilling Reading (PIR), a training-free method that inserts extraction slots into long contexts to generate intermediate reading states. Our empirical study shows that naive PIR improves over prefix-based DLM inference, but suffers from slot repetition, where multiple slots generate redundant evidence. To mitigate this issue, we propose Slot-Isolated PIR, which blocks direct slot-to-slot attention, and PIR-SFT, which uses fine-grained evidence supervision to teach DLMs effective evidence extraction while reading. Experiments on long-context question answering benchmarks show that our methods improve answer accuracy and generation efficiency, with increasingly larger gains as the context length grows.
Anthology ID:
2026.emnlp-main.282
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:
6463–6476
Language:
URL:
https://aclanthology.org/2026.emnlp-main.282/
DOI:
Bibkey:
Cite (ACL):
Han Peng, Zican Dong, Peiyu Liu, Jingyuan Zou, Pengfei Li, and Xin Zhao. 2026. Beyond Prefix Answering: Parallel Infilling Reading for Long Contexts with Diffusion Language Models. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 6463–6476, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Beyond Prefix Answering: Parallel Infilling Reading for Long Contexts with Diffusion Language Models (Peng et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.282.pdf