Timezone: »

Active Bayesian Causal Inference
Christian Toth · Lars Lorch · Christian Knoll · Andreas Krause · Franz Pernkopf · Robert Peharz · Julius von Kügelgen
Event URL: https://openreview.net/forum?id=ceQAk4W6Ho4 »

Causal discovery and causal reasoning are classically treated as separate and consecutive tasks: one first infers the causal graph, and then uses it to estimate causal effects of interventions. However, such a two-stage approach is uneconomical, especially in terms of actively collected interventional data, since the causal query of interest may not require a fully-specified causal model. From a Bayesian perspective, it is natural to treat a causal query (e.g., the causal graph or some causal effect) as subject to posterior inference while other unobserved quantities ought to be marginalized out.In this work, we propose Active Bayesian Causal Inference (ABCI), a fully-Bayesian active learning framework for integrated causal discovery and reasoning, which jointly infers a posterior over causal models and queries of interest.ABCI sequentially designs experiments that are maximally informative about the target causal query, collects the corresponding interventional data, and updates the Bayesian beliefs to choose the next experiment. Through simulations, we demonstrate that our approach is more data-efficient than several baselines that only focus on learning the full causal graph. This allows us to accurately learn downstream causal queries from fewer samples while providing well-calibrated uncertainty estimates for the quantities of interest.

Author Information

Christian Toth (Graz University of Technology)
Lars Lorch (ETH Zürich)
Christian Knoll (Graz University of Technology)
Andreas Krause (ETH Zurich)
Franz Pernkopf (Signal Processing and Speech Communication Laboratory, Graz, Austria)
Robert Peharz (Graz University of Technology)
Julius von Kügelgen (Max Planck Institute for Intelligent Systems Tübingen & University of Cambridge)

More from the Same Authors