HEBO: An Empirical Study of Assumptions in Bayesian Optimisation
Abstract
In this work we rigorously analyse assumptions inherent to black-box optimisation hyper-parameter tuning tasks. Our results on the Bayesmark benchmark indicate that heteroscedasticity and non-stationarity pose significant challenges for black-box optimisers. Based on these findings, we propose a Heteroscedastic and Evolutionary Bayesian Optimisation solver (HEBO). HEBO performs non-linear input and output warping, admits exact marginal log-likelihood optimisation and is robust to the values of learned parameters. We demonstrate HEBO’s empirical efficacy on the NeurIPS 2020 Black-Box Optimisation challenge, where HEBO placed first. Upon further analysis, we observe that HEBO significantly outperforms existing black-box optimisers on 108 machine learning hyperparameter tuning tasks comprising the Bayesmark benchmark. Our findings indicate that the majority of hyper-parameter tuning tasks exhibit heteroscedasticity and non-stationarity, multiobjective acquisition ensembles with Pareto front solutions improve queried configurations, and robust acquisition maximisers afford empirical advantages relative to their non-robust counterparts. We hope these findings may serve as guiding principles for practitioners of Bayesian optimisation.
Cite
Text
Cowen-Rivers et al. "HEBO: An Empirical Study of Assumptions in Bayesian Optimisation." Journal of Artificial Intelligence Research, 2022. doi:10.1613/JAIR.1.13643Markdown
[Cowen-Rivers et al. "HEBO: An Empirical Study of Assumptions in Bayesian Optimisation." Journal of Artificial Intelligence Research, 2022.](https://mlanthology.org/jair/2022/cowenrivers2022jair-hebo/) doi:10.1613/JAIR.1.13643BibTeX
@article{cowenrivers2022jair-hebo,
title = {{HEBO: An Empirical Study of Assumptions in Bayesian Optimisation}},
author = {Cowen-Rivers, Alexander I. and Lyu, Wenlong and Tutunov, Rasul and Wang, Zhi and Grosnit, Antoine and Griffiths, Ryan-Rhys and Maraval, Alexandre Max and Hao, Jianye and Wang, Jun and Peters, Jan and Bou-Ammar, Haitham},
journal = {Journal of Artificial Intelligence Research},
year = {2022},
pages = {1269-1349},
doi = {10.1613/JAIR.1.13643},
volume = {74},
url = {https://mlanthology.org/jair/2022/cowenrivers2022jair-hebo/}
}