arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.03382cs.LGstat.ML
收藏

基于得分匹配的可扩展因果发现

Scalable Causal Discovery with Score Matching

Università di Genova(热那亚大学);Istituto Italiano di Tecnologia(意大利技术研究院);Carnegie Mellon University(卡内基梅隆大学);Amazon Web Services(亚马逊云科技)

Francesco Montagna, Nicoletta Noceti, Lorenzo Rosasco, Kun Zhang, Francesco Locatello

AI总结:

本文提出 DAS 算法,通过可扩展的机器学习方法近似得分函数并利用对数似然二阶导数,在非线性加性高斯噪声模型中高效发现完整因果图,大幅降低修剪复杂度,在保持准确性的同时显著提升速度。

摘要:

本文展示了如何在观测性非线性加性高斯噪声模型中,从对数似然的二阶导数发现整个因果图。利用可扩展的机器学习方法来近似得分函数 $\nabla \log p(\mathbf{X})$,我们扩展了 Rolland 等人(2022)的工作,该工作仅从得分中恢复拓扑排序,并需要一个昂贵的修剪步骤来移除该排序所允许的伪边。我们的分析得出了 DAS(Discovery At Scale 的首字母缩写),这是一种实用算法,它将修剪的复杂度降低了与图大小成比例的倍数。在实践中,DAS 实现了与当前最先进方法相当的准确性,同时速度快了一个数量级以上。总体而言,我们的方法实现了有原则且可扩展的因果发现,显著降低了计算门槛。

英文摘要:

This paper demonstrates how to discover the whole causal graph from the second derivative of the log-likelihood in observational non-linear additive Gaussian noise models. Leveraging scalable machine learning approaches to approximate the score function $\nabla \log p(\mathbf{X})$, we extend the work of Rolland et al. (2022) that only recovers the topological order from the score and requires an expensive pruning step removing spurious edges among those admitted by the ordering. Our analysis leads to DAS (acronym for Discovery At Scale), a practical algorithm that reduces the complexity of the pruning by a factor proportional to the graph size. In practice, DAS achieves competitive accuracy with current state-of-the-art while being over an order of magnitude faster. Overall, our approach enables principled and scalable causal discovery, significantly lowering the compute bar.

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

↑