Constrained Causal Bayesian Optimization
Virginia Aglietti, Alan Malek, Ira Ktena, Silvia Chiappa
Abstract
We propose constrained causal Bayesian optimization (cCBO), an approach for finding interventions in a known causal graph that optimize a target variable under some constraints. cCBO first reduces the search space by exploiting the graph structure and, if available, an observational dataset; and then solves the restricted optimization problem by modelling target and constraint quantities using Gaussian processes and by sequentially selecting interventions via a constrained expected improvement acquisition function. We propose different surrogate models that enable to integrate observational and interventional data while capturing correlation among effects with increasing levels of sophistication. We evaluate cCBO on artificial and real-world causal graphs showing successful trade off between fast convergence and percentage of feasible interventions.
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 388b9e19-e592-422f-9f2b-fbedb8f4f8e3Cited by top-tier papers2
- Multi-Objective Causal Bayesian OptimizationShriya Bhatija, Paul-David Joshua Zuercher, Jakob Thumm, Thomas BohnéICML 2025
- Bayesian Optimization for Unknown Cost-Varying Variable Subsets with No-Regret CostsVu Viet Hoang, Quoc Anh Hoang Nguyen, Hung Tran TheAAAI 2025
Builds on5
- Efficient Model-Based Reinforcement Learning through Optimistic Policy Search and PlanningSebastian Curi, Felix Berkenkamp, Andreas KrauseNeurIPS 2020 · 120 citations
- Designing Optimal Dynamic Treatment Regimes: A Causal Reinforcement Learning ApproachJunzhe ZhangICML 2020 · 78 citations
- Interventions, Where and How? Experimental Design for Causal Models at ScalePanagiotis Tigas, Yashas Annadani, Andrew Jesson, Bernhard Schölkopf et al.NeurIPS 2022 · 68 citations
- Learning to search efficiently for causally near-optimal treatmentsSamuel Håkansson, Viktor Lindblom, Omer Gottesman, Fredrik D. JohanssonNeurIPS 2020 · 7 citations
- Model-based Causal Bayesian OptimizationScott Sussex, Anastasia Makarova, Andreas KrauseICLR 2023 · 1 citation
Related papers
- Dynamic Causal Bayesian OptimizationVirginia Aglietti, Neil Dhir, Javier González, Theodoros DamoulasNeurIPS 2021 · 40 citations
- Constrained Bayesian Optimization under Partial Observations: Balanced Improvements and Provable ConvergenceShengbo Wang, Ke LiAAAI 2024 · 22 citations
- Contextual Causal Bayesian OptimisationVahan Arsenyan, Antoine Grosnit, Haitham Bou-Ammar, Arnak S. DalalyanICLR 2026 · 3 citations
- Multi-task Causal Learning with Gaussian ProcessesVirginia Aglietti, Theodoros Damoulas, Mauricio A. Álvarez, Javier GonzálezNeurIPS 2020 · 23 citations
- Objective Bound Conditional Gaussian Process for Bayesian OptimizationTaewon Jeong, Heeyoung KimICML 2021 · 3 citations
