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

Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation

About

Existing parameter-efficient fine-tuning (PEFT) methods have achieved significant success on vision transformers (ViTs) adaptation by improving parameter efficiency. However, the exploration of enhancing inference efficiency during adaptation remains underexplored. This limits the broader application of pre-trained ViT models, especially when the model is computationally extensive. In this paper, we propose Dynamic Tuning (DyT), a novel approach to improve both parameter and inference efficiency for ViT adaptation. Specifically, besides using the lightweight adapter modules, we propose a token dispatcher to distinguish informative tokens from less important ones, allowing the latter to dynamically skip the original block, thereby reducing the redundant computation during inference. Additionally, we explore multiple design variants to find the best practice of DyT. Finally, inspired by the mixture-of-experts (MoE) mechanism, we introduce an enhanced adapter to further boost the adaptation performance. We validate DyT across various tasks, including image/video recognition and semantic segmentation. For instance, DyT achieves superior performance compared to existing PEFT methods while evoking only 71% of their FLOPs on the VTAB-1K benchmark.

Wangbo Zhao, Jiasheng Tang, Yizeng Han, Yibing Song, Kai Wang, Gao Huang, Fan Wang, Yang You• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU47.67
2731
Object DetectionCOCO 2017 (val)
AP40.97
2454
Instance SegmentationCOCO 2017 (val)--
1144
Image ClassificationVTAB 1K
Overall Mean Accuracy78.5
204
Semantic segmentationCOCO Stuff (val)
mIoU45.71
126
Visual Task AdaptationVTAB 1k (test)
CIFAR-100 Accuracy74.4
15
Image ClassificationImage Datasets (CIFAR-100, SVHN, Food-101, Air, Pet, Car)
Accuracy (CIFAR-100)91.37
8
Video ClassificationVideo Datasets K400, SSv2
K400 Accuracy75.43
8
Showing 8 of 8 rows

Other info

Code

Follow for update