Scalable Global Optimization via Local Bayesian Optimization
About
Bayesian optimization has recently emerged as a popular method for the sample-efficient optimization of expensive black-box functions. However, the application to high-dimensional problems with several thousand observations remains challenging, and on difficult problems Bayesian optimization is often not competitive with other paradigms. In this paper we take the view that this is due to the implicit homogeneity of the global probabilistic models and an overemphasized exploration that results from global acquisition. This motivates the design of a local probabilistic approach for global optimization of large-scale high-dimensional problems. We propose the $\texttt{TuRBO}$ algorithm that fits a collection of local models and performs a principled global allocation of samples across these models via an implicit bandit approach. A comprehensive evaluation demonstrates that $\texttt{TuRBO}$ outperforms state-of-the-art methods from machine learning and operations research on problems spanning reinforcement learning, robotics, and the natural sciences.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Receptor Docking Affinity | TDC DRD3 (leaderboard) | Affinity Score-12.6 | 48 | |
| Calibration | Brock-Hommes (test) | MSE5.22e-5 | 40 | |
| Parameter Calibration | Brock–Hommes problems (various parameter sets) | Success Rate10 | 40 | |
| Parameter Estimation | Brock–Hommes problems (test) | Parameter Estimation Error (Mean)0.0032 | 40 | |
| High-dimensional optimization | MSLR | Convergence Value-8.9199 | 21 | |
| High-dimensional optimization | Lasso-Hard | Convergence Value11.503 | 20 | |
| High-dimensional optimization | LIMO | Convergence Value-4.2479 | 20 | |
| Function Optimization | Levy D=1000 | Convergence Value6.4294 | 19 | |
| Function Optimization | Rosenbrock D=1000 | Convergence Value8.00e+4 | 19 | |
| Function Optimization | Sphere D=1000 | Final Value29.7788 | 19 |