Revisit Finetuning strategy for Few-Shot Learning to Transfer the Emdeddings
Heng Wang, Tan Yue, Xiang Ye, Zihang He, Bohan Li, Yong Li
Abstract
Few-Shot Learning (FSL) aims to learn a simple and effective bias on limited novel samples. Recently, many methods have been focused on re-training a randomly initialized linear classifier to adapt it to the novel features extracted by the pre-trained feature extractor (called Linear-Probing-based methods). These methods typically assumed the pre-trained feature extractor was robust enough, i.e., finetuning was not needed, and hence the pre-trained feature extractor does not change on the novel samples. However, the unchanged pre-trained feature extractor will distort the features of novel samples because the robustness assumption may not hold, especially on the out-of-distribution samples. To extract the undistorted features, we designed Linear-Probing-Finetuning with Firth-Bias (LP-FT-FB) to yield an accurate bias on the limited samples for better finetuning the pre-trained feature extractor, providing stronger transferring ability. In LP-FT-FB, we further proposed inverse Firth Bias Reduction (i-FBR) to regularize the over-parameterized feature extractor on which FBR does not work well.The proposed i-FBR effectively alleviates the over-fitting problem of the feature extractor in the process of finetuning and helps extract undistorted novel features. To show the effectiveness of the designed LP-FT-FB, we conducted a lot of experiments on the commonly used FSL datasets under different backbones, including in-domain and cross-domain FSL tasks. The experimental results show that the proposed FT-LP-FB outperforms the SOTA FSL methods. The code is available at https://github.com/whzyf951620/LinearProbingFinetuningFirthBias.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Cited by top-tier papers8
- Class-Aware Patch Embedding Adaptation for Few-Shot Image ClassificationFusheng Hao, Fengxiang He, Liu Liu, Fuxiang Wu et al.ICCV 2023 · 56 citations
- Quick-Tune: Quickly Learning Which Pretrained Model to Finetune and HowSebastian Pineda-Arango, Fabio Ferreira, Arlind Kadra, Frank Hutter et al.ICLR 2024 · 27 citations
- MetaCoCo: A New Few-Shot Classification Benchmark with Spurious CorrelationMin Zhang, Haoxuan Li, Fei Wu, Kun KuangICLR 2024 · 18 citations
- BECLR: Batch Enhanced Contrastive Few-Shot LearningStylianos Poulakakis-Daktylidis, Hadi Jamali RadICLR 2024 · 10 citations
- KNN Transformer with Pyramid Prompts for Few-Shot LearningWenhao Li, Qiangchang Wang, Peng Zhao, Yilong YinACM MM 2024 · 3 citations
Related papers
- Generally Boosting Few-Shot Learning with HandCrafted FeaturesYi Zhang, Sheng Huang, Fengtao ZhouACM MM 2021 · 6 citations
- Pushing the Limit of Fine-Tuning for Few-Shot Learning: Where Feature Reusing Meets Cross-Scale AttentionYing-Yu Chen, Jun-Wei Hsieh, Xin Li, Ming-Ching ChangAAAI 2024 · 3 citations
- On the Importance of Firth Bias Reduction in Few-Shot ClassificationSaba Ghaffari, Ehsan Saleh, David A. Forsyth, Yu-Xiong WangICLR 2022 · 13 citations
- Multi-directional Knowledge Transfer for Few-Shot LearningShuo Wang, Xinyu Zhang, Yanbin Hao, Chengbing Wang et al.ACM MM 2022 · 10 citations
- Feature-Proxy Transformer for Few-Shot SegmentationJian-Wei Zhang, Yifan Sun, Yi Yang, Wei ChenNeurIPS 2022 · 105 citations
