HEBO: An Empirical Study of Assumptions in Bayesian Optimisation

Alexander I. Cowen-Rivers(University of Cambridge), Wenlong Lyu(University of Cambridge), Rasul Tutunov, Zhi Wang(University of Cambridge), Antoine Grosnit(University of Cambridge), Ryan‐Rhys Griffiths(University of Cambridge), Alexandre Maraval, Jianye Hao(University of Cambridge), Jun Wang(University of Cambridge), Jan Peters, Haitham Bou Ammar(University of Cambridge)
Journal of Artificial Intelligence Research
July 11, 2022
Cited by 76Open Access
Full Text

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.


Related Papers

Adam: A Method for Stochastic Optimization
Diederik P. Kingma, Jimmy Ba|UvA-DARE (University of Amsterdam)|2014|84.7k
A fast and elitist multiobjective genetic algorithm: NSGA-II
Kalyanmoy Deb, Amrit Pratap, Sakshi Agarwal et al.|IEEE Transactions on Evolutionary Computation|2002|47.3k
An Analysis of Transformations
George E. P. Box, David R. Cox|Journal of the Royal Statistical Society Series B (Statistical Methodology)|1964|15k