Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AMU-Tuning: Effective Logit Bias for CLIP-based Few-shot Learning

About

Recently, pre-trained vision-language models (e.g., CLIP) have shown great potential in few-shot learning and attracted a lot of research interest. Although efforts have been made to improve few-shot ability of CLIP, key factors on the effectiveness of existing methods have not been well studied, limiting further exploration of CLIP's potential in few-shot learning. In this paper, we first introduce a unified formulation to analyze CLIP-based few-shot learning methods from a perspective of logit bias, which encourages us to learn an effective logit bias for further improving performance of CLIP-based few-shot learning methods. To this end, we disassemble three key components involved in computation of logit bias (i.e., logit features, logit predictor, and logit fusion) and empirically analyze the effect on performance of few-shot classification. Based on analysis of key components, this paper proposes a novel AMU-Tuning method to learn effective logit bias for CLIP-based few-shot classification. Specifically, our AMU-Tuning predicts logit bias by exploiting the appropriate $\underline{\textbf{A}}$uxiliary features, which are fed into an efficient feature-initialized linear classifier with $\underline{\textbf{M}}$ulti-branch training. Finally, an $\underline{\textbf{U}}$ncertainty-based fusion is developed to incorporate logit bias into CLIP for few-shot classification. The experiments are conducted on several widely used benchmarks, and the results show AMU-Tuning clearly outperforms its counterparts while achieving state-of-the-art performance of CLIP-based few-shot learning without bells and whistles.

Yuwei Tang, Zhenyi Lin, Qilong Wang, Pengfei Zhu, Qinghua Hu• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet A
Top-1 Acc52.05
654
Image ClassificationImageNet V2
Top-1 Acc65.42
611
Image ClassificationImageNet-R
Top-1 Acc78.09
529
Image ClassificationImageNet-Sketch
Top-1 Accuracy50.37
407
Image ClassificationImageNet-1K
Accuracy74.98
193
5-way 1-shot ClassificationCD-FSL ISIC, EuroSAT, CropDisease, ChestX (test)
Accuracy (ISIC)32.29
74
5-way 5-shot ClassificationCD-FSL ISIC, EuroSAT, CropDisease, ChestX (test)
Accuracy (ISIC)44.6
60
Image ClassificationImageNet-1K 8-shot (test)
Accuracy68.25
50
Few-shot Image ClassificationCD-FSL 5-way 1-shot (test)
ChestX Accuracy21.56
38
Few-shot Image ClassificationCD-FSL 5-way 5-shot (test)
ChestX Accuracy23.34
38
Showing 10 of 23 rows

Other info

Code

Follow for update