English

How to Estimate Model Transferability of Pre-Trained Speech Models?

Computation and Language 2024-02-07 v3 Neural and Evolutionary Computing Sound Audio and Speech Processing

Abstract

In this work, we introduce a "score-based assessment" framework for estimating the transferability of pre-trained speech models (PSMs) for fine-tuning target tasks. We leverage upon two representation theories, Bayesian likelihood estimation and optimal transport, to generate rank scores for the PSM candidates using the extracted representations. Our framework efficiently computes transferability scores without actual fine-tuning of candidate models or layers by making a temporal independent hypothesis. We evaluate some popular supervised speech models (e.g., Conformer RNN-Transducer) and self-supervised speech models (e.g., HuBERT) in cross-layer and cross-model settings using public data. Experimental results show a high Spearman's rank correlation and low pp-value between our estimation framework and fine-tuning ground truth. Our proposed transferability framework requires less computational time and resources, making it a resource-saving and time-efficient approach for tuning speech foundation models.

Keywords

Cite

@article{arxiv.2306.01015,
  title  = {How to Estimate Model Transferability of Pre-Trained Speech Models?},
  author = {Zih-Ching Chen and Chao-Han Huck Yang and Bo Li and Yu Zhang and Nanxin Chen and Shuo-Yiin Chang and Rohit Prabhavalkar and Hung-yi Lee and Tara N. Sainath},
  journal= {arXiv preprint arXiv:2306.01015},
  year   = {2024}
}

Comments

Accepted to Interspeech. Code is available at: https://github.com/virginiakm1988/LogME-CTC. Fixed a typo

R2 v1 2026-06-28T10:53:49.524Z