返回
cuPC: CUDA-Based Parallel PC Algorithm for Causal Structure Learning on GPU
DOI:10.1109/TPDS.2019.2939126.png)
摘要
En 中文
The main goal in many fields in the empirical sciences is to discover causal relationships among a set of variables from observational data. PC algorithm is one of the promising solutions to learn underlying causal structure by performing a number of conditional independence tests. In this paper, we propose a novel GPU-based parallel algorithm, called cuPC, to execute an order-independent version of PC. The proposed solution has two variants, cuPC-E and cuPC-S, which parallelize PC in two different ways for multivariate normal distribution. Experimental results show the scalability of the proposed algorithms with respect to the number of variables, the number of samples, and different graph densities. For instance, in one of the most challenging datasets, the runtime is reduced from more than 11 hours to about 4 seconds. On average, cuPC-E and cuPC-S achieve 500X and 1300X speedup, respectively, compared to serial implementation on CPU.
Keyword:
Bayesian networks
causal discovery
CUDA
GPU
machine learning
parallel processing
PC algorithm
AI总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
IF:
6
论文数:
5.2K
被引数:
1.1W
机构
引用论文
A parallel algorithm for Bayesian network structure learning from large data sets一种面向大数据集的贝叶斯网络结构学习并行算法
LEARNING HIGH-DIMENSIONAL DIRECTED ACYCLIC GRAPHS WITH LATENT AND SELECTION VARIABLES学习具有潜在变量和选择变量的高维有向无环图
ANNALS OF STATISTICS
IF3.7

