Adaptive Guidance for Retrieval-Augmented Masked Diffusion Models

Jaemin Kim, Jong Chul Ye


Abstract
Retrieval-Augmented Generation (RAG) improves factual grounding by incorporating external knowledge into language model generation. However, when retrieved context is noisy, unreliable, or inconsistent with the model’s parametric knowledge, it introduces retrieval-prior conflicts that can degrade generation quality. While this problem has been studied in autoregressive language models, it remains largely unexplored in diffusion-based language models, where the iterative denoising process introduces unique challenges for integrating retrieved context. In this work, we propose Adaptive Retrieval-Augmented Masked Diffusion (ARAM), a training-free adaptive guidance framework for Masked Diffusion Models (MDMs) in RAG settings. ARAM dynamically calibrates the guidance scale during denoising according to the Signal-to-Noise Ratio (SNR) of the distributional shift induced by retrieved context. Intuitively, the model strengthens guidance when the retrieved context provides reliable corrective evidence and suppresses it when the contextual signal is noisy or non-supportive. Extensive experiments on multiple knowledge-intensive QA benchmarks show that ARAM improves overall QA performance over decoding-based RAG baselines.
Anthology ID:
2026.findings-emnlp.80
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
1621–1636
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.80/
DOI:
Bibkey:
Cite (ACL):
Jaemin Kim and Jong Chul Ye. 2026. Adaptive Guidance for Retrieval-Augmented Masked Diffusion Models. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 1621–1636, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Adaptive Guidance for Retrieval-Augmented Masked Diffusion Models (Kim & Ye, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.80.pdf