让视觉语言模型注意力更贴合信息流,减少幻觉。
Mitigating Hallucination in Large Vision-Language Models through Aligning Attention Distribution to Information Flow
- 通过分析注意力分布与信息流的错位,定位关键语义注意力头。
- 两阶段优化使模型注意力对齐真实信息流动,显著降低幻觉率。
- 可手动调节保守程度,适配不同应用场景需求。
由于单向掩码机制,Decoder-Only模型信息从左到右传播。视觉语言模型(LVLMs)采用相同架构,视觉信息在前向传播中逐步融入语义表征。系统分析发现,大部分视觉信息被吸收进语义表征,但模型注意力分布未充分聚焦于这些核心语义内容。这种注意力分布与实际信息流之间的错位削弱了模型的视觉理解能力,导致幻觉。为解决此问题,我们利用语义表征中的核心信息增强模型视觉理解:首先基于注意力分布识别聚焦核心语义的注意力头;再通过两阶段优化,将这些注意力头的优势传播至整个模型,实现注意力分布与信息流对齐。我们在三个图像描述基准上使用五种不同的LVLMs进行评估,证明该方法能显著减少幻觉。进一步实验显示,幻觉减少与细节丰富性之间存在权衡。值得注意的是,本方法支持手动调节模型保守性,实现灵活控制以满足多样化的现实需求。
原文摘要 · Abstract (English)
Due to the unidirectional masking mechanism, Decoder-Only models propagate information from left to right. LVLMs (Large Vision-Language Models) follow the same architecture, with visual information gradually integrated into semantic representations during forward propagation. Through systematic analysis, we observe that the majority of the visual information is absorbed into the semantic representations. However, the model's attention distribution does not exhibit sufficient emphasis on semantic representations. This misalignment between the attention distribution and the actual information flow undermines the model's visual understanding ability and contributes to hallucinations. To address this issue, we enhance the model's visual understanding by leveraging the core information embedded in semantic representations. Specifically, we identify attention heads that focus on core semantic representations based on their attention distributions. Then, through a two-stage optimization paradigm, we propagate the advantages of these attention heads across the entire model, aligning the attention distribution with the actual information flow. We evaluate our method on three image captioning benchmarks using five different LVLMs, demonstrating its effectiveness in significantly reducing hallucinations. Further experiments reveal a trade-off between reduced hallucinations and richer details. Notably, our method allows for manual adjustment of the model's conservativeness, enabling flexible control to meet diverse real-world requirements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。