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
1秒前
2秒前
2秒前
昏睡的实验室小白完成签到,获得积分10
3秒前
3秒前
yar应助大青山采纳,获得10
3秒前
Rondab应助mariawang采纳,获得10
3秒前
yqb发布了新的文献求助10
3秒前
3秒前
一条裸游的鱼完成签到,获得积分10
5秒前
二白关注了科研通微信公众号
5秒前
6秒前
6秒前
超帅听枫发布了新的文献求助10
6秒前
8秒前
8秒前
fish1116发布了新的文献求助100
8秒前
8秒前
bbbbb沫完成签到,获得积分20
8秒前
峥2发布了新的文献求助10
8秒前
wzy完成签到,获得积分10
8秒前
xxxllllll发布了新的文献求助10
9秒前
干净映天完成签到 ,获得积分10
9秒前
LJ_scholar发布了新的文献求助10
10秒前
kuny完成签到,获得积分10
10秒前
勤恳的夏之完成签到,获得积分20
11秒前
wzy发布了新的文献求助10
11秒前
钵钵鸡完成签到 ,获得积分10
11秒前
JING发布了新的文献求助10
11秒前
後zgw完成签到,获得积分10
13秒前
13秒前
14秒前
明朗发布了新的文献求助10
14秒前
meredith0571完成签到,获得积分10
15秒前
15秒前
嗯哼发布了新的文献求助30
15秒前
lucky完成签到 ,获得积分10
15秒前
朴实山彤完成签到,获得积分20
16秒前
16秒前
bkagyin应助ccalvintan采纳,获得10
18秒前
高分求助中
A new approach to the extrapolation of accelerated life test data 1000
ACSM’s Guidelines for Exercise Testing and Prescription, 12th edition 500
‘Unruly’ Children: Historical Fieldnotes and Learning Morality in a Taiwan Village (New Departures in Anthropology) 400
Indomethacinのヒトにおける経皮吸収 400
Phylogenetic study of the order Polydesmida (Myriapoda: Diplopoda) 370
基于可调谐半导体激光吸收光谱技术泄漏气体检测系统的研究 350
Robot-supported joining of reinforcement textiles with one-sided sewing heads 320
热门求助领域 (近24小时)
化学 材料科学 医学 生物 工程类 有机化学 生物化学 物理 内科学 纳米技术 计算机科学 化学工程 复合材料 遗传学 基因 物理化学 催化作用 冶金 细胞生物学 免疫学
热门帖子
关注 科研通微信公众号,转发送积分 3988732
求助须知:如何正确求助?哪些是违规求助? 3531027
关于积分的说明 11252281
捐赠科研通 3269732
什么是DOI,文献DOI怎么找? 1804764
邀请新用户注册赠送积分活动 881869
科研通“疑难数据库(出版商)”最低求助积分说明 809021