NeurIPS 2022accept12 citations

Pre-Trained Model Reusability Evaluation for Small-Data Transfer Learning

Yao-Xiang Ding, Xi-Zhu Wu, Kun Zhou, Zhi-Hua Zhou

Abstract

We study {\it model reusability evaluation} (MRE) for source pre-trained models: evaluating their transfer learning performance to new target tasks. In special, we focus on the setting under which the target training datasets are small, making it difficult to produce reliable MRE scores using them. Under this situation, we propose {\it synergistic learning} for building the task-model metric, which can be realized by collecting a set of pre-trained models and asking a group of data providers to participate. We provide theoretical guarantees to show that the learned task-model metric distances can serve as trustworthy MRE scores, and propose synergistic learning algorithms and models for general learning tasks. Experiments show that the MRE models learned by synergistic learning can generate significantly more reliable MRE scores than existing approaches for small-data transfer learning.

transfer learningmetric learningmeta-learning
BibTeX
@inproceedings{
ding2022pretrained,
title={Pre-Trained Model Reusability Evaluation for Small-Data Transfer Learning},
author={Yao-Xiang Ding and Xi-Zhu Wu and Kun Zhou and Zhi-Hua Zhou},
booktitle={Advances in Neural Information Processing Systems},
editor={Alice H. Oh and Alekh Agarwal and Danielle Belgrave and Kyunghyun Cho},
year={2022},
url={https://openreview.net/forum?id=XY5g3mkVge}
}
Pre-Trained Model Reusability Evaluation for Small-Data Transfer Learning · NeurIPS 2022