Low-Rank Rescaled Vision Transformer Fine-Tuning: A Residual Design Approach

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Dong, Wei, Zhang, Xing, Chen, Bihui, Yan, Dawei, Lin, Zhijun, Yan, Qingsen, Wang, Peng, Yang, Yang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909154060271616
author Dong, Wei
Zhang, Xing
Chen, Bihui
Yan, Dawei
Lin, Zhijun
Yan, Qingsen
Wang, Peng
Yang, Yang
author_facet Dong, Wei
Zhang, Xing
Chen, Bihui
Yan, Dawei
Lin, Zhijun
Yan, Qingsen
Wang, Peng
Yang, Yang
contents Parameter-efficient fine-tuning for pre-trained Vision Transformers aims to adeptly tailor a model to downstream tasks by learning a minimal set of new adaptation parameters while preserving the frozen majority of pre-trained parameters. Striking a balance between retaining the generalizable representation capacity of the pre-trained model and acquiring task-specific features poses a key challenge. Currently, there is a lack of focus on guiding this delicate trade-off. In this study, we approach the problem from the perspective of Singular Value Decomposition (SVD) of pre-trained parameter matrices, providing insights into the tuning dynamics of existing methods. Building upon this understanding, we propose a Residual-based Low-Rank Rescaling (RLRR) fine-tuning strategy. This strategy not only enhances flexibility in parameter tuning but also ensures that new parameters do not deviate excessively from the pre-trained model through a residual design. Extensive experiments demonstrate that our method achieves competitive performance across various downstream image classification tasks, all while maintaining comparable new parameters. We believe this work takes a step forward in offering a unified perspective for interpreting existing methods and serves as motivation for the development of new approaches that move closer to effectively considering the crucial trade-off mentioned above. Our code is available at \href{https://github.com/zstarN70/RLRR.git}{https://github.com/zstarN70/RLRR.git}.
format Preprint
id arxiv_https___arxiv_org_abs_2403_19067
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Low-Rank Rescaled Vision Transformer Fine-Tuning: A Residual Design Approach
Dong, Wei
Zhang, Xing
Chen, Bihui
Yan, Dawei
Lin, Zhijun
Yan, Qingsen
Wang, Peng
Yang, Yang
Computer Vision and Pattern Recognition
Parameter-efficient fine-tuning for pre-trained Vision Transformers aims to adeptly tailor a model to downstream tasks by learning a minimal set of new adaptation parameters while preserving the frozen majority of pre-trained parameters. Striking a balance between retaining the generalizable representation capacity of the pre-trained model and acquiring task-specific features poses a key challenge. Currently, there is a lack of focus on guiding this delicate trade-off. In this study, we approach the problem from the perspective of Singular Value Decomposition (SVD) of pre-trained parameter matrices, providing insights into the tuning dynamics of existing methods. Building upon this understanding, we propose a Residual-based Low-Rank Rescaling (RLRR) fine-tuning strategy. This strategy not only enhances flexibility in parameter tuning but also ensures that new parameters do not deviate excessively from the pre-trained model through a residual design. Extensive experiments demonstrate that our method achieves competitive performance across various downstream image classification tasks, all while maintaining comparable new parameters. We believe this work takes a step forward in offering a unified perspective for interpreting existing methods and serves as motivation for the development of new approaches that move closer to effectively considering the crucial trade-off mentioned above. Our code is available at \href{https://github.com/zstarN70/RLRR.git}{https://github.com/zstarN70/RLRR.git}.
title Low-Rank Rescaled Vision Transformer Fine-Tuning: A Residual Design Approach
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.19067