通过层间视觉注意力差异,减少大模型生成时对视觉证据的遗忘,降低幻觉。
Finding the Correct Visual Evidence Without Forgetting: Mitigating Hallucination in LVLMs via Inter-Layer Visual Attention Discrepancy

- 利用不同层间对视觉信息的关注差异,识别出应被强调的关键视觉证据。
- 在生成过程中增强对关键视觉证据的注意力,有效缓解视觉信息遗忘问题。
- 无需训练、即插即用,适用于多种主流视觉语言模型,提升生成真实性。
大型视觉语言模型(LVLMs)在众多视觉语言任务中表现优异,但仍易产生与图像内容不符的幻觉。本文发现,当模型对正确视觉证据关注不足且在生成过程中逐渐遗忘时,幻觉更易发生。我们实证发现,尽管整体上模型对视觉证据关注不足,但在特定层中对正确证据仍具敏感性,存在显著的层间注意力差异。基于此,我们提出一种名为互层视觉注意力差异(ILVAD)的新方法:通过分析早期生成标记在各层对视觉标记的注意力权重,识别出反复激活的视觉证据,构建显著性图谱,并据此增强生成过程中的视觉注意力,减少视觉遗忘。同时,利用该图谱获取文本标记与视觉证据的关联得分,筛选并强化有视觉依据的文本部分。该方法无需训练、可即插即用。在五种最新发布的模型上的多基准评估表明,该方法能持续有效缓解不同架构下的幻觉问题。代码已开源。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) have shown remarkable performance on a wide range of vision-language tasks. Despite this progress, they are still prone to hallucination, generating responses that are inconsistent with visual content. In this work, we find that LVLMs tend to hallucinate when they pay insufficient attention to the correct visual evidence and gradually forget it during the generation process. We empirically find that although LVLMs overall attend insufficiently to visual evidence, they exhibit sensitivity to the correct visual evidence in specific layers, with notable inter-layer discrepancy. Motivated by this observation, we propose a novel hallucination mitigation method that enhances visual evidence based on Inter-Layer Visual Attention Discrepancy (ILVAD). Specifically, we obtain the attention weights from early generated tokens to visual tokens across layers and identify the tokens that are repeatedly activated as visual evidence, forming a saliency map. We then enhance attention to visual evidence during generation through the saliency map to reduce visual forgetting. In addition, we leverage the saliency map to obtain attention scores of generated text to visual evidence, in order to select and emphasize text tokens that are strongly grounded in visual evidence. Our method is training-free and plug-and-play. Multiple benchmark evaluations conducted on five recently released models show that our method can consistently mitigate hallucinations in different LVLMs over various architectures. Code is available at https://github.com/ytx-ML/ILVAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。