| |||||||
ShanghaiTech University Knowledge Management System
A Closer Look at Self-Supervised Lightweight Vision Transformers | |
2023-05-03 | |
状态 | 已发表 |
摘要 | 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 pro-mote lightweight ViTs’ performance is consider-ably 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 surpris-ingly find that if proper pre-training is adopted, even vanilla lightweight ViTs show compara-ble 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 dur -ing pre-training is developed, which leads to fur-ther downstream performance improvement for MAE-based pre-training. Code is available at |
DOI | arXiv:2205.14443 |
相关网址 | 查看原文 |
出处 | Arxiv |
WOS记录号 | PPRN:67041783 |
WOS类目 | Computer Science, Software Engineering |
资助项目 | National Key R&D Program of China[ |
文献类型 | 预印本 |
条目标识符 | https://kms.shanghaitech.edu.cn/handle/2MSLDSTB/348136 |
专题 | 信息科学与技术学院 |
作者单位 | 1.Chinese Acad Sci, Inst Automat, State Key Lab Multimodal Artificial Intelligence Syst, Beijing, Peoples R China 2.Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing, Peoples R China 3.Megvii Res, Beijing, Peoples R China 4.Wenzhou Univ, Key Lab Intelligent Informat Safety & Emergency Zhejiang Prov, Wenzhou, Peoples R China 5.ShanghaiTech Univ, Sch Informat Sci & Technol, Shanghai, Peoples R China |
推荐引用方式 GB/T 7714 | Wang, Shaoru,Gao, Jin,Li, Zeming,et al. A Closer Look at Self-Supervised Lightweight Vision Transformers. 2023. |
条目包含的文件 | ||||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 |
修改评论
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。