Bayesian Optimization with Cost-varying Variable Subsets
Sebastian Tay, Chuan Sheng Foo, Daisuke Urano, Richalynn Leong, Bryan Kian Hsiang Low
Abstract
We introduce the problem of Bayesian optimization with cost-varying variable subsets (BOCVS) where in each iteration, the learner chooses a subset of query variables and specifies their values while the rest are randomly sampled. Each chosen subset has an associated cost. This presents the learner with the novel challenge of balancing between choosing more informative subsets for more directed learning versus leaving some variables to be randomly sampled to reduce incurred costs. This paper presents a novel Gaussian process upper confidence bound-based algorithm for solving the BOCVS problem that is provably no-regret. We analyze how the availability of cheaper control sets helps in exploration and reduces overall regret. We empirically show that our proposed algorithm can find significantly better solutions than comparable baselines with the same budget.
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 b1d96f06-176a-403b-80fd-e38e59eba40bCited by top-tier papers3
- Towards AutoAI: Optimizing a Machine Learning System with Black-box and Differentiable ComponentsZhiliang Chen, Chuan-Sheng Foo, Bryan Kian Hsiang LowICML 2024 · 10 citations
- DUET: Optimizing LLM Training Data Mixtures via Noisy Feedback from Unseen, Downstream Evaluation TasksZhiliang Chen, Gregory Kang Ruey Lau, Chuan Sheng Foo, Bryan Kian Hsiang LowICLR 2026 · 8 citations
- Bayesian Optimization for Unknown Cost-Varying Variable Subsets with No-Regret CostsVu Viet Hoang, Quoc Anh Hoang Nguyen, Hung Tran TheAAAI 2025
Builds on8
- BoTorch: A Framework for Efficient Monte-Carlo Bayesian OptimizationMaximilian Balandat, Brian Karrer, Daniel R. Jiang, Samuel Daulton et al.NeurIPS 2020 · 686 citations
- Multi-fidelity Bayesian Optimization with Max-value Entropy Search and its ParallelizationShion Takeno, Hitoshi Fukuoka, Yuhki Tsukada, Toshiyuki Koyama et al.ICML 2020 · 83 citations
- Bayesian Optimization of Risk MeasuresSait Cakmak, Raul Astudillo, Peter I. Frazier, Enlu ZhouNeurIPS 2020 · 65 citations
- Dynamic Causal Bayesian OptimizationVirginia Aglietti, Neil Dhir, Javier González, Theodoros DamoulasNeurIPS 2021 · 40 citations
- Value-at-Risk Optimization with Gaussian ProcessesQuoc Phong Nguyen, Zhongxiang Dai, Bryan Kian Hsiang Low, Patrick JailletICML 2021 · 34 citations
Related papers
- BILBO: BILevel Bayesian OptimizationWan Theng Ruth Chew, Quoc Phong Nguyen, Bryan Kian Hsiang LowICML 2025
- Monte Carlo Tree Search based Variable Selection for High Dimensional Bayesian OptimizationLei Song, Ke Xue, Xiaobin Huang, Chao QianNeurIPS 2022 · 57 citations
- On Regret Bounds of Thompson Sampling for Bayesian OptimizationShion Takeno, Shogo IwazakiICML 2026 · 3 citations
- Active Set OrderingQuoc Phong Nguyen, Sunil Gupta, Svetha Venkatesh, Bryan Kian Hsiang Low et al.NeurIPS 2024 · 1 citation
- Near-linear time Gaussian process optimization with adaptive batching and resparsificationDaniele Calandriello, Luigi Carratino, Alessandro Lazaric, Michal Valko et al.ICML 2020 · 23 citations
