Vision Transformer Slimming: Multi-Dimension Searching in Continuous Optimization Space

失败 变压器 计算机科学 过程(计算) 人工智能 计算机工程 算法 并行计算 工程类 电气工程 程序设计语言 电压
作者
Chavan, Arnav,Zhi-Qiang Shen,Zhuang Liu,Zechun Liu,Kwang-Ting Cheng,Eric P. Xing
出处
期刊:Cornell University - arXiv
标识
DOI:10.48550/arxiv.2201.00814
摘要

This paper explores the feasibility of finding an optimal sub-model from a vision transformer and introduces a pure vision transformer slimming (ViT-Slim) framework. It can search a sub-structure from the original model end-to-end across multiple dimensions, including the input tokens, MHSA and MLP modules with state-of-the-art performance. Our method is based on a learnable and unified $\ell_1$ sparsity constraint with pre-defined factors to reflect the global importance in the continuous searching space of different dimensions. The searching process is highly efficient through a single-shot training scheme. For instance, on DeiT-S, ViT-Slim only takes ~43 GPU hours for the searching process, and the searched structure is flexible with diverse dimensionalities in different modules. Then, a budget threshold is employed according to the requirements of accuracy-FLOPs trade-off on running devices, and a re-training process is performed to obtain the final model. The extensive experiments show that our ViT-Slim can compress up to 40% of parameters and 40% FLOPs on various vision transformers while increasing the accuracy by ~0.6% on ImageNet. We also demonstrate the advantage of our searched models on several downstream datasets. Our code is available at https://github.com/Arnav0400/ViT-Slim.
最长约 10秒,即可获得该文献文件

科研通智能强力驱动
Strongly Powered by AbleSci AI
更新
PDF的下载单位、IP信息已删除 (2025-6-4)

科研通是完全免费的文献互助平台,具备全网最快的应助速度,最高的求助完成率。 对每一个文献求助,科研通都将尽心尽力,给求助人一个满意的交代。
实时播报
杰尼龟的鱼完成签到 ,获得积分10
3秒前
安然完成签到 ,获得积分10
4秒前
张希伦完成签到 ,获得积分10
4秒前
任性翠安完成签到 ,获得积分10
8秒前
dong完成签到 ,获得积分10
10秒前
神说完成签到,获得积分0
12秒前
量子星尘发布了新的文献求助10
18秒前
Aimee完成签到 ,获得积分10
18秒前
徐彬荣完成签到,获得积分10
19秒前
研友_8yN60L完成签到,获得积分10
20秒前
搜集达人应助科研通管家采纳,获得10
20秒前
光亮的自行车完成签到 ,获得积分10
20秒前
李东东完成签到 ,获得积分10
35秒前
王多肉完成签到,获得积分10
39秒前
Iiiilr完成签到 ,获得积分10
40秒前
杨幂完成签到,获得积分10
41秒前
48秒前
hellokitty完成签到,获得积分10
51秒前
51秒前
小四发布了新的文献求助10
52秒前
58秒前
西瓜完成签到 ,获得积分10
58秒前
包容的忆灵完成签到 ,获得积分10
1分钟前
高兴尔冬发布了新的文献求助10
1分钟前
xiang完成签到 ,获得积分0
1分钟前
小四完成签到,获得积分10
1分钟前
FashionBoy应助slayers采纳,获得30
1分钟前
量子星尘发布了新的文献求助10
1分钟前
黑眼圈完成签到 ,获得积分10
1分钟前
jia完成签到 ,获得积分10
1分钟前
如履平川完成签到 ,获得积分10
1分钟前
科目三应助忧伤的步美采纳,获得10
1分钟前
大椒完成签到 ,获得积分10
1分钟前
1分钟前
1分钟前
wisdom完成签到,获得积分10
1分钟前
slayers发布了新的文献求助30
1分钟前
1分钟前
e746700020完成签到,获得积分10
1分钟前
高兴尔冬完成签到,获得积分10
1分钟前
高分求助中
【提示信息,请勿应助】关于scihub 10000
Les Mantodea de Guyane: Insecta, Polyneoptera [The Mantids of French Guiana] 3000
徐淮辽南地区新元古代叠层石及生物地层 3000
The Mother of All Tableaux: Order, Equivalence, and Geometry in the Large-scale Structure of Optimality Theory 3000
Handbook of Industrial Diamonds.Vol2 1100
Global Eyelash Assessment scale (GEA) 1000
Picture Books with Same-sex Parented Families: Unintentional Censorship 550
热门求助领域 (近24小时)
化学 材料科学 医学 生物 工程类 有机化学 生物化学 物理 内科学 纳米技术 计算机科学 化学工程 复合材料 遗传学 基因 物理化学 催化作用 冶金 细胞生物学 免疫学
热门帖子
关注 科研通微信公众号,转发送积分 4038039
求助须知:如何正确求助?哪些是违规求助? 3575756
关于积分的说明 11373782
捐赠科研通 3305574
什么是DOI,文献DOI怎么找? 1819239
邀请新用户注册赠送积分活动 892655
科研通“疑难数据库(出版商)”最低求助积分说明 815022