Batch Bayesian Optimization For Replicable Experimental Design
Zhongxiang Dai, Quoc Phong Nguyen, Sebastian Tay, Daisuke Urano, Richalynn Leong, Bryan Kian Hsiang Low, Patrick Jaillet
Abstract
Many real-world experimental design problems (a) evaluate multiple experimental conditions in parallel and (b) replicate each condition multiple times due to large and heteroscedastic observation noise. Given a fixed total budget, this naturally induces a trade-off between evaluating more unique conditions while replicating each of them fewer times vs. evaluating fewer unique conditions and replicating each more times. Moreover, in these problems, practitioners may be risk-averse and hence prefer an input with both good average performance and small variability. To tackle both challenges, we propose the Batch Thompson Sampling for Replicable Experimental Design (BTS-RED) framework, which encompasses three algorithms. Our BTS-RED-Known and BTS-RED-Unknown algorithms, for, respectively, known and unknown noise variance, choose the number of replications adaptively rather than deterministically such that an input with a larger noise variance is replicated more times. As a result, despite the noise heteroscedasticity, both algorithms enjoy a theoretical guarantee and are asymptotically no-regret. Our Mean-Var-BTS-RED algorithm aims at risk-averse optimization and is also asymptotically no-regret. We also show the effectiveness of our algorithms in two practical real-world applications: precision agriculture and AutoML.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 74f835a1-74ec-4f3f-9796-94855bd2a28bCited by top-tier papers2
- Towards AutoAI: Optimizing a Machine Learning System with Black-box and Differentiable ComponentsZhiliang Chen, Chuan-Sheng Foo, Bryan Kian Hsiang LowICML 2024 · 10 citations
- Quantum Lipschitz BanditsBongsoo Yi, Yue Kang, Yao LiAAAI 2026 · 3 citations
Builds on15
- Biological Sequence Design with GFlowNetsMoksh Jain, Emmanuel Bengio, Alex Hernández-García, Jarrid Rector-Brooks et al.ICML 2022 · 224 citations
- Re-Examining Linear Embeddings for High-Dimensional Bayesian OptimizationBenjamin Letham, Roberto Calandra, Akshara Rai, Eytan BakshyNeurIPS 2020 · 152 citations
- Federated Bayesian Optimization via Thompson SamplingZhongxiang Dai, Bryan Kian Hsiang Low, Patrick JailletNeurIPS 2020 · 144 citations
- Bayesian Optimization of Risk MeasuresSait Cakmak, Raul Astudillo, Peter I. Frazier, Enlu ZhouNeurIPS 2020 · 65 citations
- Differentially Private Federated Bayesian Optimization with Distributed ExplorationZhongxiang Dai, Bryan Kian Hsiang Low, Patrick JailletNeurIPS 2021 · 64 citations
Related papers
- Risk-averse Heteroscedastic Bayesian OptimizationAnastasia Makarova, Ilnura Usmanova, Ilija Bogunovic, Andreas KrauseNeurIPS 2021 · 47 citations
- Bayesian Optimization for Categorical and Category-Specific Continuous InputsDang Nguyen, Sunil Gupta, Santu Rana, Alistair Shilton et al.AAAI 2020 · 59 citations
- Scalable Thompson Sampling using Sparse Gaussian Process ModelsSattar Vakili, Henry B. Moss, Artem Artemev, Vincent Dutordoir et al.NeurIPS 2021 · 52 citations
- Only Pay for What Is Uncertain: Variance-Adaptive Thompson SamplingAadirupa Saha, Branislav KvetonICLR 2024 · 3 citations
- Adaptive Variance Inflation in Thompson Sampling: Efficiency, Safety, Robustness, and BeyondFeng Zhu, David Simchi-LeviNeurIPS 2025 · 3 citations
