arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.03411stat.MLcs.LG

可扩展的未知软干预下的对比因果发现

Scalable Contrastive Causal Discovery under Unknown Soft Interventions

Department of Systems Biology, Columbia University Irving Institute for Cancer Dynamics, Columbia University(系统生物学系,哥伦比亚大学伊万尼癌症动力学研究所,哥伦比亚大学);Irving Institute for Cancer Dynamics, Columbia University(伊万尼癌症动力学研究所,哥伦比亚大学);Department of Computer Science, Columbia University(计算机科学系,哥伦比亚大学);Department of Biomedical Engineering, Columbia University(生物医学工程系,哥伦比亚大学);Data Science Institute, Columbia University(数据科学研究所,哥伦比亚大学)

Mingxuan Zhang, Khushi Desai, Sopho Kevlishvili, Elham Azizi

AI总结:

本文提出了一种可扩展的因果发现模型,通过对比方法处理未知软干预,提升因果结构恢复和泛化能力。

摘要:

观测性因果发现只能在马尔可夫等价类内被识别。虽然干预可以减少这种不确定性,但在实践中干预通常是软的,有多个未知的目标。在许多现实场景中,只观察到单一干预制度。我们提出了一种可扩展的因果发现模型,用于具有共享底层因果结构的配对观测性和干预性设置,以及未知的软干预。该模型聚合子集级别的PDAG并应用对比跨制度定向规则,以在Meek闭合下构建一致的全局最大PDAG,从而能够推广到分布内和分布外的设置。理论上,我们证明了该模型在由子集限制设置中的信息诱导出的受限Ψ等价类中是正确的。我们进一步表明,该模型在渐近上能够恢复相应的可识别PDAG,并且能够比非对比子集限制方法定向更多的边。在合成数据上的实验显示了改进的因果结构恢复、对未见图的推广能力以及对更大图的可扩展性,消融实验支持了理论结果。

英文摘要:

Observational causal discovery is only identifiable up to the Markov equivalence class. While interventions can reduce this ambiguity, in practice interventions are often soft with multiple unknown targets. In many realistic scenarios, only a single intervention regime is observed. We propose a scalable causal discovery model for paired observational and interventional settings with shared underlying causal structure and unknown soft interventions. The model aggregates subset-level PDAGs and applies contrastive cross-regime orientation rules to construct a globally consistent maximal PDAG under Meek closure, enabling generalization to both in-distribution and out-of-distribution settings. Theoretically, we prove that our model is sound with respect to a restricted $Ψ$ equivalence class induced solely by the information available in the subset-restricted setting. We further show that the model asymptotically recovers the corresponding identifiable PDAG and can orient additional edges compared to non-contrastive subset-restricted methods. Experiments on synthetic data demonstrate improved causal structure recovery, generalization to unseen graphs with held-out causal mechanisms, and scalability to larger graphs, with ablations supporting the theoretical results.

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

↑