Parallel Bayesian Optimization of Multiple Noisy Objectives with Expected Hypervolume Improvement
About
Optimizing multiple competing black-box objectives is a challenging problem in many fields, including science, engineering, and machine learning. Multi-objective Bayesian optimization (MOBO) is a sample-efficient approach for identifying the optimal trade-offs between the objectives. However, many existing methods perform poorly when the observations are corrupted by noise. We propose a novel acquisition function, NEHVI, that overcomes this important practical limitation by applying a Bayesian treatment to the popular expected hypervolume improvement (EHVI) criterion and integrating over this uncertainty in the Pareto frontier. We argue that, even in the noiseless setting, generating multiple candidates in parallel is an incarnation of EHVI with uncertainty in the Pareto frontier and therefore can be addressed using the same underlying technique. Through this lens, we derive a natural parallel variant, $q$NEHVI, that reduces computational complexity of parallel EHVI from exponential to polynomial with respect to the batch size. $q$NEHVI is one-step Bayes-optimal for hypervolume maximization in both noisy and noiseless environments, and we show that it can be optimized effectively with gradient-based methods via sample average approximation. Empirically, we demonstrate not only that $q$NEHVI is substantially more robust to observation noise than existing MOBO approaches, but also that it achieves state-of-the-art optimization performance and competitive wall-times in large-batch environments.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Acquisition function optimization | DTLZ1 5 objectives m = 5 | Mean Wall Time (s)2.14 | 16 | |
| Acquisition function optimization | DTLZ1 3 objectives m = 3 | Mean Wall Time (s)3.49 | 16 | |
| Acquisition function optimization | DTLZ1 10 objectives m = 10 | Mean Wall Time (s)5.97 | 14 | |
| Multi-Objective Optimization | DTLZ2 noisy 5-objective | Log Distance-1.3 | 12 | |
| Multi-Objective Optimization | Inverted DTLZ2 noisy 5-objective | Log Distance0.15 | 6 | |
| Multi-Objective Optimization | DTLZ2 5 objectives noisy | Mean HV4.3 | 6 | |
| Multi-Objective Optimization | Inverted DTLZ1 noisy 5-objective | Log Distance3.1 | 6 | |
| Multi-Objective Optimization | DTLZ1 5 objectives noisy | Mean HV7.20e+13 | 6 | |
| Multi-Objective Optimization | Noisy Convex DTLZ2 5 objectives | Mean Hypervolume (HV)11 | 6 | |
| Multi-Objective Optimization | Noisy Scaled DTLZ2 5 objectives | HV (Mean)7.7 | 6 |