ALoRE aggregates multiple low-rank experts in a Kronecker-product space and merges them into the frozen backbone, reporting top accuracy on FGVC and VTAB-1k with only 0.15M trainable parameters.
Time-, Memory- and Parameter-Efficient Visual Adaptation
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
As foundation models become more popular, there is a growing need to efficiently finetune them for downstream tasks. Although numerous adaptation methods have been proposed, they are designed to be efficient only in terms of how many parameters are trained. They, however, typically still require backpropagating gradients throughout the model, meaning that their training-time and -memory cost does not reduce as significantly. We propose an adaptation method which does not backpropagate gradients through the backbone. We achieve this by designing a lightweight network in parallel that operates on features from the frozen, pretrained backbone. As a result, our method is efficient not only in terms of parameters, but also in training-time and memory usage. Our approach achieves state-of-the-art accuracy-parameter trade-offs on the popular VTAB benchmark, and we further show how we outperform prior works with respect to training-time and -memory usage too. We further demonstrate the training efficiency and scalability of our method by adapting a vision transformer backbone of 4 billion parameters for the computationally demanding task of video classification, without any intricate model parallelism. Here, we outperform a prior adaptor-based method which could only scale to a 1 billion parameter backbone, or fully-finetuning a smaller backbone, with the same GPU and less training time.
citation-role summary
citation-polarity summary
fields
cs.CV 1years
2024 1verdicts
CONDITIONAL 1roles
background 1polarities
unclear 1representative citing papers
citing papers explorer
-
ALoRE: Efficient Visual Adaptation via Aggregating Low Rank Experts
ALoRE aggregates multiple low-rank experts in a Kronecker-product space and merges them into the frozen backbone, reporting top accuracy on FGVC and VTAB-1k with only 0.15M trainable parameters.