Bayesian Active Causal Discovery with Multi-Fidelity Experiments
Zeyu Zhang, Chaozhuo Li, Xu Chen, Xing Xie
Abstract
This paper studies the problem of active causal discovery when the experiments can be done based on multi-fidelity oracles, where higher fidelity experiments are more precise and expensive, while the lower ones are cheaper but less accurate. In this paper, we formally define the task of multi-fidelity active causal discovery, and design a probabilistic model for solving this problem. In specific, we first introduce a mutual-information based acquisition function to determine which variable should be intervened at which fidelity, and then a cascading model is proposed to capture the correlations between different fidelity oracles. Beyond the above basic framework, we also extend it to the batch intervention scenario. We find that the theoretical foundations behind the widely used and efficient greedy method do not hold in our problem. To solve this problem, we introduce a new concept called ϵ -submodular, and design a constraint based fidelity model to theoretically validate the greedy method. We conduct extensive experiments to demonstrate the effectiveness of our model.
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 papers2
- Amortized Active Causal Induction with Deep Reinforcement LearningYashas Annadani, Panagiotis Tigas, Stefan Bauer, Adam FosterNeurIPS 2024 · 13 citations
- Causal Discovery via Bayesian OptimizationBao Duong, Sunil Gupta, Thin NguyenICLR 2025
Builds on7
- DiBS: Differentiable Bayesian Structure LearningLars Lorch, Jonas Rothfuss, Bernhard Schölkopf, Andreas KrauseNeurIPS 2021 · 144 citations
- Multi-Fidelity Bayesian Optimization via Deep Neural NetworksShibo Li, Wei W. Xing, Robert M. Kirby, Shandian ZheNeurIPS 2020 · 74 citations
- Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic ReparameterizationSamuel Daulton, Xingchen Wan, David Eriksson, Maximilian Balandat et al.NeurIPS 2022 · 71 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
- Batch Multi-Fidelity Active Learning with Budget ConstraintsShibo Li, Jeff M. Phillips, Xin Yu, Robert M. Kirby et al.NeurIPS 2022 · 23 citations
Related papers
- Differentiable Multi-Target Causal Bayesian Experimental DesignPanagiotis Tigas, Yashas Annadani, Desi R. Ivanova, Andrew Jesson et al.ICML 2023 · 15 citations
- Optimal Multi-Fidelity Best-Arm IdentificationRiccardo Poiani, Rémy Degenne, Emilie Kaufmann, Alberto Maria Metelli et al.NeurIPS 2024 · 9 citations
- Nonmyopic Multifidelity Acitve SearchQuan Nguyen, Arghavan Modiri, Roman GarnettICML 2021 · 6 citations
- Near-Optimal Multi-Perturbation Experimental Design for Causal Structure LearningScott Sussex, Caroline Uhler, Andreas KrauseNeurIPS 2021 · 24 citations
- Multi-Fidelity Best-Arm IdentificationRiccardo Poiani, Alberto Maria Metelli, Marcello RestelliNeurIPS 2022 · 12 citations
