Fast hyperparameter tuning using Bayesian optimization with directional derivatives
Fast hyperparameter tuning using Bayesian optimization with directional derivatives
复制标题
DOI:
10.1016/j.knosys.2020.106247
复制
发表时间:
2020-10-12
影响因子:
8.8
通讯作者:
Venkatesh, Svetha
中科院分区:
文献类型:
--
作者:
Joy, Tinu Theckel;Rana, Santu;Venkatesh, Svetha
In this paper we develop a Bayesian optimization based hyperparameter tuning framework inspired by statistical learning theory for classifiers. We utilize two key facts from PAC learning theory; the generalization bound will be higher for a small subset of data compared to the whole, and the highest accuracy for a small subset of data can be achieved with a simple model. We initially tune the hyperparameters on a small subset of training data using Bayesian optimization. While tuning the hyperparameters on the whole training data, we leverage the insights from the learning theory to seek more complex models. We realize this by using directional derivative signs strategically placed in the hyperparameter search space to seek a more complex model than the one obtained with small data. We demonstrate the performance of our method on the tasks of tuning the hyperparameters of several machine learning algorithms. (C) 2020 Elsevier B.V. All rights reserved.