HaloProbe: Bayesian Detection and Mitigation of Object Hallucinations in Vision-Language Models
AI 摘要
HaloProbe通过贝叶斯框架检测并缓解视觉-语言模型中的物体幻觉问题,并提出了一种非侵入式的缓解方法。
主要贡献
- 揭示了基于注意力机制的幻觉检测的局限性
- 提出了HaloProbe贝叶斯框架,用于更准确地估计幻觉概率
- 提出了一种基于HaloProbe的非侵入式幻觉缓解方法
方法论
采用贝叶斯框架,分解外部描述统计信息和内部解码信号,结合平衡训练和学习到的先验来估计token级别的幻觉概率。
原文摘要
Large vision-language models can produce object hallucinations in image descriptions, highlighting the need for effective detection and mitigation strategies. Prior work commonly relies on the model's attention weights on visual tokens as a detection signal. We reveal that coarse-grained attention-based analysis is unreliable due to hidden confounders, specifically token position and object repetition in a description. This leads to Simpson's paradox: the attention trends reverse or disappear when statistics are aggregated. Based on this observation, we introduce HaloProbe, a Bayesian framework that factorizes external description statistics and internal decoding signals to estimate token-level hallucination probabilities. HaloProbe uses balanced training to isolate internal evidence and combines it with learned prior over external features to recover the true posterior. While intervention-based mitigation methods often degrade utility or fluency by modifying models' internals, we use HaloProbe as an external scoring signal for non-invasive mitigation. Our experiments show that HaloProbe-guided decoding reduces hallucinations more effectively than state-of-the-art intervention-based methods while preserving utility.