Abstract:Optimizing expensive-to-evaluate black-box functions of discrete (and potentially continuous) design parameters is a ubiquitous problem in scientific and engineering applications. Bayesian optimization (BO) is a popular, sample-efficient method that leverages a probabilistic surrogate model and an acquisition function (AF) to select promising designs to evaluate. However, maximizing the AF over mixed or high-cardinality discrete search spaces is challenging standard gradient-based methods cannot be used directly or evaluating the AF at every point in the search space would be computationally prohibitive. To address this issue, we propose using probabilistic reparameterization (PR). Instead of directly optimizing the AF over the search space containing discrete parameters, we instead maximize the expectation of the AF over a probability distribution defined by continuous parameters. We prove that under suitable reparameterizations, the BO policy that maximizes the probabilistic objective is the same as that which maximizes the AF, and therefore, PR enjoys the same regret bounds as the original BO policy using the underlying AF. Moreover, our approach provably converges to a stationary point of the probabilistic objective under gradient ascent using scalable, unbiased estimators of both the probabilistic objective and its gradient. Therefore, as the number of starting points and gradient steps increase, our approach will recover of a maximizer of the AF (an often-neglected requisite for commonly used BO regret bounds). We validate our approach empirically and demonstrate state-of-the-art optimization performance on a wide range of real-world applications. PR is complementary to (and benefits) recent work and naturally generalizes to settings with multiple objectives and black-box constraints.

Bayesian Optimization Based on Pseudo Labels

Bayesian Optimization Using Pseudo-Points

Poisson Process for Bayesian Optimization

PG-LBO: Enhancing High-Dimensional Bayesian Optimization with Pseudo-Label and Gaussian Process Guidance

Principled Bayesian Optimisation in Collaboration with Human Experts

Pseudo-Bayesian Optimization

RLBOF: Reinforcement Learning from Bayesian Optimization Feedback

Bayesian Optimization for Hyperparameters Tuning in Neural Networks

Provably Efficient Bayesian Optimization with Unbiased Gaussian Process Hyperparameter Estimation

Noisy Multiobjective Black-Box Optimization Using Bayesian Optimization.

Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic Reparameterization

Modulating Surrogates for Bayesian Optimization

In-Context Freeze-Thaw Bayesian Optimization for Hyperparameter Optimization

Robust Bayesian Optimization via Localized Online Conformal Prediction

Learning Optimal Data Augmentation Policies via Bayesian Optimization for Image Classification Tasks

Understanding and Improving Early Stopping for Learning with Noisy Labels

Bayesian Optimization Meets Self-Distillation

Bayesian Optimistic Optimization: Optimistic Exploration for Model-based Reinforcement Learning

PFNs4BO: In-Context Learning for Bayesian Optimization

On Provably Robust Meta-Bayesian Optimization

Practical Bayesian Optimization of Machine Learning Algorithms