arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.07545cs.LGcs.AI
收藏

$χ$SPN:面向混合域因果推断的特征干预和积网络

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

Indian Institute of Technology Bombay(印度孟买理工学院);Technical University of Darmstadt(达姆施塔特工业大学);Hessian Center for Artificial Intelligence (hessian.AI)(黑森人工智能中心);German Research Center for Artificial Intelligence (DFKI)(德国人工智能研究中心);Eindhoven University of Technology(埃因霍温理工大学)

Harsh Poonia, Moritz Willig, Zhongjie Yu, Matej Zečević, Kristian Kersting, Devendra Singh Dhami

AI总结:

针对混合域因果推断难题,提出融合特征函数与神经网络的$χ$SPN模型,可统一处理离散连续变量,在3个合成数据集上表现优异且能跨干预场景泛化。

摘要:

由离散与连续变量混合构成的混合域因果推断是一项极具挑战性的难题。本文朝着该方向推进,提出了特征干预和积网络($χ$SPN),该模型能够在随机变量服从混合分布的场景下估计干预分布。$χ$SPN在干预和积网络(iSPN)的叶节点中使用特征函数,通过概率测度的Fourier-Stieltjes变换,为离散与连续随机变量提供了统一的表示视角。模型采用神经网络,基于干预数据来估计所学习的iSPN的参数。我们在3个合成异构数据集上开展的实验表明,$χ$SPN既能有效捕捉离散与连续变量的干预分布,同时具备良好的表达能力与因果充分性。研究还显示,$χ$SPN仅在单种干预数据上训练,就能泛化至多种干预场景。

英文摘要:

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 ($χ$SPN) that is capable of estimating interventional distributions in presence of random variables drawn from mixed distributions. $χ$SPN 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 $χ$SPN can effectively capture the interventional distributions for both discrete and continuous variables while being expressive and causally adequate. We also show that $χ$SPN generalize to multiple interventions while being trained only on a single intervention data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑