跳到主要导航 跳到搜索 跳到主要内容

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

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

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

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.

源语言英语
主期刊名International Conference on Electronic Information Engineering and Computer Science, EIECS 2022
编辑Yang Yue
出版商SPIE
ISBN(电子版)9781510663312
DOI
出版状态已出版 - 2023
已对外发布
活动2022 International Conference on Electronic Information Engineering and Computer Science, EIECS 2022 - Changchun, 中国
期限: 16 9月 202218 9月 2022

出版系列

姓名Proceedings of SPIE - The International Society for Optical Engineering
12602
ISSN(印刷版)0277-786X
ISSN(电子版)1996-756X

会议

会议2022 International Conference on Electronic Information Engineering and Computer Science, EIECS 2022
国家/地区中国
Changchun
时期16/09/2218/09/22

学术指纹

探究 'How to Make Use of Pretrained Models in Few-Shot Classification' 的科研主题。它们共同构成独一无二的学术指纹。

引用此