无需标注伪相关特征,提升视觉语言模型的鲁棒性。
Debiased Prompt Tuning in Vision-Language Model without Annotations
- 利用零样本识别能力自动发现伪相关特征
- 在CelebA、Waterbirds等数据集上显著提升最差组准确率
- 适合关注模型公平性与泛化能力的研究者
视觉语言模型(如CLIP)的提示调优可快速适应下游任务,但易受虚假相关性影响,依赖背景、性别等无关特征,导致分布外数据下性能下降。现有消除虚假相关性的方法需人工标注伪相关属性,现实难行。本文利用视觉语言模型的零样本图像识别能力自动识别伪相关特征,构建伪属性标注,并据此自适应调整不同群体的训练权重。大量实验表明,该方法在CelebA、Waterbirds和MetaShift数据集上有效提升了最差组准确率,实现了最佳的组间鲁棒性差距,且无需人工标注。
原文摘要 · Abstract (English)
Prompt tuning of Vision-Language Models (VLMs) such as CLIP, has demonstrated the ability to rapidly adapt to various downstream tasks. However, recent studies indicate that tuned VLMs may suffer from the problem of spurious correlations, where the model relies on spurious features (e.g. background and gender) in the data. This may lead to the model having worse robustness in out-of-distribution data. Standard methods for eliminating spurious correlation typically require us to know the spurious attribute labels of each sample, which is hard in the real world. In this work, we explore improving the group robustness of prompt tuning in VLMs without relying on manual annotation of spurious features. We notice the zero - shot image recognition ability of VLMs and use this ability to identify spurious features, thus avoiding the cost of manual annotation. By leveraging pseudo-spurious attribute annotations, we further propose a method to automatically adjust the training weights of different groups. Extensive experiments show that our approach efficiently improves the worst-group accuracy on CelebA, Waterbirds, and MetaShift datasets, achieving the best robustness gap between the worst-group accuracy and the overall accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。