Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation

Abstract

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.

Cite

Text

Zhao et al. "Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation." Neural Information Processing Systems, 2024. doi:10.52202/079017-3643

Markdown

[Zhao et al. "Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation." Neural Information Processing Systems, 2024.](https://mlanthology.org/neurips/2024/zhao2024neurips-dynamic/) doi:10.52202/079017-3643

BibTeX

@inproceedings{zhao2024neurips-dynamic,
  title     = {{Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation}},
  author    = {Zhao, Wangbo and Tang, Jiasheng and Han, Yizeng and Song, Yibing and Wang, Kai and Huang, Gao and Wang, Fan and You, Yang},
  booktitle = {Neural Information Processing Systems},
  year      = {2024},
  doi       = {10.52202/079017-3643},
  url       = {https://mlanthology.org/neurips/2024/zhao2024neurips-dynamic/}
}