Learning Regions of Interest for Bayesian Optimization with Adaptive Level-Set Estimation
Fengxue Zhang, Jialin Song, James C. Bowden, Alexander Ladd, Yisong Yue, Thomas Desautels, Yuxin Chen
Abstract
We study Bayesian optimization (BO) in high-dimensional and non-stationary scenarios. Existing algorithms for such scenarios typically require extensive hyperparameter tuning, which limits their practical effectiveness. We propose a framework, called BALLET, which adaptively filters for a high-confidence region of interest (ROI) as a superlevel-set of a nonparametric probabilistic model such as a Gaussian process (GP). Our approach is easy to tune, and is able to focus on local region of the optimization space that can be tackled by existing BO methods. The key idea is to use two probabilistic models: a coarse GP to identify the ROI, and a localized GP for optimization within the ROI. We show theoretically that BALLET can efficiently shrink the search space, and can exhibit a tighter regret bound than standard BO without ROI filtering. We demonstrate empirically the effectiveness of BALLET on both synthetic and real-world optimization tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers5
- Transition Constrained Bayesian Optimization via Markov Decision ProcessesJose Pablo Folch, Calvin Tsay, Robert M. Lee, Behrang Shafei et al.NeurIPS 2024 · 10 citations
- Optimizing the Unknown: Black Box Bayesian Optimization with Energy-Based Model and Reinforcement LearningRuiyao Miao, Junren Xiao, Shiya Tsang, Hui Xiong et al.NeurIPS 2025 · 2 citations
- CoffeeBoost: Gradient Boosting Native Conformal Inference for Bayesian OptimizationYuanhao Lai, Pengfei Zheng, Chenpeng Ji, Cheng Qiu et al.AAAI 2025 · 1 citation
- Optimistic Games for Combinatorial Bayesian Optimization with Application to Protein DesignMelis Ilayda Bal, Pier Giuseppe Sessa, Mojmir Mutny, Andreas KrauseICLR 2025 · 1 citation
- Variational Search DistributionsDaniel M. Steinberg, Rafael Oliveira, Cheng Soon Ong, Edwin V. BonillaICLR 2025
Builds on8
- Sample-Efficient Optimization in the Latent Space of Deep Generative Models via Weighted RetrainingAustin Tripp, Erik A. Daxberger, José Miguel Hernández-LobatoNeurIPS 2020 · 186 citations
- Learning Search Space Partition for Black-box Optimization using Monte Carlo Tree SearchLinnan Wang, Rodrigo Fonseca, Yuandong TianNeurIPS 2020 · 163 citations
- Re-Examining Linear Embeddings for High-Dimensional Bayesian OptimizationBenjamin Letham, Roberto Calandra, Akshara Rai, Eytan BakshyNeurIPS 2020 · 152 citations
- Few-Shot Bayesian Optimization with Deep Kernel SurrogatesMartin Wistuba, Josif GrabockaICLR 2021 · 87 citations
- Increasing the Scope as You Learn: Adaptive Bayesian Optimization in Nested SubspacesLeonard Papenmeier, Luigi Nardi, Matthias PoloczekNeurIPS 2022 · 76 citations
Related papers
- Scalable Bayesian Optimization via Focalized Sparse Gaussian ProcessesYunyue Wei, Vincent Zhuang, Saraswati Soedarmadji, Yanan SuiNeurIPS 2024 · 10 citations
- Bayesian Optimistic Optimisation with Exponentially Decaying RegretHung Tran-The, Sunil Gupta, Santu Rana, Svetha VenkateshICML 2021 · 4 citations
- Policy Search via Bayesian Optimization with Temporal Difference Gaussian ProcessesArmin Lederer, Anuj Srivastava, Marco Bagatella, Andreas KrauseICML 2026
- Sub-linear Regret Bounds for Bayesian Optimisation in Unknown Search SpacesHung Tran-The, Sunil Gupta, Santu Rana, Huong Ha et al.NeurIPS 2020 · 8 citations
- "Why Not Looking backward?" A Robust Two-Step Method to Automatically Terminate Bayesian OptimizationShuang Li, Ke Li, Wei LiNeurIPS 2023 · 6 citations
