Entropy-based pruning for learning Bayesian networks using BIC

被引:28
作者
de Campos, Cassio P. [1 ,2 ]
Scanagatta, Mauro [3 ]
Corani, Giorgio [3 ]
Zaffalon, Marco [3 ]
机构
[1] Univ Utrecht, Utrecht, Netherlands
[2] Queens Univ Belfast, Belfast, Antrim, North Ireland
[3] Ist Dalle Molle Studi Intelligenza Artificiale ID, Lugano, Switzerland
基金
瑞士国家科学基金会;
关键词
Structure learning; Bayesian networks; BIC; Parent set pruning;
D O I
10.1016/j.artint.2018.04.002
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
For decomposable score-based structure learning of Bayesian networks, existing approaches first compute a collection of candidate parent sets for each variable and then optimize over this collection by choosing one parent set for each variable without creating directed cycles while maximizing the total score. We target the task of constructing the collection of candidate parent sets when the score of choice is the Bayesian Information Criterion (BIC). We provide new non-trivial results that can be used to prune the search space of candidate parent sets of each node. We analyze how these new results relate to previous ideas in the literature both theoretically and empirically. We show in experiments with UCI data sets that gains can be significant. Since the new pruning rules are easy to implement and have low computational costs, they can be promptly integrated into all state-of-the-art methods for structure learning of Bayesian networks. (C) 2018 Elsevier B.V. All rights reserved.
引用
收藏
页码:42 / 50
页数:9
相关论文
共 13 条
[1]   Integer Linear Programming for the Bayesian network structure learning problem [J].
Bartlett, Mark ;
Cussens, James .
ARTIFICIAL INTELLIGENCE, 2017, 244 :258-271
[2]  
Chickering DM, 2004, J MACH LEARN RES, V5, P1287
[3]  
COOPER GF, 1992, MACH LEARN, V9, P309, DOI 10.1007/BF00994110
[4]  
de Campos CP, 2010, AAAI CONF ARTIF INTE, P431
[5]  
de Campos CP, 2011, J MACH LEARN RES, V12, P663
[6]  
HECKERMAN D, 1995, MACH LEARN, V20, P197, DOI 10.1007/BF00994016
[7]  
Ji Qiang, 2009, P 26 ANN INT C MACHI, P113
[8]   Parent assignment is hard for the MDL, AIC, and NML costs [J].
Koivisto, Mikko .
LEARNING THEORY, PROCEEDINGS, 2006, 4005 :289-303
[9]  
LICHMAN M., 2013, UCI MACHINE LEARNING
[10]  
Pearl J., 1988, Probabilistic Reasoning in Intelligent Systems: Networks of Plausible Inference, DOI DOI 10.1016/C2009-0-27609-4