References & Citations
Computer Science > Machine Learning
Title: Towards Scalable Bayesian Learning of Causal DAGs
(Submitted on 30 Sep 2020 (v1), last revised 18 Nov 2020 (this version, v2))
Abstract: We give methods for Bayesian inference of directed acyclic graphs, DAGs, and the induced causal effects from passively observed complete data. Our methods build on a recent Markov chain Monte Carlo scheme for learning Bayesian networks, which enables efficient approximate sampling from the graph posterior, provided that each node is assigned a small number $K$ of candidate parents. We present algorithmic techniques to significantly reduce the space and time requirements, which make the use of substantially larger values of $K$ feasible. Furthermore, we investigate the problem of selecting the candidate parents per node so as to maximize the covered posterior mass. Finally, we combine our sampling method with a novel Bayesian approach for estimating causal effects in linear Gaussian DAG models. Numerical experiments demonstrate the performance of our methods in detecting ancestor-descendant relations, and in causal effect estimation our Bayesian method is shown to outperform previous approaches.
Submission history
From: Jussi Viinikka [view email][v1] Wed, 30 Sep 2020 08:46:46 GMT (291kb,D)
[v2] Wed, 18 Nov 2020 13:48:01 GMT (360kb,D)
Link back to: arXiv, form interface, contact.