Regional Expected Improvement for Efficient Trust Region Selection in High-Dimensional Bayesian Optimization
Nobuo Namura, Sho Takemori
TL;DR
This work addresses the challenge of scalable Bayesian optimization in medium to high dimensions by introducing regional expected improvement (REI) and its Monte Carlo variant qREI, integrated into the TuRBO trust-region framework. The authors develop a region-averaging operator that computes acquisition functions over trust regions, and prove that region averaging reduces the problem complexity via a regret bound U_N, with U_N(Sf) ≤ U_N(f). Empirically, REI/qREI-guided trust-region selection improves optimization performance across several real-world high-dimensional tasks, especially when multiple restarts are possible, and demonstrates robustness to unknown problem characteristics. The proposed approach offers a practical, theoretically grounded enhancement for global optimization in high-dimensional, expensive-black-box settings, with publicly available code for replication.
Abstract
Real-world optimization problems often involve complex objective functions with costly evaluations. While Bayesian optimization (BO) with Gaussian processes is effective for these challenges, it suffers in high-dimensional spaces due to performance degradation from limited function evaluations. To overcome this, simplification techniques like dimensionality reduction have been employed, yet they often rely on assumptions about the problem characteristics, potentially underperforming when these assumptions do not hold. Trust-region-based methods, which avoid such assumptions, focus on local search but risk stagnation in local optima. In this study, we propose a novel acquisition function, regional expected improvement (REI), designed to enhance trust-region-based BO in medium to high-dimensional settings. REI identifies regions likely to contain the global optimum, improving performance without relying on specific problem characteristics. We provide a theoretical proof that REI effectively identifies optimal trust regions and empirically demonstrate that incorporating REI into trust-region-based BO outperforms conventional BO and other high-dimensional BO methods in medium to high-dimensional real-world problems.
