Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

BOHB: Robust and Efficient Hyperparameter Optimization at Scale

About

Modern deep learning methods are very sensitive to many hyperparameters, and, due to the long training times of state-of-the-art models, vanilla Bayesian hyperparameter optimization is typically computationally infeasible. On the other hand, bandit-based configuration evaluation approaches based on random search lack guidance and do not converge to the best configurations as quickly. Here, we propose to combine the benefits of both Bayesian optimization and bandit-based methods, in order to achieve the best of both worlds: strong anytime performance and fast convergence to optimal configurations. We propose a new practical state-of-the-art hyperparameter optimization method, which consistently outperforms both Bayesian optimization and Hyperband on a wide range of problem types, including high-dimensional toy functions, support vector machines, feed-forward neural networks, Bayesian neural networks, deep reinforcement learning, and convolutional neural networks. Our method is robust and versatile, while at the same time being conceptually simple and easy to implement.

Stefan Falkner, Aaron Klein, Frank Hutter• 2018

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 NAS-Bench-201 (test)
Accuracy93.61
225
Image ClassificationCIFAR-100 NAS-Bench-201 (test)
Accuracy72.37
198
Image ClassificationCIFAR-10 NAS-Bench-201 (val)
Accuracy90.82
169
Image ClassificationImageNet-16-120 NAS-Bench-201 (test)
Accuracy45.26
167
Neural Architecture SearchNAS-Bench-201 ImageNet-16-120 (test)
Accuracy45.26
140
Image ClassificationCIFAR-100 NAS-Bench-201 (val)
Accuracy72.59
139
Image ClassificationImageNet 16-120 NAS-Bench-201 (val)
Accuracy45.44
123
Neural Architecture SearchCIFAR-10 NAS-Bench-201 (val)
Accuracy90.82
111
Neural Architecture SearchNAS-Bench-201 CIFAR-10 (test)
Accuracy93.61
110
Neural Architecture SearchImageNet16-120 NAS-Bench-201 (val)
Accuracy45.44
104
Showing 10 of 32 rows

Other info

Follow for update