English

Back to the Basics on Predicting Transfer Performance

Machine Learning 2024-06-03 v1 Computer Vision and Pattern Recognition

Abstract

In the evolving landscape of deep learning, selecting the best pre-trained models from a growing number of choices is a challenge. Transferability scorers propose alleviating this scenario, but their recent proliferation, ironically, poses the challenge of their own assessment. In this work, we propose both robust benchmark guidelines for transferability scorers, and a well-founded technique to combine multiple scorers, which we show consistently improves their results. We extensively evaluate 13 scorers from literature across 11 datasets, comprising generalist, fine-grained, and medical imaging datasets. We show that few scorers match the predictive performance of the simple raw metric of models on ImageNet, and that all predictors suffer on medical datasets. Our results highlight the potential of combining different information sources for reliably predicting transferability across varied domains.

Keywords

Cite

@article{arxiv.2405.20420,
  title  = {Back to the Basics on Predicting Transfer Performance},
  author = {Levy Chaves and Eduardo Valle and Alceu Bissoto and Sandra Avila},
  journal= {arXiv preprint arXiv:2405.20420},
  year   = {2024}
}

Comments

15 pages, 3 figures, 2 tables

R2 v1 2026-06-28T16:47:46.420Z