用文本预测概率快速检测视觉模型幻觉,效率更高。
Leveraging NTPs for Efficient Hallucination Detection in VLMs
- 用视觉模型的下一个词概率作为不确定度信号,训练轻量级模型检测幻觉。
- 在1400条人工标注数据上,该方法性能接近强模型,且推理更快。
- 结合文本回流概率和模型预测分数,进一步提升检测效果,适合部署优化。
视觉语言模型(VLM)的幻觉问题——即生成文本与图像内容不一致——严重影响其可靠性。现有检测方法多依赖同一或另一模型评估输出,计算开销大、延迟高。本文提出一种高效的实时幻觉检测方法:基于VLM的下一个词概率(NTPs)训练传统机器学习模型。NTPs可直接反映模型不确定性,我们假设低NTP值(高不确定性)与幻觉强相关。为此,构建了一个包含1400条人工标注语句的数据集,每条标注为幻觉或非幻觉。结果表明,基于NTP的特征能有效预测幻觉,使轻量级模型性能媲美强VLM。进一步地,引入仅输入生成文本回流得到的语言学NTPs,可提升检测效果;将VLM的幻觉预测分数融入NTP模型,性能优于单独使用任一方法。本研究为提升VLM可靠性提供了简单高效的解决方案。
原文摘要 · Abstract (English)
Hallucinations of vision-language models (VLMs), which are misalignments between visual content and generated text, undermine the reliability of VLMs. One common approach for detecting them employs the same VLM, or a different one, to assess generated outputs. This process is computationally intensive and increases model latency. In this paper, we explore an efficient on-the-fly method for hallucination detection by training traditional ML models over signals based on the VLM's next-token probabilities (NTPs). NTPs provide a direct quantification of model uncertainty. We hypothesize that high uncertainty (i.e., a low NTP value) is strongly associated with hallucinations. To test this, we introduce a dataset of 1,400 human-annotated statements derived from VLM-generated content, each labeled as hallucinated or not, and use it to test our NTP-based lightweight method. Our results demonstrate that NTP-based features are valuable predictors of hallucinations, enabling fast and simple ML models to achieve performance comparable to that of strong VLMs. Furthermore, augmenting these NTPs with linguistic NTPs, computed by feeding only the generated text back into the VLM, enhances hallucination detection performance. Finally, integrating hallucination prediction scores from VLMs into the NTP-based models led to better performance than using either VLMs or NTPs alone. We hope this study paves the way for simple, lightweight solutions that enhance the reliability of VLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。