Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognition

About

Pretraining Vision Transformers (ViTs) has achieved great success in visual recognition. A following scenario is to adapt a ViT to various image and video recognition tasks. The adaptation is challenging because of heavy computation and memory storage. Each model needs an independent and complete finetuning process to adapt to different tasks, which limits its transferability to different visual domains. To address this challenge, we propose an effective adaptation approach for Transformer, namely AdaptFormer, which can adapt the pre-trained ViTs into many different image and video tasks efficiently. It possesses several benefits more appealing than prior arts. Firstly, AdaptFormer introduces lightweight modules that only add less than 2% extra parameters to a ViT, while it is able to increase the ViT's transferability without updating its original pre-trained parameters, significantly outperforming the existing 100\% fully fine-tuned models on action recognition benchmarks. Secondly, it can be plug-and-play in different Transformers and scalable to many visual tasks. Thirdly, extensive experiments on five image and video datasets show that AdaptFormer largely improves ViTs in the target domains. For example, when updating just 1.5% extra parameters, it achieves about 10% and 19% relative improvement compared to the fully fine-tuned models on Something-Something~v2 and HMDB51, respectively. Code is available at https://github.com/ShoufaChen/AdaptFormer.

Shoufa Chen, Chongjian Ge, Zhan Tong, Jiangliu Wang, Yibing Song, Jue Wang, Ping Luo• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy91.86
3518
Semantic segmentationADE20K (val)
mIoU51.3
2731
Object DetectionCOCO 2017 (val)
AP38.71
2454
Instance SegmentationCOCO 2017 (val)--
1144
Image ClassificationImageNet 1k (test)
Top-1 Accuracy82.33
798
Image Super-resolutionManga109
PSNR25.19
656
Image Super-resolutionSet5 (test)
PSNR32.7
544
Image ClassificationFood-101--
494
Image ClassificationImageNet-R
Top-1 Acc69.5
474
Oriented Object DetectionDOTA v1.0 (test)--
378
Showing 10 of 144 rows
...

Other info

Code

Follow for update