Attention-Aware Remasking: A Training-Free Approach for Enhancing Visual Fidelity in Multimodal Diffusion Language Models

Yechan Cha, Dong-Kyu Chae


Abstract
Multimodal diffusion Large Language Models (dMLLMs) emerge as a compelling alternative to traditional autoregressive paradigms. But they still suffer from hallucinations driven by an inherent language bias. Through an analysis of the dMLLM inference dynamics, we observe that visual tokens frequently receive disproportionately low attention scores. We identify that the conventional low-confidence remasking strategy redundantly leverages logit-based probabilities during the token selection process, thereby exacerbating this language bias. To address this vulnerability, we propose Attention-Aware Remasking, a training-free, plug-and-play strategy that incorporates newly defined visual confidence scores to select the final output tokens, effectively promoting visual grounding. To enhance the logical integrity and inference efficiency of dLLMs, we further propose clue-based answering and echo-filling. Evaluations on representative Visual-Language Natural Language Explanation (VL-NLE) tasks, including VQA-X and e-SNLI-VE, demonstrate that our method achieves the highest BERTScore on both VQA-X and e-SNLI-VE, and the best answer accuracy on VQA-X among the compared methods. Our study reveals that the widely adopted logit-based remasking is not necessarily the optimal approach, highlighting promising new avenues for visually grounded inference in dMLLMs. Our code is available at: https://github.com/yechancha/Attention-aware-Remasking.
Anthology ID:
2026.findings-emnlp.765
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:
16137–16151
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.765/
DOI:
Bibkey:
Cite (ACL):
Yechan Cha and Dong-Kyu Chae. 2026. Attention-Aware Remasking: A Training-Free Approach for Enhancing Visual Fidelity in Multimodal Diffusion Language Models. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 16137–16151, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Attention-Aware Remasking: A Training-Free Approach for Enhancing Visual Fidelity in Multimodal Diffusion Language Models (Cha & Chae, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.765.pdf