arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.04992cs.LGcs.AIstat.ME
收藏

不受无环性约束的可扩展变分因果发现

Scalable Variational Causal Discovery Unconstrained by Acyclicity

Applied Artificial Intelligence Institute (A2I2)(应用人工智能研究所(A2I2));Deakin University(迪肯大学)

Nu Hoang, Bao Duong, Thin Nguyen

AI总结:

本研究提出一种不受无环性约束的可扩展贝叶斯因果发现方法,通过新颖的可微DAG采样将隐式拓扑顺序的无约束分布映射到DAG分布,利用变分推理高效学习因果图后验,在模拟与真实数据上表现优越。

摘要:

贝叶斯因果发现能够量化广泛的结构多样的因果理论(以有向无环图(DAGs)形式表示)之间的认识不确定性,这些理论可能解释数据。然而,由于复杂的无环性约束,现有方法在高效DAG采样方面存在困难。本研究提出了一种可扩展的贝叶斯方法,得益于无需显式强制无环性即可生成DAG的能力,能够有效地在给定观测数据的情况下学习因果图的后验分布。具体而言,我们引入了一种新颖的可微DAG采样方法,通过将隐式拓扑顺序的无约束分布映射到DAG上的分布,生成有效的无环因果图。基于这种高效的DAG采样方案,我们能够使用连续域上的简单变分分布对因果图的后验分布进行建模,该分布可通过变分推理框架进行学习。在模拟和真实数据集上的大量实证实验表明,与几种最先进的基线方法相比,所提出的模型具有优越的性能。

英文摘要:

Bayesian causal discovery offers the power to quantify epistemic uncertainties among a broad range of structurally diverse causal theories potentially explaining the data, represented in forms of directed acyclic graphs (DAGs). However, existing methods struggle with efficient DAG sampling due to the complex acyclicity constraint. In this study, we propose a scalable Bayesian approach to effectively learn the posterior distribution over causal graphs given observational data thanks to the ability to generate DAGs without explicitly enforcing acyclicity. Specifically, we introduce a novel differentiable DAG sampling method that can generate a valid acyclic causal graph by mapping an unconstrained distribution of implicit topological orders to a distribution over DAGs. Given this efficient DAG sampling scheme, we are able to model the posterior distribution over causal graphs using a simple variational distribution over a continuous domain, which can be learned via the variational inference framework. Extensive empirical experiments on both simulated and real datasets demonstrate the superior performance of the proposed model compared to several state-of-the-art baselines.

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

↑