How far pre-trained models are from neural collapse on the target dataset informs their transferability

Z Wang, Y Luo, L Zheng, Z Huang… - Proceedings of the …, 2023 - openaccess.thecvf.com
This paper focuses on model transferability estimation, ie, assessing the performance of pre-
trained models on a downstream task without performing fine-tuning. Motivated by the …

Transferability estimation using bhattacharyya class separability

M Pándy, A Agostinelli, J Uijlings… - Proceedings of the …, 2022 - openaccess.thecvf.com
Transfer learning has become a popular method for leveraging pre-trained models in
computer vision. However, without performing computationally expensive fine-tuning, it is …

What to pre-train on? efficient intermediate task selection

C Poth, J Pfeiffer, A Rücklé, I Gurevych - arXiv preprint arXiv:2104.08247, 2021 - arxiv.org
Intermediate task fine-tuning has been shown to culminate in large transfer gains across
many NLP tasks. With an abundance of candidate datasets as well as pre-trained language …

Exploring model transferability through the lens of potential energy

X Li, Z Hu, Y Ge, Y Shan… - Proceedings of the IEEE …, 2023 - openaccess.thecvf.com
Transfer learning has become crucial in computer vision tasks due to the vast availability of
pre-trained deep learning models. However, selecting the optimal pre-trained model from a …

Frustratingly easy transferability estimation

LK Huang, J Huang, Y Rong… - … on machine learning, 2022 - proceedings.mlr.press
Transferability estimation has been an essential tool in selecting a pre-trained model and
the layers in it for transfer learning, to transfer, so as to maximize the performance on a target …

Etran: Energy-based transferability estimation

M Gholami, M Akbari, X Wang… - Proceedings of the …, 2023 - openaccess.thecvf.com
This paper addresses the problem of ranking pre-trained models for object detection and
image classification. Selecting the best pre-trained model by fine-tuning is an expensive and …

Model spider: Learning to rank pre-trained models efficiently

YK Zhang, TJ Huang, YX Ding… - Advances in Neural …, 2024 - proceedings.neurips.cc
Abstract Figuring out which Pre-Trained Model (PTM) from a model zoo fits the target task is
essential to take advantage of plentiful model resources. With the availability of numerous …

Not all models are equal: Predicting model transferability in a self-challenging fisher space

W Shao, X Zhao, Y Ge, Z Zhang, L Yang… - … on Computer Vision, 2022 - Springer
This paper addresses an important problem of ranking the pre-trained deep neural networks
and screening the most transferable ones for downstream tasks. It is challenging because …

Transferability metrics for selecting source model ensembles

A Agostinelli, J Uijlings, T Mensink… - Proceedings of the …, 2022 - openaccess.thecvf.com
We address the problem of ensemble selection in transfer learning: Given a large pool of
source models we want to select an ensemble of models which, after fine-tuning on the …

Pactran: Pac-bayesian metrics for estimating the transferability of pretrained models to classification tasks

N Ding, X Chen, T Levinboim, S Changpinyo… - … on Computer Vision, 2022 - Springer
With the increasing abundance of pretrained models in recent years, the problem of
selecting the best pretrained checkpoint for a particular downstream classification task has …