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

Scaling & Shifting Your Features: A New Baseline for Efficient Model Tuning

About

Existing fine-tuning methods either tune all parameters of the pre-trained model (full fine-tuning), which is not efficient, or only tune the last linear layer (linear probing), which suffers a significant accuracy drop compared to the full fine-tuning. In this paper, we propose a new parameter-efficient fine-tuning method termed as SSF, representing that researchers only need to Scale and Shift the deep Features extracted by a pre-trained model to catch up with the performance of full fine-tuning. In this way, SSF also surprisingly outperforms other parameter-efficient fine-tuning approaches even with a smaller number of tunable parameters. Furthermore, different from some existing parameter-efficient fine-tuning methods (e.g., Adapter or VPT) that introduce the extra parameters and computational cost in the training and inference stages, SSF only adds learnable parameters during the training stage, and these additional parameters can be merged into the original pre-trained model weights via re-parameterization in the inference phase. With the proposed SSF, our model obtains 2.46% (90.72% vs. 88.54%) and 11.48% (73.10% vs. 65.57%) performance improvement on FGVC and VTAB-1k in terms of Top-1 accuracy compared to the full fine-tuning but only fine-tuning about 0.3M parameters. We also conduct amounts of experiments in various model families (CNNs, Transformers, and MLPs) and datasets. Results on 26 image classification datasets in total and 3 robustness & out-of-distribution datasets show the effectiveness of SSF. Code is available at https://github.com/dongzelian/SSF.

Dongze Lian, Daquan Zhou, Jiashi Feng, Xinchao Wang• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet-1K 1.0 (val)
Top-1 Accuracy83.1
1866
Image Super-resolutionManga109
PSNR24.98
656
Image Super-resolutionSet5 (test)
PSNR29.56
544
Image ClassificationStanford Cars
Accuracy89.2
477
Text-to-Image RetrievalFlickr30K
R@166.8
460
Text-to-Video RetrievalDiDeMo (test)
R@138.3
376
Fine-grained Image ClassificationStanford Cars (test)
Accuracy89.2
348
Text-to-Video RetrievalMSR-VTT
Recall@140.2
313
Image Super-resolutionSet14 (test)
PSNR26.84
292
Image ClassificationCUB-200 2011
Accuracy89.5
257
Showing 10 of 78 rows
...

Other info

Follow for update