arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.05269stat.MLcs.LGstat.ME

从不完全数据学习贝叶斯网络中的硬EM与软EM

Hard and Soft EM in Bayesian Network Learning from Incomplete Data

Istituto Dalle Molle di Studi sull'Intelligenza Artificiale (IDSIA)(达勒·莫勒人工智能研究所)

Andrea Ruggieri, Francesco Stranieri, Fabio Stella, Marco Scutari

AI总结:

本文比较贝叶斯网络从不完全数据学习时硬EM插补与软EM置信传播对模型质量的影响,并基于模拟结果构建决策树以指导算法选择。

摘要:

不完全数据是从临床试验到工业应用等众多领域中的常见特征。贝叶斯网络(BN)因其图形化和因果解释能力而常被用于这些领域。从不完全数据进行BN参数学习通常采用期望最大化算法(EM)实现,该算法利用置信传播计算相关充分统计量(“软EM”)。类似地,结构期望最大化算法(Structural EM)利用为完全数据设计的算法,从这些充分统计量中学习BN的网络结构。然而,参数与结构学习的实际实现中,出于实现简便性和计算速度的考虑,常常通过插补缺失数据(“硬EM”)来计算充分统计量,而非使用置信传播。本文研究如下问题:使用插补而非置信传播,对所得BN的质量有何影响?通过一项使用合成数据和参考BN的模拟研究,我们发现可以在若干场景下根据数据特征推荐其中一种方法优于另一种。随后,我们利用这些信息构建了一棵简单的决策树,以指导从业者选择最适合其问题的EM算法。

英文摘要:

Incomplete data are a common feature in many domains, from clinical trials to industrial applications. Bayesian networks (BNs) are often used in these domains because of their graphical and causal interpretations. BN parameter learning from incomplete data is usually implemented with the Expectation-Maximisation algorithm (EM), which computes the relevant sufficient statistics ("soft EM") using belief propagation. Similarly, the Structural Expectation-Maximisation algorithm (Structural EM) learns the network structure of the BN from those sufficient statistics using algorithms designed for complete data. However, practical implementations of parameter and structure learning often impute missing data ("hard EM") to compute sufficient statistics instead of using belief propagation, for both ease of implementation and computational speed. In this paper, we investigate the question: what is the impact of using imputation instead of belief propagation on the quality of the resulting BNs? From a simulation study using synthetic data and reference BNs, we find that it is possible to recommend one approach over the other in several scenarios based on the characteristics of the data. We then use this information to build a simple decision tree to guide practitioners in choosing the EM algorithm best suited to their problem.

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

↑