arXiv · 2402.02644
Permutation-based Inference for Variational Learning of Directed Acyclic Graphs
Abstract
Estimating the structure of Bayesian networks as directed acyclic graphs (DAGs) from observational data is a fundamental challenge, particularly in causal discovery. Bayesian approaches excel by quantifying uncertainty and addressing identifiability, but key obstacles remain: (i) representing distributions over DAGs and (ii) estimating a posterior in the underlying combinatorial space. We introduce PIVID, a method that jointly infers a distribution over permutations and DAGs using variational inference and continuous relaxations of discrete distributions. Through experiments on synthetic and real-world datasets, we show that PIVID can outperform deterministic and Bayesian approaches, achieving superior accuracy-uncertainty trade-offs while scaling efficiently with the number of variables.
Explore related subjects
Keep this discovery
Edwin V. Bonilla, Pantelis Elinas, He Zhao, Maurizio Filippone, Vassili Kitsios, Terry O'Kane. 2024-02-04. Permutation-based Inference for Variational Learning of Directed Acyclic Graphs. https://arxiv.org/abs/2402.02644
Cite the original work for its findings. Save a collection to share your selection of sources.