Attention Hijackers: Detect and Disentangle Attention Hijacking in LVLMs for Hallucination Mitigation
Beitao Chen, Xinyu Lyu, Lianli Gao, Jingkuan Song, Heng Tao Shen
TL;DR
The paper tackles LVLM hallucinations by revealing Attention Hijacking: instruction tokens can distort visual attention during decoding, producing hallucinations. It introduces AID, a training-free framework with Attention Hijackers Detection, Attention Disentanglement, and Re-Disentanglement to identify and neutralize hijacker influence while preserving beneficial instruction guidance. Across multiple benchmarks (POPE, CHAIR, MME, MM-Vet, MMBench) and backbones, AID markedly reduces hallucinations without extra training and shows robust generalization and ablations supporting its components. The approach enhances reliability of multimodal generation and suggests combining with retrieval-based methods for further factual accuracy improvements.
Abstract
Despite their success, Large Vision-Language Models (LVLMs) remain vulnerable to hallucinations. While existing studies attribute the cause of hallucinations to insufficient visual attention to image tokens, our findings indicate that hallucinations also arise from interference from instruction tokens during decoding. Intuitively, certain instruction tokens continuously distort LVLMs' visual perception during decoding, hijacking their visual attention toward less discriminative visual regions. This distortion prevents them integrating broader contextual information from images, ultimately leading to hallucinations. We term this phenomenon 'Attention Hijacking', where disruptive instruction tokens act as 'Attention Hijackers'. To address this, we propose a novel, training-free strategy namely Attention HIjackers Detection and Disentanglement (AID), designed to isolate the influence of Hijackers, enabling LVLMs to rely on their context-aware intrinsic attention map. Specifically, AID consists of three components: First, Attention Hijackers Detection identifies Attention Hijackers by calculating instruction-driven visual salience. Next, Attention Disentanglement mechanism is proposed to mask the visual attention of these identified Hijackers, and thereby mitigate their disruptive influence on subsequent tokens. Finally, Re-Disentanglement recalculates the balance between instruction-driven and image-driven visual salience to avoid over-masking effects. Extensive experiments demonstrate that AID significantly reduces hallucination across various LVLMs on several benchmarks.
