HalluRNN: Mitigating Hallucinations via Recurrent Cross-Layer Reasoning in Large Vision-Language Models
Le Yu, Kaishen Wang, Jianlong Xiong, Yue Cao, Lei Zhang, Zhang Yi Tao He
TL;DR
HalluRNN tackles hallucinations in Large Vision-Language Models by introducing a cross-layer recurrent mechanism. It implements a Dual-Gated Depth Propagation Unit (DG-DPU) shared across Transformer layers, enabling adaptive information propagation and explicit modeling of inter-layer state evolution. The DG-DPU employs a Constraint Gate and a Correction Gate to control coarse-to-fine updates, and only the DG-DPU is fine-tuned for efficiency. Empirical results across POPE, MM-Vet, CHAIR, and MMMU benchmarks show HalluRNN consistently reduces hallucinations and transfers across models, highlighting its robustness and broad applicability.
Abstract
Though Large Vision-Language Models (LVLMs) have achieved remarkable performance across various tasks, they are still prone to hallucinations-generating outputs that are textually plausible but visually ungrounded. While prior approaches generally address this issue through data-centric fine-tuning or innovative decoding strategies, these methods often require substantial resources or task-specific configurations. In this work, we introduce an architecture-level solution, HalluRNN, which enhances model stability through recurrent cross-layer reasoning. Specifically, we propose a novel Dual-Gated Depth Propagation Unit (DG-DPU) module, which is shared across layers and recurrently refines hidden states. This allows for the adaptive propagation of information throughout the model, enforces consistency across layers, and mitigates hallucinations caused by representational drift. By fine-tuning only the DG-DPU module, HalluRNN achieves strong and robust performance across multiple benchmarks.
