CVLGFeb 5, 2024

Time-, Memory- and Parameter-Efficient Visual Adaptation

arXiv:2402.02887v129 citationsh-index: 24CVPR
Originality Highly original
AI Analysis

This addresses the problem of high training-time and memory costs for adapting large foundation models in computer vision, offering a more scalable solution for downstream tasks.

The paper tackles the inefficiency of existing adaptation methods for foundation models by proposing a method that avoids backpropagation through the backbone, using a lightweight parallel network on frozen features. It achieves state-of-the-art accuracy-parameter trade-offs on the VTAB benchmark and scales to a 4 billion parameter vision transformer for video classification with improved training-time and memory efficiency.

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.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes