arXiv:2410.19294cs.CV2024-10NeurIPS被引 34

无需标注数据,通过分布学习与偏置修正提升零样本视觉模型性能

Enhancing Zero-Shot Vision Models by Label-Free Prompt Distribution Learning and Bias Correcting

  • 学习提示原型分布,自适应融合原CLIP模型
  • 在16个数据集上平均提升2.6%,ImageNet上提升1.5%
  • 训练免费且无需调参,适合零样本迁移任务

视觉语言模型如CLIP在使用恰当文本描述时展现出强大泛化能力。尽管在下游标注数据上优化提示能有效提升性能,但该方法依赖人工标注且受限于标注质量。此外,由于CLIP在高度不平衡的网络规模数据上预训练,存在固有标签偏置,导致性能不佳。为此,我们提出无需标注的提示分布学习与偏置校正框架Frolic,可显著提升零样本性能。Frolic通过学习提示原型分布,捕捉多样视觉表征,并通过置信度匹配自适应融合原始CLIP;再通过无标签逻辑调整进一步校正标签偏置。该方法无需训练也无需超参数调优。在16个数据集上的实验表明其有效性,尤其在10个数据集上比当前最优方法平均高出2.6%(使用CLIP ViT-B/16),在ImageNet及其五个分布偏移数据集上平均提升1.5%。

原文摘要 · Abstract (English)

Vision-language models, such as CLIP, have shown impressive generalization capacities when using appropriate text descriptions. While optimizing prompts on downstream labeled data has proven effective in improving performance, these methods entail labor costs for annotations and are limited by their quality. Additionally, since CLIP is pre-trained on highly imbalanced Web-scale data, it suffers from inherent label bias that leads to suboptimal performance. To tackle the above challenges, we propose a label-Free prompt distribution learning and bias correction framework, dubbed as **Frolic**, which boosts zero-shot performance without the need for labeled data. Specifically, our Frolic learns distributions over prompt prototypes to capture diverse visual representations and adaptively fuses these with the original CLIP through confidence matching. This fused model is further enhanced by correcting label bias via a label-free logit adjustment. Notably, our method is not only training-free but also circumvents the necessity for hyper-parameter tuning. Extensive experimental results across 16 datasets demonstrate the efficacy of our approach, particularly outperforming the state-of-the-art by an average of $2.6\%$ on 10 datasets with CLIP ViT-B/16 and achieving an average margin of $1.5\%$ on ImageNet and its five distribution shifts with CLIP ViT-B/16. Codes are available in https://github.com/zhuhsingyuu/Frolic.

零样本学习提示学习偏置校正CLIP

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