arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.09194stat.MEcs.LGstat.CO
收藏

广义线性因果网络的分布式学习

Distributed Learning of Generalized Linear Causal Networks

University of California, Los Angeles(加州大学洛杉矶分校)

Qiaoling Ye, Arash A. Amini, Qing Zhou

AI总结:

本文提出一种新型分布式因果结构学习方法DARLS,通过模拟退火搜索拓扑排序并结合分布式优化求解广义线性模型参数化的有向无环图,在理论上证明了其全局收敛性,实验表明其在分布式和汇总数据下均具高精度与强预测力。

摘要:

我们考虑从存储在多台机器上的数据中学习因果结构的任务,并提出一种名为基于正则化似然分数的分布式退火(DARLS)的新型结构学习方法来解决这一问题。我们使用以广义线性模型参数化的有向无环图对因果结构进行建模,从而使我们的方法适用于各种类型的数据。为了获得高分的因果图,DARLS模拟退火过程以在拓扑排序的空间上进行搜索,其中通过与排序兼容的最优图结构由分布式优化方法找到。这种分布式优化依赖于本地和中央机器之间的多轮通信来估计最优结构。我们建立了其向全局优化器的收敛性,该优化器对应于跨本地机器的所有数据上计算的总分数。据我们所知,DARLS是首个具有此类理论保证的分布式因果图学习方法。通过广泛的模拟研究,DARLS在分布式数据上表现出与现有方法相媲美的性能,并在结构学习精度和测试数据似然方面,与应用于所有本地机器汇总数据的竞争方法取得了相当的结果。在利用分布式ChIP-Sequencing数据对蛋白质-DNA结合网络进行建模的实际应用中,DARLS也展现出比其他方法更高的预测能力,证明了其在从分布式数据估计因果网络方面的巨大优势。

英文摘要:

We consider the task of learning causal structures from data stored on multiple machines, and propose a novel structure learning method called distributed annealing on regularized likelihood score (DARLS) to solve this problem. We model causal structures by a directed acyclic graph that is parameterized with generalized linear models, so that our method is applicable to various types of data. To obtain a high-scoring causal graph, DARLS simulates an annealing process to search over the space of topological sorts, where the optimal graphical structure compatible with a sort is found by a distributed optimization method. This distributed optimization relies on multiple rounds of communication between local and central machines to estimate the optimal structure. We establish its convergence to a global optimizer of the overall score that is computed on all data across local machines. To the best of our knowledge, DARLS is the first distributed method for learning causal graphs with such theoretical guarantees. Through extensive simulation studies, DARLS has shown competing performance against existing methods on distributed data, and achieved comparable structure learning accuracy and test-data likelihood with competing methods applied to pooled data across all local machines. In a real-world application for modeling protein-DNA binding networks with distributed ChIP-Sequencing data, DARLS also exhibits higher predictive power than other methods, demonstrating a great advantage in estimating causal networks from distributed data.

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

↑