arXiv:2501.11485cs.CV2025-01被引 9

利用视觉语言模型中类别间语义一致性提升异常检测效果

SimLabel: Consistency-Guided OOD Detection with Pretrained Vision-Language Models

  • 基于同类语义关联构建更鲁棒的图像-类别相似度度量
  • 在多个零样本场景下超越现有方法,显著提升异常检测性能
  • 适用于各类视觉语言模型,适合安全关键领域应用

在真实世界机器学习应用中,尤其是安全敏感场景,检测分布外(OOD)数据至关重要。现有方法通常利用视觉语言模型(VLMs)中的语言信息,通过丰富的类别文本信息改进置信度估计。然而,现有工作在构建基于分布内(ID)图文相似性的OOD检测评分时,要么关注单个类别,要么关注整个类别集合,忽略了类别间的内在联系。我们发现,不同分布内类别之间的语义信息对有效进行OOD检测具有帮助。因此,我们研究了视觉语言模型中语义相关类别间的图文理解能力,并提出一种新颖的后处理策略SimLabel。该方法通过考虑一组相似类别标签的一致性,建立更稳健的图像-类别相似度度量,从而增强分布内与分布外样本的可分性。大量实验表明,SimLabel在多种零样本OOD检测基准上均表现优异。所提方法还可扩展至多种VLM主干网络,展现出良好的泛化能力。演示与实现代码已开源:https://github.com/ShuZou-1/SimLabel。

原文摘要 · Abstract (English)

Detecting out-of-distribution (OOD) data is crucial in real-world machine learning applications, particularly in safety-critical domains. Existing methods often leverage language information from vision-language models (VLMs) to enhance OOD detection by improving confidence estimation through rich class-wise text information. However, when building OOD detection score upon on in-distribution (ID) text-image affinity, existing works either focus on each ID class or whole ID label sets, overlooking inherent ID classes' connection. We find that the semantic information across different ID classes is beneficial for effective OOD detection. We thus investigate the ability of image-text comprehension among different semantic-related ID labels in VLMs and propose a novel post-hoc strategy called SimLabel. SimLabel enhances the separability between ID and OOD samples by establishing a more robust image-class similarity metric that considers consistency over a set of similar class labels. Extensive experiments demonstrate the superior performance of SimLabel on various zero-shot OOD detection benchmarks. The proposed model is also extended to various VLM-backbones, demonstrating its good generalization ability. Our demonstration and implementation codes are available at: https://github.com/ShuZou-1/SimLabel.

异常检测视觉语言模型零样本一致性

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