STREGO: A Scalable Implementation of a Trust-Region Framework for Noisy Bayesian Optimization
Authors: Youssef Diouane, Issmail El Hallaoui, Meisseme Kadri, Amina Lamghari
Organizations: GERAD & Department of Mathematics and Industrial Engineering, Polytechnique Montr´eal. · GERAD & Department of Management, Universit´e du Qu´ebec `a Trois-Rivi`eres.
TREGO, a trust-region Bayesian optimization (BO) method for expensive blackbox problems, has convergence guarantees, but only for exact function values. In high dimension, its performance tends to deteriorate. This paper asks whether a trust-region BO method can be extended to handle noisy evaluations while being competitive in high dimension. We first generalize TREGO into a trust-region framework for noisy BO and extend its convergence analysis to noisy function estimates that are sufficiently accurate with sufficiently high probability. We then propose STREGO, a scalable implementation of this framework. Its global phase selects a batch of candidates by solving a bi-objective problem on the posterior mean of a Gaussian process and the integrated variance reduction. Unlike the predictive variance, this second objective does not favor boundary points merely because they are far from the data. Its local phase minimizes the posterior mean of a second Gaussian process, trained only on points close to the incumbent. STREGO is compared with TREGO, TURBO, BAXUS and NSMA-BO on synthetic and real-world problems of dimension 4 to 124, with and without noise. It gives the best average results in every synthetic setting tested and the best or tied-best solution on four of the six real-world problems. An implementation of STREGO is available at https://github.com/Mayousk/STREGO.
Figures & tables
Figure 1: Overview of TREGO, adapted from [ 16 ] (here 0<β<1<γ ).
Figure 2: An R2 illustration of the annular trust region Ωk (using the infinity norm) for different step sizes and incumbent positions xk∈Ω .
X={x∈X∥x−xk∥≤ρrmaxδk},
Algorithm 3 Step 3 of Algorithm 1 .
Figure 3: Comparison of STREGO with the global bi-objective criterion (μ,−IVR) against its variance counterpart (μ,−σ2) , both using the exploitative local phase, on Ackley, Alpine-01, Rotated Rastrigin and Schwefel in 4 d, 20 d, 50 d and 100 d under different noise settings.
Figure 4: Normalized final objective value (lower is better) and CPU time for d=20 , 50 and 100 in the noise-free setting. Tables report the mean ± standard error over 20 trials and 5 functions.
Figure 5: Normalized final objective value (lower is better) and CPU time across different noise settings for d=4 . Tables report the mean ± standard error over 20 trials and 4 functions.
Figure 6: Normalized final objective value (lower is better) and CPU time across different noise settings for d=50 . Tables report the mean ± standard error over 20 trials and 5 functions.
Figure 7: Convergence of STREGO, TREGO, TURBO, BAXUS and NSMA-BO on six real-world benchmarks of increasing dimensionality ( 16 d to 124 d).
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
Hyperparameter
Setting
Initial DoE size ( N0 )
10
Evaluation budget
200
Number of trials
20
Base seed
42
Initial trust-region radius ( δ0 )
0.5(1/5)1/d
Expansion factor ( γ )
2.0
Appendix
Table 1: STREGO hyperparameters used for a d -dimensional problem.
Figure 8: Comparison of STREGO with global batch sizes G∈{1,2,3,4} , all other settings held fixed, across dimensions and noise settings, averaged over 20 trials.
Trust Region Bayesian Optimization (TuRBO) is an effective strategy for alleviating the curse of dimensionality in high-dimensional black-box optimization. However, inappropriate lengthscale design can cause the local Gaussian process (GP) model within the trust region to degenerate, leading to suboptimal performance in high dimensions. In this work, we show that TuRBO's local GP may remain either excessively complex or overly simple as the dimension D and trust region side length L vary. To address this issue, we propose a straightforward variant, AdaScale-TuRBO, which scales the GP lengthscale with both the problem dimension and trust region size, thereby preserving kernel geometry and maintaining consistent prior complexity. Empirically, we show that AdaScale-TuRBO can robustly outperform standard TuRBO and other popular high-dimensional BO methods on synthetic benchmarks and real-world trajectory planning tasks.
We introduce LAGO, a LocAl-Global Optimization framework coupling Bayesian Optimization (BO) and gradient-based trust region local refinement through an adaptive competition mechanism for smooth expensive-to-evaluate objective functions with available gradients. At each iteration, global and local optimization strategies independently propose candidate points, and the next evaluation is selected based on predicted improvement. LAGO separates global exploration from local refinement at the proposal level: the BO acquisition function is optimized outside the active trust region, while local candidates are proposed within the trust region. Points in the vicinity of the accepted local step are incorporated in the global GP dataset only when satisfying a lengthscale-based minimum-distance criterion, hence reducing the risk of numerical instability during local exploitation. LAGO enhances BO with efficient local refinement when reaching promising regions, and reverts to exploratory behavior when local steps are not competitive.
Eliott Van Dieren, Tommaso Vanzan, Fabio Nobile
Institute of Mathematics, EPFL, Lausanne, Switzerland · Dipartimento di Scienze Matematiche, Politecnico di Torino, Turin, Italy
Bayesian optimization (BO) is a powerful paradigm for optimizing expensive black-box functions. Traditional BO methods typically rely on separate hand-crafted acquisition functions and surrogate models for the underlying function, and often operate in a myopic manner. In this paper, we propose a novel direct regret optimization approach that jointly learns the optimal model and non-myopic acquisition by distilling from a set of candidate models and acquisitions, and explicitly targets minimizing the multi-step regret. Our framework leverages an ensemble of Gaussian Processes (GPs) with varying hyperparameters to generate simulated BO trajectories, each guided by an acquisition function drawn from a pool of conventional choices and terminated by a Bayesian early stop criterion. These trajectories train an end-to-end Decision Transformer that selects the next query so as to improve the ultimate objective, following a dense training sparse learning paradigm: the transformer is trained on abundant simulated data, while a limited number of real evaluations refine the GPs online. On synthetic and real-world benchmarks, our method attains the best or near-best final simple regret against standard, lookahead, trust-region and amortized BO baselines, with the largest gains in high-dimensional settings. Ablations attribute the gains jointly to region-of-interest filtering and the learned policy, and matched-budget comparisons against explicit two-step lookahead acquisitions show that the advantage is not shared by lookahead alone.
Fengxue Zhang, Yuxin Chen
Department of Computer Science University of Chicago Chicago, IL 60637