Fine-Tuning for Few-Shot Image Classification by Multimodal Prototype Regularization

被引:2
作者
Wu, Qianhao [1 ]
Qi, Jiaxin [2 ]
Zhang, Dong [1 ]
Zhang, Hanwang
Tang, Jinhui [1 ]
机构
[1] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Peoples R China
[2] Nanyang Technol Univ, Sch Comp Sci & Engn, Singapore 639798, Singapore
基金
新加坡国家研究基金会;
关键词
Training; Visualization; Testing; Task analysis; Prototypes; Feature extraction; Tuning; Few-shot classification; large pre-trained vision-language models; model fine-tuning; prototype regularization; NETWORK; MODELS;
D O I
10.1109/TMM.2024.3379896
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Large pre-trained vision-language models, such as CLIP [Radford et al. 2021], have demonstrated remarkable performance in few-shot image classification. To facilitate the rapid adaptation of CLIP in downstream tasks with limited visual samples, two primary frameworks have been proposed. The first framework centers on the image encoder and introduces a trainable visual classifier after the backbone to generate logits for each object class. Nevertheless, this framework heavily depends on limited visual features extracted by the pre-trained visual encoder, which can result in over-fitting issues. The second framework aims to optimize the text encoder by using trainable soft language prompts and computing logits for each class based on the similarity between image features and optimized prompt features. However, this framework encounters the issue of imperfect alignment between the representations extracted by the image and text encoders, making it difficult to fine-tune the language prompts using visual samples. This paper proposes a Multi-Modal Prototype Regularization (MMPR) method for CLIP-based few-shot fine-tuning for image classification. MMPR can address the challenges of effectively utilizing both image and text features. MMPR fine-tunes a classifier and regularizes its weights using both image-based (ImgPR) and text-based (TexPR) prototypes. ImgPR represents the mean of image representations within the same class, derived from the image encoder, to distill specific visual distribution knowledge for classifier adaptation. TexPR represents the hand-crafted prompt associated with the class, derived from the text encoder, to incorporate general encyclopedic knowledge and mitigate visual over-fitting. MMPR significantly leverages both image and text information without increasing computational complexity during the inference stage compared to existing methods. Experimental results on various challenging public benchmarks demonstrate the superiority of the proposed MMPR method over state-of-the-art methods.
引用
收藏
页码:8543 / 8556
页数:14
相关论文
共 84 条
  • [31] Li JH, 2021, ADV NEUR IN, V34
  • [32] Li JN, 2022, PR MACH LEARN RES
  • [33] Multimodality Helps Unimodality: Cross-Modal Few-Shot Learning with Multimodal Models
    Lin, Zhiqiu
    Yu, Samuel
    Kuang, Zhiyi
    Pathak, Deepak
    Ramanan, Deva
    [J]. 2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 19325 - 19337
  • [34] Loshchilov I., 2016, P 5 INT C LEARN REPR, P1
  • [35] Loshchilov I., 2019, INT C LEARNING REPRE
  • [36] Lu Y., 2021, arXiv
  • [37] Maji S, 2013, Arxiv, DOI arXiv:1306.5151
  • [38] Mikolov T., 2013, P 26 INT C NEUR INF, P3111
  • [39] Mikolov T, 2013, Arxiv, DOI [arXiv:1301.3781, DOI 10.48550/ARXIV.1301.3781]
  • [40] Domain-Oriented Semantic Embedding for Zero-Shot Learning
    Min, Shaobo
    Yao, Hantao
    Xie, Hongtao
    Zha, Zheng-Jun
    Zhang, Yongdong
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2021, 23 : 3919 - 3930