Distributionally Robust Bayesian Optimization with φ-divergences
Hisham Husain, Vu Nguyen, Anton van den Hengel
Abstract
The study of robustness has received much attention due to its inevitability in data-driven settings where many systems face uncertainty. One such example of concern is Bayesian Optimization (BO), where uncertainty is multi-faceted, yet there only exists a limited number of works dedicated to this direction. In particular, there is the work of Kirschner et al. (2020), which bridges the existing literature of Distributionally Robust Optimization (DRO) by casting the BO problem from the lens of DRO. While this work is pioneering, it admittedly suffers from various practical shortcomings such as finite contexts assumptions, leaving behind the main question Can one devise a computationally tractable algorithm for solving this DRO-BO problem? In this work, we tackle this question to a large degree of generality by considering robustness against data-shift in -divergences, which subsumes many popular choices, such as the -divergence, Total Variation, and the extant Kullback-Leibler (KL) divergence. We show that the DRO-BO problem in this setting is equivalent to a finite-dimensional optimization problem which, even in the continuous context setting, can be easily implemented with provable sublinear regret bounds. We then show experimentally that our method surpasses existing methods, attesting to the theoretical results.
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 d1cfebaf-3e4a-4c8e-b0df-c73dcb27ad9aCited by top-tier papers7
- Contextual Gaussian Process Bandits with Neural NetworksHaoting Zhang, Jinghai He, Rhonda Righter, Zuo-Jun Max Shen et al.NeurIPS 2023 · 8 citations
- Offline-to-Online Hyperparameter Transfer for Stochastic BanditsDravyansh Sharma, Arun SuggalaAAAI 2025 · 8 citations
- Distributionally Robust Optimization via Generative Ambiguity ModelingJiaqi Wen, Jianyi YangICLR 2026 · 3 citations
- Stochastic Bayesian Optimization with Unknown Continuous Context Distribution via Kernel Density EstimationXiaobin Huang, Lei Song, Ke Xue, Chao QianAAAI 2024 · 3 citations
- 3D-Learning: Diffusion-Augmented Distributionally Robust Decision-Focused LearningJiaqi Wen, Lei Fan, Jianyi YangINFOCOM 2026 · 1 citation
Builds on9
- Bayesian Optimisation over Multiple Continuous and Categorical InputsBin Xin Ru, Ahsan S. Alvi, Vu Nguyen, Michael A. Osborne et al.ICML 2020 · 119 citations
- Provably Efficient Online Hyperparameter Optimization with Population-Based BanditsJack Parker-Holder, Vu Nguyen, Stephen J. RobertsNeurIPS 2020 · 105 citations
- Coping with Label Shift via Distributionally Robust OptimisationJingzhao Zhang, Aditya Krishna Menon, Andreas Veit, Srinadh Bhojanapalli et al.ICLR 2021 · 79 citations
- A Framework for robustness Certification of Smoothed Classifiers using F-DivergencesKrishnamurthy (Dj) Dvijotham, Jamie Hayes, Borja Balle, J. Zico Kolter et al.ICLR 2020 · 74 citations
- Knowing The What But Not The Where in Bayesian OptimizationVu Nguyen, Michael A. OsborneICML 2020 · 42 citations
Related papers
- Large-Scale Non-convex Stochastic Constrained Distributionally Robust OptimizationQi Zhang, Yi Zhou, Ashley Prater-Bennette, Lixin Shen et al.AAAI 2024 · 6 citations
- A Unified Framework for Bayesian Optimization under Contextual UncertaintySebastian Shenghong Tay, Chuan-Sheng Foo, Daisuke Urano, Richalynn Leong et al.ICLR 2024
- DORO: Distributional and Outlier Robust OptimizationRuntian Zhai, Chen Dan, J. Zico Kolter, Pradeep RavikumarICML 2021 · 74 citations
- Distributionally Robust Counterfactual Risk MinimizationLouis Faury, Ugo Tanielian, Elvis Dohmatob, Elena Smirnova et al.AAAI 2020 · 48 citations
- Efficient Distributionally Robust Assortment Optimization in MNL BanditsYunfan Zhang, Yuxuan Han, Zhengyuan ZhouICML 2026
