Amortized Auto-Tuning: Cost-Efficient Bayesian Transfer Optimization for Hyperparameter Recommendation
Document Type
Article
Publication Title
arXiv
Abstract
With the surge in the number of hyperparameters and training times of modern machine learning models, hyperparameter tuning is becoming increasingly expensive. However, after assessing 40 tuning methods systematically, we find that each faces certain limitations. In particular, methods that speed up tuning via knowledge transfer typically require the final performance of hyperparameters and do not focus on low-fidelity information. As we demonstrate empirically, this common practice is suboptimal and can incur an unnecessary use of resources. It is more cost-efficient to instead leverage low-fidelity tuning observations to measure inter-task similarity and transfer knowledge from existing to new tasks accordingly. However, performing multi-fidelity tuning comes with its own challenges in the transfer setting: the noise in additional observations and the need for performance forecasting. Therefore, we propose and conduct a thorough analysis of a multi-task multi-fidelity Bayesian optimization framework, which leads to the best instantiation-AmorTized Auto-Tuning (AT2). We further present an offline-computed 27-task Hyperparameter Recommendation (HyperRec) database to serve the community. Extensive experiments on HyperRec and other real-world databases illustrate the effectiveness of our AT2 method. Copyright © 2021, The Authors. All rights reserved.
DOI
10.48550/arXiv.2106.09179
Publication Date
6-17-2021
Keywords
Automated machine learning, Autotuning, Bayesian optimization, Cost-efficient, Hyper-parameter, Hyperparameter transfer tuning, Low fidelities, Multi fidelities, Multi tasks, Multi-task multi-fidelity bayesian optimization
Recommended Citation
Y. Xiao, E.P. Xing, and W. Neiswanger, "Amortized Auto-Tuning: Cost-Efficient Bayesian Transfer Optimization for Hyperparameter Recommendation", 2021, arXiv:2106.09179
Comments
IR Deposit conditions: non-described
Preprint available on arXiv