Bayesian Causal Discovery Networks for Linear Mixed Data

Moabi Mokhoro, Ioan Gabriel Bucur, Tom Heskes, Jildau Bouwman, Tom Claassen

Keywords: Bayesian Causal Discovery, Uncertainty, Variational Inference, Mixed data

TL;DR: Extend Bayesian Causal Discovery Networks (BCD Nets) to handle linear mixed data.

Abstract: Causal discovery from observational data is challenging due to limited sample sizes and noise, motivating probabilistic approaches that represent uncertainty over causal structures and parameters. Bayesian Causal Discovery Networks (BCD Nets) and related methods approximate posterior distributions over graphs, but existing approaches primarily focus on continuous data, with limited support for mixed discrete–continuous settings common in healthcare and economics. In this work, we extend BCD Nets to handle linear mixed data. Our approach incorporates an appropriate likelihood function for mixed data into the BCD Nets framework, enabling it to jointly model discrete and continuous variables. Experiments on synthetic and real-world datasets show that our method significantly outperforms a state-of-the-art causal discovery model for mixed data in both structural and causal effects accuracy.

Submission Number: 90