arXiv:2410.04514cs.CLcs.CV2024-10EMNLP被引 64

通过分析注意力机制,减少大模型对背景的错误关注,有效降低视觉问答中的幻觉问题。

DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination

论文配图:DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination
图 1 · 摘自论文原文
  • 利用ViT的分类令牌过滤背景异常高注意力区域
  • 在不训练的前提下显著降低对象幻觉率,提升答案准确性
  • 适用于各类主流视觉语言模型,适合研究幻觉问题的开发者

尽管大型视觉语言模型(LVLMs)取得了巨大成功,但仍不可避免地出现幻觉。我们发现,LVLM中语言模型解码器对图像标记的注意力分布与视觉编码器高度一致,且均倾向于聚焦于特定背景标记而非所指物体。我们归因于视觉编码器本身存在固有缺陷,误导语言模型过度关注冗余信息,从而产生对象幻觉。为此,我们提出DAMRO——一种无需训练的新策略,深入分析并修正注意力机制以减少幻觉。具体而言,该方法利用ViT的分类标记(CLS)识别并过滤背景中分散的高注意力异常标记,并在解码阶段消除其影响。我们在包括LLaVA-1.5、LLaVA-NeXT和InstructBLIP在内的多个模型上,基于POPE、CHAIR、MME及GPT-4V辅助评估等基准进行测试,结果表明该方法能有效降低异常标记的影响,显著缓解幻觉现象。代码已开源:https://github.com/coder-gx/DAMRO。

原文摘要 · Abstract (English)

Despite the great success of Large Vision-Language Models (LVLMs), they inevitably suffer from hallucination. As we know, both the visual encoder and the Large Language Model (LLM) decoder in LVLMs are Transformer-based, allowing the model to extract visual information and generate text outputs via attention mechanisms. We find that the attention distribution of LLM decoder on image tokens is highly consistent with the visual encoder and both distributions tend to focus on particular background tokens rather than the referred objects in the image. We attribute to the unexpected attention distribution to an inherent flaw in the visual encoder itself, which misguides LLMs to over emphasize the redundant information and generate object hallucination. To address the issue, we propose DAMRO, a novel training-free strategy that $D$ive into $A$ttention $M$echanism of LVLM to $R$educe $O$bject Hallucination. Specifically, our approach employs classification token (CLS) of ViT to filter out high-attention outlier tokens scattered in the background and then eliminate their influence during decoding stage. We evaluate our method on LVLMs including LLaVA-1.5, LLaVA-NeXT and InstructBLIP, using various benchmarks such as POPE, CHAIR, MME and GPT-4V Aided Evaluation. The results demonstrate that our approach significantly reduces the impact of these outlier tokens, thus effectively alleviating the hallucination of LVLMs. The code is released at https://github.com/coder-gx/DAMRO.

视觉语言模型注意力机制幻觉抑制ViT

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。