English

$\chi$SPN: Characteristic Interventional Sum-Product Networks for Causal Inference in Hybrid Domains

Machine Learning 2024-08-15 v1 Artificial Intelligence

Abstract

Causal inference in hybrid domains, characterized by a mixture of discrete and continuous variables, presents a formidable challenge. We take a step towards this direction and propose Characteristic Interventional Sum-Product Network (χ\chiSPN) that is capable of estimating interventional distributions in presence of random variables drawn from mixed distributions. χ\chiSPN uses characteristic functions in the leaves of an interventional SPN (iSPN) thereby providing a unified view for discrete and continuous random variables through the Fourier-Stieltjes transform of the probability measures. A neural network is used to estimate the parameters of the learned iSPN using the intervened data. Our experiments on 3 synthetic heterogeneous datasets suggest that χ\chiSPN can effectively capture the interventional distributions for both discrete and continuous variables while being expressive and causally adequate. We also show that χ\chiSPN generalize to multiple interventions while being trained only on a single intervention data.

Keywords

Cite

@article{arxiv.2408.07545,
  title  = {$\chi$SPN: Characteristic Interventional Sum-Product Networks for Causal Inference in Hybrid Domains},
  author = {Harsh Poonia and Moritz Willig and Zhongjie Yu and Matej Zečević and Kristian Kersting and Devendra Singh Dhami},
  journal= {arXiv preprint arXiv:2408.07545},
  year   = {2024}
}

Comments

17 pages, 11 figures. Accepted as poster at UAI (Uncertainty in Artificial Intelligence) 2024