English

Top-Tuning: a study on transfer learning for an efficient alternative to fine tuning for image classification with fast kernel methods

Machine Learning 2023-11-10 v3

Abstract

The impressive performance of deep learning architectures is associated with a massive increase in model complexity. Millions of parameters need to be tuned, with training and inference time scaling accordingly, together with energy consumption. But is massive fine-tuning always necessary? In this paper, focusing on image classification, we consider a simple transfer learning approach exploiting pre-trained convolutional features as input for a fast-to-train kernel method. We refer to this approach as \textit{top-tuning} since only the kernel classifier is trained on the target dataset. In our study, we perform more than 3000 training processes focusing on 32 small to medium-sized target datasets, a typical situation where transfer learning is necessary. We show that the top-tuning approach provides comparable accuracy with respect to fine-tuning, with a training time between one and two orders of magnitude smaller. These results suggest that top-tuning is an effective alternative to fine-tuning in small/medium datasets, being especially useful when training time efficiency and computational resources saving are crucial.

Keywords

Cite

@article{arxiv.2209.07932,
  title  = {Top-Tuning: a study on transfer learning for an efficient alternative to fine tuning for image classification with fast kernel methods},
  author = {Paolo Didier Alfano and Vito Paolo Pastore and Lorenzo Rosasco and Francesca Odone},
  journal= {arXiv preprint arXiv:2209.07932},
  year   = {2023}
}
R2 v1 2026-06-28T01:27:11.774Z