Source-linked AI summary
Surrogate modeling for Bayesian optimization beyond a single Gaussian process
Qin Lu, Konstantinos D. Polyzos, Bingcong Li, Georgios B. Giannakis
TL;DR
BO needs surrogate and acquisition choices for expensive black-box objectives, and common GP-based approaches preselect a kernel. This paper uses a weighted GP ensemble with TS and random-feature sampling; its analysis establishes sublinear Bayesian regret under stated assumptions, and experiments report advantages over alternatives.
Problem
Method
EGP-TS adaptively weights GP models, uses random-feature function sampling with TS to select queries, and accommodates parallel evaluations.
Source: §1.2 · p. 2Results
EGP-TS has sublinear cumulative Bayesian regret when the stated condition 0 ≤ c < 1 holds, and tests report merits relative to single-GP TS and alternative ensembles.
Source: §1.2 · p. 2, §4 · p. 6Takeaways & Limitations
The proposed ensemble approach combines adaptive GP surrogate selection with TS, parallel operation, and Bayesian-regret convergence results in the settings studied.
Source: §1.2 · p. 2, §4 · p. 6The regret guarantee relies on Assumption 1, which the paper relates to commonly used stationary kernels that are four times differentiable, including Gaussian kernels and Matérn kernels with ν > 2.
Source: §4 · p. 5
Abstract
from arXiv · showhide
Bayesian optimization (BO) has well-documented merits for optimizing black-box functions with an expensive evaluation cost. Such functions emerge in applications as diverse as hyperparameter tuning, drug discovery, and robotics. BO hinges on a Bayesian surrogate model to sequentially select query points so as to balance exploration with exploitation of the search space. Most existing works rely on a single Gaussian process (GP) based surrogate model, where the kernel function form is typically preselected using domain knowledge. To bypass such a design process, this paper leverages an ensemble (E) of GPs to adaptively select the surrogate model fit on-the-fly, yielding a GP mixture posterior with enhanced expressiveness for the sought function. Acquisition of the next evaluation input using this EGP-based function posterior is then enabled by Thompson sampling (TS) that requires no additional design parameters. To endow function sampling with scalability, random feature-based kernel approximation is leveraged per GP model. The novel EGP-TS readily accommodates parallel operation. To further establish convergence of the proposed EGP-TS to the global optimum, analysis is conducted based on the notion of Bayesian regret for both sequential and parallel settings. Tests on synthetic functions and real-world applications showcase the merits of the proposed method.
1 INTRODUCTION
BO can optimize expensive black-box functions under limited evaluation budgets, but designing its surrogate, acquisition function, and tuning parameters still requires expertise; parallel operation and convergence guarantees are also sought. The paper introduces an adaptive GP ensemble with Thompson sampling (TS), random-feature sampling, parallel operation, and Bayesian regret analysis.
1 INTRODUCTION
- The work targets BO that reduces design effort, supports parallel evaluations, and has convergence guarantees, addressing needs identified for time-consuming black-box optimization.The introduction also motivates these goals by the expertise required to choose surrogate models, kernels, and acquisition functions.Source: §1 · p. 1
1.1 Related works
- Prior work had explored ensembles of acquisition functions for a single GP, but not the complementary setting of an ensemble of GP surrogate models with a given acquisition function.The related-work discussion also identifies automatic kernel selection for GP-based BO as unexplored.Source: §1.1 · pp. 1–2
- TS offers BO acquisition without additional design parameters, but its mixture-prior counterpart with regret analysis had not been studied in BO.Parallel BO using other acquisition rules typically relies on additional hyperparameters or selection rules for query diversity.Source: §1.1 · pp. 1–2
1.2 Contributions
- EGP-TS adaptively weights an ensemble of GPs, uses random-feature approximation for scalable function sampling, and supports synchronous or asynchronous parallel evaluations.The method uses TS without additional design parameters.Source: §1.2 · p. 2
- The paper establishes sublinear Bayesian regret for EGP-TS in sequential and parallel settings and tests it on synthetic functions, three-model hyperparameter tuning, and robot pushing.The tests compare EGP-TS with single GP-based TS and alternative ensemble approaches.Source: §1.2 · p. 2
- Unlike the prior supervised-learning use of EGP, this BO adaptation actively selects query points and uses random features to scale TS function sampling with limited labeled data.The contribution discussion notes that kernel-hyperparameter updates require extra weight and model reinitialization.Source: §1.2 · p. 2
2 PRELIMINARIES
The preliminaries frame BO as maximizing an expensive, analytically unavailable objective by using evaluated data to guide subsequent queries. They outline GP-based surrogate modeling and TS as the acquisition strategy.
2 PRELIMINARIES
- BO seeks the maximizing feasible input of a black-box objective that is expensive to evaluate and lacks an available analytic expression.These properties make exhaustive enumeration and conventional gradient-based solvers unsuitable in the described setting.Source: §2 · p. 2
- BO iteratively fits a surrogate posterior from observed input-output pairs and maximizes an acquisition function designed to balance exploration and exploitation.The procedure selects the next query using the posterior over the objective.Source: §2 · p. 2
2.1 GP-based surrogate model and TS for acquisition
- A GP prior models the unknown function using a positive-definite kernel to encode pairwise input similarity, while observations are linked to function values through a noisy likelihood.The posterior is updated from the acquired data, and TS selects a query by sampling a function from that posterior and maximizing it.Source: §2.1 · pp. 2–3
- Random-feature approximation converts learning with stationary kernels into a parametric form that can support scalable sampling of functions from the GP posterior.The text presents it as one approach to function sampling for TS.Source: §2.1 · p. 3
3 ENSEMBLE GPS WITH TS FOR BO
The method combines GP models into a weighted mixture posterior and uses TS to choose evaluations. Random features support function sampling, while the method also provides synchronous and asynchronous parallel operation.
3 ENSEMBLE GPS WITH TS FOR BO
- EGP forms a non-Gaussian surrogate from an ensemble of GPs with distinct kernels, weighting each model’s contribution to the sought function.The posterior mixture weights are informed by the acquired data’s marginal likelihood under each GP.Source: §3 · p. 3
3.1 RF-based EGP-TS
- EGP-TS samples a GP model according to its mixture weight and draws a function path using random features, enabling differentiable and incrementally updated sampling.The resulting sampled function is maximized to select the next evaluation input.Source: §3 · p. 3, §3.1 · p. 3
- After each evaluation, Bayes’ rule updates the per-GP weights; kernel hyperparameters are periodically re-estimated using all acquired data, followed by weight reinitialization.Between such updates, the function posterior is propagated incrementally.Source: §3.1 · p. 4
3.2 Parallel EGP-TS
- EGP-TS supports synchronous batches by sampling once per worker and asynchronous operation by updating the posterior whenever a worker finishes.Both parallel settings use the same posterior-based query-selection procedure without extra design.Source: §3.2 · p. 4
- The EGP surrogate can also be paired with EI or UCB, but devising those rules directly from the GP-mixture posterior is left for future investigation.When kernels share a type, the EGP prior corresponds to a pseudo-Bayesian GP with hyperparameters chosen from a finite set.Source: §3.2 · pp. 4–5
4 BAYESIAN REGRET ANALYSIS
The section bounds EGP-TS’s cumulative Bayesian regret under a smoothness assumption. The bounds imply diminishing average regret in sequential and parallel settings.
4 BAYESIAN REGRET ANALYSIS
- Assumption 1 requires smooth GP sample paths on a compact, convex domain; it holds for several four-times-differentiable stationary kernels, including Gaussian and Matérn kernels with ν > 2.The section also identifies MIG as an information-theoretic measure of BO’s statistical difficulty and uses it in the regret bound.Source: §4 · p. 5
- The sequential proof splits regret into two terms, bounding one with single-GP results and the other through latent-model confidence sets and posterior standard deviations.The second term is further bounded using maximum information gain (MIG).Source: §4 · p. 5
- For K-worker asynchronous and synchronous evaluations, the paper gives cumulative regret bounds; in the asynchronous setting, the first term scales by √ρ_K relative to the sequential bound.The bounds concern evaluation counts, which typically exceed those in the sequential setup.Source: §4 · pp. 5–6
- Across sequential and parallel settings, cumulative Bayesian regret is O(MT^(c+1) log T), yielding diminishing average regret when 0 ≤ c < 1.The paper states that the bound is sublinear in the number of evaluations in this range.Source: §4 · p. 6
5 NUMERICAL TESTS
The tests compare EGP-TS with GP and ensemble baselines on synthetic functions, robot pushing, and model hyperparameter tuning, then examine a preliminary EGP-EI variant. Across these settings, EGP-TS generally performs well, while the EGP-EI comparison favors it in most tested functions and both robotic tasks.
5 NUMERICAL TESTS
- The experiments compare EGP-TS with GP-EI, GP-TS variants, fully Bayesian GP-TS, BanditBO, and EXP3BO, using 10 repetitions and reporting average performance and standard deviation.Unless stated otherwise, EGP-TS hyperparameters are refitted every 50 iterations, versus every iteration for other baselines; TS methods use 50 spectral features for fairness.Source: §5 · p. 6
5.1 Tests on synthetic functions
- On synthetic functions, EGP-TS improves on GP-TS counterparts and BanditBO/EXP3BO; with 11 RBF lengthscales, it outperforms all competitors.The tests include Ackley-5d, Zakharov, Drop-wave, and Eggholder; GP-EI is superior to GP-TS on Zakharov, but EGP-TS performs better than GP-EI there.Source: §5.1 · pp. 6–7
5.2 Robot pushing tasks
- On both robot-pushing tasks, EGP-TS outperforms the tested competitors with a four-kernel dictionary and also beats fully Bayesian GP-TS in simple regret while running faster.The tasks use 3 and 4 input variables; the four-kernel dictionary combines RBF kernels with and without ARD and Matérn kernels with ν = 3/2 and 5/2.Source: §5.2 · p. 7, Figure 3 · p. 7
5.3 Hyperparameter tuning tasks
- Across NN, SVM, and gradient-boosting hyperparameter tuning, EGP-TS performs better than competitors in most cases with either kernel-form or RBF-lengthscale dictionaries.The evaluation covers Breast cancer, Iris, Transportation, and Wine datasets; validation accuracy is measured on held-out data.Source: §5.3 · p. 7
5.4 Preliminary results for EGP-EI
- For EGP-EI, results favor it over GP-EI on three of four synthetic functions and both robot-pushing tasks.The preliminary comparison uses kernel hyperparameters updated each iteration and no random-feature approximation.Source: §5.4 · p. 7, Figure 10 · p. 9, Figure 11 · p. 9
6 CONCLUSIONS
The conclusion presents EGP-TS as an adaptive-kernel BO method with sequential and parallel convergence guarantees. Numerical tests show its merits relative to existing alternatives.
6 CONCLUSIONS
- EGP-TS combines an adaptive ensemble GP prior with random-feature sampling and Thompson sampling, and has sublinear cumulative Bayesian regret in sequential and parallel settings.The conclusion identifies other acquisition functions and frequentist-regret analysis as future work.Source: §6 · p. 9
7 PROOFS
The proofs establish Bayesian-regret bounds for EGP-TS across sequential and parallel settings and derive a random-feature approximation for its GP models.
7.1.1 Proof for Lemma 4
- The confidence set C_t misses the optimal model m* with probability at most 2MT^-1 at each time t.Source: §7.1 · p. 10
7.1 Proof of Theorem 1
- Theorem 1’s proof combines a GP sample-path supremum bound and confidence-set control to bound EGP-TS cumulative Bayesian regret.It uses η = 2√log T and β^(1/2)_T ≈ √(d log T).Source: §7.1 · pp. 9–11
7.1.2 Proof for Lemma 5
- The proof bounds the expected difference between the sampled model’s posterior mean and function value by 3B.Source: §7.1 · p. 9, §7.1.2 · p. 12
7.2 Proof of Theorem 2
- The asynchronous parallel proof establishes a cumulative Bayesian-regret bound for EGP-TS over T function evaluations.Source: §7.2 · p. 12
7.3 Proof of Theorem 3
- The synchronous parallel proof likewise derives the cumulative Bayesian-regret bound in Theorem 3.Source: §7.3 · pp. 12–13
APPENDIX A RANDOM FEATURE (RF) APPROXIMATION FOR GP
- Random Fourier features approximate shift-invariant kernels, allowing the nonparametric GP prior to be represented through a finite feature-based posterior.The approximation draws i.i.d. spectral samples and uses them to construct the feature map.Source: Appendix A · pp. 15–16
B.1 Ablation studies
The appendix presents ablation studies to further demonstrate the performance of EGP-TS.
- The appendix presents ablation studies to further assess EGP-TS performance.Source: §B.1 · p. 16
B.1.1 Function sampled from a GP prior
In a GP-prior test, EGP-TS with mixed kernels achieved regret comparable to a GP-TS model using the generating kernel, while its matching expert weight rose to one. Further ablations found diminishing gains from adding models and negligible gains from increasing spectral features from 50 to 1000.
B.1.1 Function sampled from a GP prior
- EGP-TS(mixed) had slightly inferior but comparable regret to GP-TS (Matern ν = 3/2), while the matching expert weight w3 went to 1 as data accumulated.The black-box function was drawn from a Matern ν = 3/2 GP prior with lengthscale 1, matching the GP-TS kernel.Source: §B.1.1 · p. 16
B.1.2 Effect of the number M of kernels
- On Robot pushing 3D, increasing M improved performance, but gains shrank from 3 to 7; the useful dictionary size depends on the problem and available domain knowledge.The authors suggest gains may saturate when the sought posterior has more modes than the ensemble has components.Source: §B.1.2 · p. 16
B.1.3 Effect of the number D of spectral features
- On Robot pushing 3D, increasing spectral features D from 50 to 1000 produced negligible performance gain.Source: §B.1.3 · p. 16
B.2 Results of parallel EGP-TS
Synchronous parallel EGP-TS trades slightly higher regret for shorter runtime as worker count increases, and it outperformed competing parallel TS-based methods on Ackley-5d.
B.2 Results of parallel EGP-TS
- Increasing K reduced runtime with slightly larger regret on Robot pushing 3D and 4D; EGP-TS also stood out against parallel TS-based baselines on Ackley-5d.Source: §B.2 · p. 16
B.3 Average runtimes on synthetic functions and robotic tasks
The runtime comparison covers competing methods on synthetic functions and robotic tasks; EGP-TS takes slightly longer than single GP-TS alternatives.
B.3 Average runtimes on synthetic functions and robotic tasks
- EGP-TS requires slightly increased runtime relative to the single GP-TS alternatives.The comparison concerns average runtimes on synthetic functions and robotic tasks.Source: §B.3 · p. 16
- The fully Bayesian GP method is omitted because its time-consuming MCMC step requires a much larger runtime.Source: §B.3 · p. 16
B.4 Additional information about synthetic functions
This section provides synthetic-function details, including optimizer values and a parallel-setting regret figure for Ackley-5d.
B.4 Additional information about synthetic functions
- For parallel Ackley-5d regret, the figure varies K using a dictionary of 11 RBF kernels with characteristic lengthscales {10c}6 c=−4.Source: Figure 15 · p. 17
- Table 3 provides candidate values, the corresponding maximizer x∗, and f(x∗) for the synthetic functions.Source: §B.4 · p. 17, Table 3 · p. 17
- The latter three synthetic functions are the opposites of their standard forms to make them maximization problems.Source: §B.4 · p. 17
Related papers
This paper cites
- BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding2018
- Structure Discovery in Nonparametric Regression through Compositional Kernel Search2013
- Bayesian Optimization is Superior to Random Search for Machine Learning Hyperparameter Tuning: Analysis of the Black-Box Optimization Challenge 20202021