文章快速检索  
  高级检索
基于人工蜂群算法的贝叶斯网络结构学习
张平 , 刘三阳 , 朱明敏
西安电子科技大学 数学与统计学院,陕西 西安 710071
基金项目: 国家自然科学基金资助项目(61075055);西安电子科技大学基本科研业务基金资助项目(K5051270013)    
摘要: 从数据集中学习贝叶斯网络结构是一个NP难问题。针对此问题提出基于遗传算子的人工蜂群算法。首先,将贝叶斯网络结构映射为一种二进制编码;其次, 根据贝叶斯网络的结构特点, 设计了蜜源的更新策略, 从而将学习贝叶斯网络结构的过程转化为蜂群寻找最优蜜源的过程。实验结果表明, 该算法应用于贝叶斯网络结构学习中的有效性。
关键词: 贝叶斯网络     NP难     人工蜂群算法     遗传算子     结构学习    
Structure learning of Bayesian networks by use of the artificial bee colony algorithm
ZHANG Ping , LIU Sanyang , ZHU Mingmin
School of Mathematics and Statistics, Xidian University, Xi'an 710071, China
Abstract: The learning structure of Bayesian networks from a data set is an NP-hard problem. To deal with this problem, an artificial bee colony algorithm based on genetic operators is proposed in this paper. The structure of the Bayesian network is mapped to binary encoding, and the updated strategy of nectar is designed according to the characteristics of the Bayesian network structure. Thus the process of structure learning of the Bayesian network is transformed into the process of the bee colony finding the optimal nectar. The experimental results show that the algorithm is valid in the structure learning of Bayesian networks.
Key words: Bayesian networks     NP-hard     artificial bee colony     genetic operators     structure learning    

贝叶斯网络(Bayesian networks, BN)[1-2]是概率论与图论结合的产物,已经成为处理不确定性问题的有效工具。近年来,如何从数据集中高效地学习贝叶斯网络结构受到了众多学者的广泛关注。一般地,贝叶斯网络结构学习的方法可以分为两大类:1)基于独立性测试的方法[3-4];2)基于评分搜索的方法[5-7]。基于评分搜索的方法包括2个要素,即评分标准和搜索策略。由于贝叶斯网络结构学习是一个NP难问题,因此基于评分搜索的贝叶斯网络结构学习方法中的搜索算法一般采用启发式搜索算法。国内外的学者提出了许多基于评分搜索的方法,例如Cooper等1992年提出了K2算法[8]。Chickering在2002年提出贪婪算法(greedy search, GS)[9]。李显杰等在2008年将量子遗传算法用于贝叶斯网络结构学习,取得较好的效果,但该方法编码方式较复杂。高晓利在2011年提出了一种改进的学习贝叶斯结构的贪婪算法,该算法结合了条件独立性测试方法,当节点个数增大时,条件独立性测试呈指数增长,这种方法适合节点较少的网络结构。

人工蜂群(artificial bee colony,ABC)算法是D Karaboga [10]2005年提出的一种群体智能优化算法.该算法结构简单、参数较少、易于实现,受到了众多学者的关注和研究,并成功应用于函数优化[11-12]、神经网络训练[13-15]、控制工程[16]等问题。本文将ABC算法用于贝叶斯网络结构学习问题,具体论述了算法的实现过程。最后,通过在Asia和Car网络上测试,结果表明了本文算法的合理性和有效性。

1 贝叶斯网络

贝叶斯网络是一个二元组,即BN=(G, P),其中G=(V, E)为一个有向无环图,表示BN的结构;P表示节点的条件概率分布集合,表示节点之间的依赖程度。根据条件独立性,给定BN,存在一个离散变量集合X={V1, V2, …, Vn}上的联合概率分布P(V):

(1)

式中:Pa(Vi)是Vi的父节点。

假设一组随机变量x1, x2, …, xnD=(D1, D2, …, Dn)是关于这组变量独立分布的数据集。贝叶斯网络的结构学习就是尽可能结合先验知识,找到和样本数据D拟合最好的网络结构。测试网络结构与样本集匹配程度的函数称为评分函数。本文采用基于贝叶斯统计思想的贝叶斯评分函数:

(2)

式中:NijkD中满足xi=kPa(xi)=j的样本个数,Nij=Nijk, αij=αijk,取一个等价样本,其数量为Nαijk=

2 人工蜂群算法原理

ABC算法中,蜜源的位置表示问题的候选解,花蜜数量反映解的质量。最优蜜源的确定依靠人工蜂群的迭代搜索。人工蜂群包括3类:引领蜂、跟随蜂和侦查蜂。引领蜂标记较优蜜源,并将信息传递给跟随蜂,跟随蜂根据蜜源的花蜜数量选择蜜源进行更新,侦查蜂随机搜索新蜜源。

设在D维空间中,种群规模为2×SN(引领蜂个数=跟随蜂个数=SN),蜜源与引领蜂一一对应,即蜜源数目为SN,第i个蜜源的位置记为Xi=xi1, xi2, …, xiD。算法的迭代过程中,跟随蜂根据引领蜂分享的信息,以轮盘赌的方式根据式(3)选择一个蜜源:

(3)

式中:fiti是花蜜数量(适应度),按照式(4)计算:

(4)

式中fi是第i个解的目标函数值。

引领蜂和跟随蜂按照式(5)由蜜源Xi=(xi1, xi2, …, xiD)得到新蜜源:

(5)

式中:k∈{1, 2, …, SN}且kij∈{1, 2, …, D}是随机产生的整数;r∈[-1, 1]是一个随机数。若在一定循环次数(limit)后,蜜源质量没有提高,则引领蜂放弃该蜜源,转变为侦查蜂,按照式(6)随机产生新蜜源:

(6)

式中:ljuj分别是变量xij的下界和上界。

3 人工蜂群算法的贝叶斯网络结构学习

ABC算法中,学习最优贝叶斯网络结构的过程就是蜂群寻找高收益蜜源的过程。引领蜂能够保持优良蜜源,有精英特性;跟随蜂能够增加较优蜜源对应的蜜蜂数量,加速算法的收敛;侦查蜂任意搜索新蜜源,增加了蜜源的多样性,有利于算法跳出局部最优。2个过程的对应关系如表 1所示。

表 1 2个过程的对应关系 Table 1 The corresponding relation of the two processes
蜂群寻找蜜源过程贝叶斯网络结构学习过程
蜜源位置所有贝叶斯网络结构
蜜源的收益贝叶斯网络的得分
高收益蜜源最优贝叶斯网络结构

3.1 贝叶斯网络结构的编码

文中采用文献[17]中的矩阵编码方式,一个包含n个节点的贝叶斯网络可以表示为n×n的矩阵A={aij},其中aij定义如下:

本文算法中表示贝叶斯网络结构的个体为

3.2 基于遗传算子的蜜源更新策略

人工蜂群算法一般用于解决连续优化问题,而贝叶斯网络结构为二进制编码,因此不能直接应用ABC算法中式(5)产生新蜜源。考虑式(5),实际上是两个蜜源之间信息的分享以及个体蜜源自身特性的继承,这类似于遗传算法中的交叉和变异算子,所以采用交叉和变异两种算子对蜜源进行更新。

在交叉算子中,本文采用rand和best两种交叉策略。其中rand交叉策略是指随机选择一个蜜源与待更新蜜源进行两点交叉,能够增强种群之间的信息共享;best交叉策略是指利用当前最优蜜源与待更新蜜源进行两点交叉,有利于最优蜜源特性的继承。文中通过参数p来决定采取哪种交叉策略,p的取值将在实验部分讨论。

在变异算子中,有加边、删边和反转边3种变异操作。本文通过随机均匀产生1、2、3的任意一个整数决定进行哪种操作。若该数为1,则进行加边操作;该数为2,则进行边反转操作;该数为3,则进行删边操作。具体步骤如下:

1)参数设置,概率p

2)产生随机数rnd;

3)如果rnd < p,随机选择一个蜜源与待更新蜜源进行两点交叉;否则,最优蜜源与待更新蜜源进行两点交叉;

4)随机均匀产生1、2、3的任意一个整数;

5))进行加边、反转边或者删边操作。

3.3 修复非法结构图

蜜源进行交叉和变异后可能会成为非法贝叶斯网络结构,图 1是常见的非法结构图。因此,文中采用修复算子[18]对非法贝叶斯网络结构进行修复。修复步骤如下:

图 1 常见非法结构图 Fig. 1 The common illegal structure

1)求出网络结构图对应矩阵的传递闭包;

2)若传递闭包矩阵对角线上的元素全为0,则该图是合法的网络结构;否则,保留主对角线上不为0的元素对应的节点(这些节点全位于环内);

3)任取环内的一点,求出它位于闭环内的所有父节点;

4)删除或者反转这些父节点指向该节点的任一边,使网络结构图中不存在环。

3.4 算法实现步骤

由于初始种群的选择对ABC算法的寻优性能影响较大,因此本文采用Chow等[19]提出的MWST算法得到与贝叶斯网络结构拟合度最高的树结构。然后以该树结构为初始图模型,通过对该图进行加边、删边或反转边的操作产生该图的所有邻近图。在邻近图中选取一定数量的图作为初始蜜源。具体步骤如下:

1)初始化参数SN,limit,p

2)由MWST算法得到初始结构图,并生成该图的所有邻近图,选择SN个结构图为初始蜜源;

3)引领蜂阶段

For i=1、2、…、SN, 重复以下步骤:

①利用交叉和变异算子得到新蜜源;

②判断新蜜源是否存在环。如果存在,则进行修复;

③计算新蜜源的得分,若大于待更新蜜源的得分,则用新蜜源取代待更新蜜源。

4)跟随蜂阶段

For i=1, 2, …, SN, 重复以下步骤:①跟随蜂i根据概率选择一个蜜源;

②对该蜜源利用交叉和变异算子得到新蜜源;

③判断新蜜源是否存在环。若存在,则进行修复;

④计算新蜜源的得分,若大于待更新蜜源的得分,则用新蜜源取代待更新蜜源。

5)侦查蜂阶段

若蜜源连续limit次没有得到提高,则放弃该蜜源,随机产生新蜜源;

6)记忆目前最优蜜源;

7)判断是否达到终止条件,若达到,则输出结果;否则,转3)。

4 实验

本文算法中,种群规模为40(SN=20),limit=100, 最大迭代次数为100。蜜源的更新策略中,参数p的取值通过实验方法得到,p的取值对学习Asia网络结构的影响如图 2所示。

图 2 p的取值对学习Asia网络结构的影响 Fig. 2 The effect of the value of p on Asia network structure

图 2表示样本容量为2 000时,学习Asia网络结构的10次实验的平均汉明距离(SHD),汉明距离用来度量学习所得网络结构与真实网络结构之间的差距,即学习所得网络结构转换到真实网络需要的操作数。

图 3 Asia网络 Fig. 3 Asia network

图 4的曲线可以看出,p=0.4时,SHD最小。究其原因,p的取值过小,种群受当前最优蜜源的影响太大,容易造成早熟收敛;p的取值过大,种群不能充分利用当前最优蜜源的信息,得到最优网络结构的概率减小。

图 4 Car网络 Fig. 4 Car network

为检验算法应用于贝叶斯网络结构学习的有效性,对经典的Asia和Car网络进行结构学习,并与贪婪算法(GS)和爬山算法(HC)进行比较。为了保证3种算法在相同的实验条件下进行,3种算法均采用MWST算法得到初始结构图,在不同样本数据集上分别独立运行10次。测试结果如表 2表 3所示,其中A(G)表示学习所得的网络结构中额外增加的边数;D(G)表示学习所得的网络结构中丢掉的边数;I(G)表示学习所得的网络结构中错误指向的边数;Score表示网络结构的得分值。3种算法对Asia网络和Car网络的学习结果如表 2表 3

表 2 3种算法对Asia网络的学习结果 Table 2 Learning results of the three algorithms for Asia network
算法 1 000 1 500 2 000
A(G)D(G)I(G) A(G)D(G)I(G) A(G)D(G)I(G)
ABC-BN0.51.01.00.51.00.501.00.4
GS0.81.01.70.71.01.50.81.01.0
HC0.81.01.60.81.01.50.61.01.0

表 3 3种算法对Car网络的学习结果 Table 3 Learning results of the three algorithms for Car network
算法 1 000 1 500 2 000
A(G)D(G)I(G) A(G)D(G)I(G) A(G)D(G)I(G)
ABC-BN0.61.01.40.51.01.00.41.00.5
GS1.81.21.71.31.01.20.91.01.1
HC1.81.01.51.51.01.30.61.21.0

表 23的数据可以看出,在样本容量相同的情况下,ABC-BN算法得到的网络结构最接近真实的网络结构,而GS和HC算法得到的网络结构错误边数相对较多。这说明了ABC-BN算法的有效性。

为了说明ABC-BN算法的学习性能。将本文算法与智能算法GA和HPGA-BN[18]比较。各智能算法的种群规模均为40。最大迭代次数为100。表 4列出了3种算法学习Asia网络的有效收敛时间和相应的得分。实验中,每种算法独立运行10次,取平均值。由表 4的数据可以看出,ABC-BN算法得到的网络结构得分最高且有效收敛时间最短。由此证明了ABC-BN算法能较快地收敛到高精度的解。

表 4 3种算法学习Asia网络的结果比较 Table 4 Comparison of the results of the three algorithms for learning Asia network
样本
容量
GA HPGA-BN ABC-BN
得分值 时间/s 得分值 时间/s 得分值 时间/s
1 000-2 292.488.16-2 281.026.54-2 280.243.21
1 500-3 457.108.83-3 446.147.28-3 444.183.85
2 000-4 523.619.14-4 515.427.90-4 515.004.02

5 结束语

本文提出一种基于遗传算子的人工蜂群算法。该算法将遗传算子嵌入到人工蜂群算法中,并成功应用于贝叶斯网络结构学习问题。实验结果表明,该算法具有较快的收敛速率和较高的学习质量。在实际应用中,往往得不到容量较大的样本数据,所以,下一步的研究方向是在样本容量较小的情况下,通过设计更有效的操作算子得到质量较高的网络结构。

参考文献
[1] CAI Z, SUN S, SI S, et al. Identifying product failure rate based on a conditional Bayesian network classifier[J]. Expert Systems with Applications , 2011, 38 (5) : 5036-5043 DOI:10.1016/j.eswa.2010.09.146
[2] HSIEH N C, HUNG L P. A data driven ensemble classifier for credit scoring analysis[J]. Expert Systems with Applications , 2010, 37 (1) : 534-545 DOI:10.1016/j.eswa.2009.05.059
[3] De CAMPOS L M. Independency relationships and learning algorithms for singly connected networks[J]. Journal of Experimental & Theoretical Artificial Intelligence , 1998, 10 (4) : 511-549
[4] De CAMPOS L M, HUETE J F. A new approach for learning belief networks using independence criteria[J]. International Journal of Approximate Reasoning , 2000, 24 (1) : 11-37 DOI:10.1016/S0888-613X(99)00042-0
[5] COOPER G F, HERSKOVITS E. A Bayesian method for the induction of probabilistic networks from data[J]. Machine Learning , 1992, 9 (4) : 309-347
[6] HECKERMAN D, GEIGER D, CHICKERING D M. Learning Bayesian networks: The combination of knowledge and statistical data[J]. Machine Learning , 1995, 20 (3) : 197-243
[7] LAM W, BACCHUS F. Learning Bayesian belief networks: an approach based on the MDL principle[J]. Computational Intelligence , 1994, 10 (3) : 269-293 DOI:10.1111/coin.1994.10.issue-3
[8] COOPER G F, HERSKOVITS E. A Bayesian method for the induction of probabilistic networks from data[J]. Machine Learning , 1992, 9 (4) : 309-347
[9] CHICKERING D M. Optimal structure identification with greedy search[J]. The Journal of Machine Learning Research , 2003 (3) : 507-554
[10] KARABOGA D. An idea based on honey bee swarm for numerical optimization[R]. Erciyes university, engineering faculty, computer engineering department, 2005.
[11] KARABOGA D, BASTURK B. A powerful and efficient algorithm for numerical function optimization: artificial bee colony (ABC) algorithm[J]. Journal of Global Optimization , 2007, 39 (3) : 459-471 DOI:10.1007/s10898-007-9149-x
[12] KARABOGA D, BASTURK B. On the performance of artificial bee colony (ABC) algorithm[J]. Applied soft Computing , 2008, 8 (1) : 687-697 DOI:10.1016/j.asoc.2007.05.007
[13] KARABOGA D, AKAY B. Artificial bee colony (abc) algorithm on training artificial neural networks[C]//2007 IEEE 15th Signal Processing and Communications Applications. Eskisehir: IEEE Press, 2007: 1-4.
[14] KARABOGA D, OZTURK C. Neural networks training by artificial bee colony algorithm on pattern classification[J]. Neural Netw World , 2009, 19 (3) : 279-292
[15] OZTURK C, KARABOGA D. Hybrid artificial bee colony algorithm for neural network training[C]//2011 IEEE Congress on Evolutionary Computation. New Orleans, LA:IEEE Press, 2011: 84-88.
[16] ABACHIZADEH M, YAZDI M, YOUSEFI-KOMA A. Optimal tuning of PID controllers using artificial bee colony algorithm[C]//2010 IEEE/ASME International Conference on Advanced Intelligent. Montreal: IEEE Press, 2010: 379-384.
[17] LARRANAGA P, POZA M, YURRAMENDI Y. Structure learning of Bayesian networks by genetic algorithms: a performance analysis of control parameters[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 1996, 18 (9) : 912-926 DOI:10.1109/34.537345
[18] 许丽佳, 黄建国, 王厚军, 等. 混合优化的贝叶斯网络结构学习[J]. 计算机辅助设计与图形学报 , 2009, 21 (5) : 633-639 XU Lijia, HUANG Jianguo, WANG Houjun, et al. Hybrid optimized algorithm for learning Bayesian network structure[J]. Journal of Computer-Aided Design & Computer Graphics , 2009, 21 (5) : 633-639
[19] CHOW C, LIU C. Approximation discrete probability distributions with dependence trees[J]. IEEE Transactions on Information Theory , 1968, 14 (3) : 462-467 DOI:10.1109/TIT.1968.1054142
DOI: 10.3969/j.issn.1673-4785.201310014
中国人工智能学会和哈尔滨工程大学联合主办。
0

文章信息

张平, 刘三阳, 朱明敏
ZHANG Ping, LIU Sanyang, ZHU Mingmin
基于人工蜂群算法的贝叶斯网络结构学习
Structure learning of Bayesian networks by use of the artificial bee colony algorithm
智能系统学报, 2014, 9(3): 325-329
CAAI Transactions on Intelligent Systems, 2014, 9(3): 325-329
http://dx.doi.org/10.3969/j.issn.1673-4785.201310014

文章历史

收稿日期: 2013-11-14
网络出版日期: 2014-06-14

相关文章

工作空间