让CLIP在少量样本下更懂语言,提升图像分类准确率
Language-Aware Information Maximization for Transductive Few-Shot CLIP
- 设计语言感知的信息最大化损失,融合视觉与文本信息
- 在多个数据集上超越最新方法,部分任务提升超10个百分点
- 首次系统探索参数高效微调在该场景的潜力,适合做视觉语言模型优化的研究者
横向少样本学习在仅依赖视觉模型时已有大量研究,但在当前主流的视觉-语言模型(VLMs)背景下仍处于起步阶段。仅有少数近期工作尝试解决此问题,表明横向推理在VLM中的潜力,并呼吁开发专用方法。基于此趋势,我们结合信息论思想与参数高效微调(PEFT)进展,提出一种高效的横向少样本CLIP方法。具体而言,引入一种新型语言感知信息最大化(LIMO)损失,包含三项互补项:(i) 视觉输入与文本类别描述之间的互信息;(ii) KL散度惩罚网络输出偏离文本驱动的零样本预测;(iii) 基于有标签样本的标准交叉熵损失。此外,我们挑战了该领域常见的微调范式,探索了此前被忽视的PEFT策略。令人意外的是,性能显著提升,揭示了在横向少样本设置中仅调整部分参数的巨大潜力。我们在多个数据集上进行了全面评估,结果表明LIMO大幅优于最新横向少样本CLIP方法,并在多项任务上超越表现最佳的归纳方法。代码已公开于:https://github.com/ghassenbaklouti/LIMO
原文摘要 · Abstract (English)
Transductive few-shot learning has triggered an abundant literature focusing on vision-only models, but is still at a nascent stage within the recent context of foundational vision-language models (VLMs). Only a few recent methods addressed the problem, pointing to the potential of tranduction in VLMs and to the need for VLM-tailored methods. Building on this momentum, we leverage information-theoretic concepts and recent progress in parameter-efficient fine-tuning (PEFT), developing a highly competitive transductive few-shot CLIP method. Specifically, we introduce a novel Language-aware Information MaximizatiOn (LIMO) loss integrating three complementary terms: (i) the mutual information between the vision inputs and the textual class descriptions; (ii) a Kullback-Leibler (KL) divergence penalizing deviation of the network's probabilistic outputs from the text-driven zero-shot predictions; and (iii) a standard cross-entropy loss based on the labeled shots. Furthermore, we challenge the commonly followed fine-tuning practices in the context of transductive few-shot learning, and explore PEFT strategies, completely overlooked in this context. Surprisingly, we observe substantial boosts in performances, which points to the potential of adapting a subset of the model's parameters in the transductive few-shot setting. We report comprehensive evaluations, which show that LIMO outperforms the very recent transductive few-shot CLIP methods by a large margin and yields significant gains over the best-performing inductive methods. Our code is publicly available at:\[ \href{https://github.com/ghassenbaklouti/LIMO}{\text{here}} \]
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。