arXiv:2411.04097cs.CVcs.AI2024-11NeurIPS被引 27

提出新方法发现并消除视觉语言模型中的虚假关联,提升泛化能力。

RaVL: Discovering and Mitigating Spurious Correlations in Fine-Tuned Vision-Language Models

  • 基于局部图像区域聚类定位导致错误的特征
  • 用新损失函数让模型关注真实相关区域,提升最差组准确率8.2%
  • 适用于各类模型与领域,尤其对医疗等敏感场景有帮助

微调后的视觉-语言模型常捕捉图像特征与文本属性间的虚假关联,导致测试时零样本性能下降。现有方法主要在全局图像层面操作,且多用于单模态场景。本文提出RaVL,从细粒度视角提升VLM鲁棒性:首先通过区域级聚类识别导致零样本分类错误的精确图像特征;再设计一种区域感知损失函数,在微调过程中引导模型关注相关区域、忽略虚假关联。我们在654个不同架构、数据域和虚假关联模式的VLM上评估该方法,结果表明RaVL在虚假关联发现上比最优基线提升191%,在最差组图像分类准确率上提升8.2%。通用领域与医学领域模型的定性验证进一步确认了效果。

原文摘要 · Abstract (English)

Fine-tuned vision-language models (VLMs) often capture spurious correlations between image features and textual attributes, resulting in degraded zero-shot performance at test time. Existing approaches for addressing spurious correlations (i) primarily operate at the global image-level rather than intervening directly on fine-grained image features and (ii) are predominantly designed for unimodal settings. In this work, we present RaVL, which takes a fine-grained perspective on VLM robustness by discovering and mitigating spurious correlations using local image features rather than operating at the global image level. Given a fine-tuned VLM, RaVL first discovers spurious correlations by leveraging a region-level clustering approach to identify precise image features contributing to zero-shot classification errors. Then, RaVL mitigates the identified spurious correlation with a novel region-aware loss function that enables the VLM to focus on relevant regions and ignore spurious relationships during fine-tuning. We evaluate RaVL on 654 VLMs with various model architectures, data domains, and learned spurious correlations. Our results show that RaVL accurately discovers (191% improvement over the closest baseline) and mitigates (8.2% improvement on worst-group image classification accuracy) spurious correlations. Qualitative evaluations on general-domain and medical-domain VLMs confirm our findings.

视觉语言模型虚假关联鲁棒性细粒度分析

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