中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
A Closer Look at Self-Supervised Lightweight Vision Transformers

文献类型:会议论文

作者Wang, Shaoru2,4; Gao, Jin2,4; Li, Zeming3; Zhang, Xiaoqin1; Weiming, Hu2,4,5
出版日期2023-07
会议日期2023-7
会议地点Honolulu, Hawaii, USA
关键词Vision Transformer Self-supervised Learning Lightweight Networks Knowledge Distillation
英文摘要

Self-supervised learning on large-scale Vision Transformers (ViTs) as pre-training methods has achieved promising downstream performance. Yet, how much these pre-training paradigms promote lightweight ViTs' performance is considerably less studied. In this work, we develop and benchmark several self-supervised pre-training methods on image classification tasks and some downstream dense prediction tasks. We surprisingly find that if proper pre-training is adopted, even vanilla lightweight ViTs show comparable performance to previous SOTA networks with delicate architecture design. It breaks the recently popular conception that vanilla ViTs are not suitable for vision tasks in lightweight regimes. We also point out some defects of such pre-training, e.g., failing to benefit from large-scale pre-training data and showing inferior performance on data-insufficient downstream tasks. Furthermore, we analyze and clearly show the effect of such pre-training by analyzing the properties of the layer representation and attention maps for related models. Finally, based on the above analyses, a distillation strategy during pre-training is developed, which leads to further downstream performance improvement for MAE-based pre-training. Code is available at https://github.com/wangsr126/mae-lite.

语种英语
源URL[http://ir.ia.ac.cn/handle/173211/52415]  
专题自动化研究所_模式识别国家重点实验室_视频内容安全团队
通讯作者Gao, Jin
作者单位1.Wenzhou University
2.Institute of Automation, Chinese Academy of Sciences
3.Megvii Research
4.School of Artificial Intelligence, University of Chinese Academy of Sciences
5.ShanghaiTech University
推荐引用方式
GB/T 7714
Wang, Shaoru,Gao, Jin,Li, Zeming,et al. A Closer Look at Self-Supervised Lightweight Vision Transformers[C]. 见:. Honolulu, Hawaii, USA. 2023-7.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。