arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2102.00084cs.CVcs.LG
收藏

一个线性化的框架和一个新的基准模型选择微调

A linearized framework and a new benchmark for model selection for fine-tuning

Amazon Web Services(亚马逊云科技)

Aditya Deshpande, Alessandro Achille, Avinash Ravichandran, Hao Li, Luca Zancato, Charless Fowlkes, Rahul Bhotika, Stefano Soatto, Pietro Perona

摘要:

从不同领域预先训练的模型集合(“模型动物园”)中进行微调正在成为一种提高低数据状态下测试准确性的技术。然而,模型选择,即如何从模型动物园中预先选择正确的模型进行微调,而无需执行任何训练,仍然是一个开放的话题。我们使用线性化的框架来近似微调,并引入两个新的基线模型选择-标签梯度和标签特征相关性。由于文献中的所有模型选择算法都已经在不同的用例上进行了测试,并且从未直接进行过比较,因此我们引入了一个新的模型选择综合基准,包括:i)单和多领域模型的模型动物园,以及ii)许多目标任务。我们的基准测试突出了与微调Imagenet模型相比,模型动物园的准确性增益。我们表明,我们的模型选择基线可以选择最佳的模型进行微调,在几个选择,并具有最高的排名相关性微调精度相比,现有的算法。

英文摘要:

Fine-tuning from a collection of models pre-trained on different domains (a "model zoo") is emerging as a technique to improve test accuracy in the low-data regime. However, model selection, i.e. how to pre-select the right model to fine-tune from a model zoo without performing any training, remains an open topic. We use a linearized framework to approximate fine-tuning, and introduce two new baselines for model selection -- Label-Gradient and Label-Feature Correlation. Since all model selection algorithms in the literature have been tested on different use-cases and never compared directly, we introduce a new comprehensive benchmark for model selection comprising of: i) A model zoo of single and multi-domain models, and ii) Many target tasks. Our benchmark highlights accuracy gain with model zoo compared to fine-tuning Imagenet models. We show our model selection baseline can select optimal models to fine-tune in few selections and has the highest ranking correlation to fine-tuning accuracy compared to existing algorithms.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑