Skip to main navigation Skip to search Skip to main content

How to Make Use of Pretrained Models in Few-Shot Classification

  • Mingyu Fu
  • , Peng Wang
  • Northwestern Polytechnical University Xian

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Few-shot learning(FSL) aims to generalize model to novel categoeries by few labelled samples, which is challenging for machine. Large-scaled pretrained models, especially vision transformers achieve excellent performances benefiting from numerous and diverse data. Researchers have exploited pretrained models in few-shot classification by simply updating the whole parameters and finetuning on few samples. In this paper, we explore two methods: vision prompt tuning and a reparameterization method called 'scaling&&shift' to leverage pretrained models in few-shot classification. Vision prompt tuning is for vision transformer only and we first evaluate the method in few-shot setting. 'Scaling&&shift' is originally applied in convolution neural networks(CNN). We extend it to vision transformer. The two methods are evaluated on standard benchmarks such as miniImageNet, CUB, CIFAR-FS, clipart and sketch. The results show that 'scaling&&shift' reaches the same level compared to updating the whole parameters. Vision prompt tuning is 0%~5% lower than updating the whole parameters over five datasets while it has quite smaller amount of parameters updated.

Original languageEnglish
Title of host publicationInternational Conference on Electronic Information Engineering and Computer Science, EIECS 2022
EditorsYang Yue
PublisherSPIE
ISBN (Electronic)9781510663312
DOIs
StatePublished - 2023
Externally publishedYes
Event2022 International Conference on Electronic Information Engineering and Computer Science, EIECS 2022 - Changchun, China
Duration: 16 Sep 202218 Sep 2022

Publication series

NameProceedings of SPIE - The International Society for Optical Engineering
Volume12602
ISSN (Print)0277-786X
ISSN (Electronic)1996-756X

Conference

Conference2022 International Conference on Electronic Information Engineering and Computer Science, EIECS 2022
Country/TerritoryChina
CityChangchun
Period16/09/2218/09/22

Keywords

  • few-shot learning
  • pretrained model
  • reparameterization
  • vision prompt tuning
  • vision transformer

Fingerprint

Dive into the research topics of 'How to Make Use of Pretrained Models in Few-Shot Classification'. Together they form a unique fingerprint.

Cite this