Bayesian network structure learning based on HC-PSO algorithm

• Published in: Journal of intelligent & fuzzy systems Vol. 46; no. 2; pp. 4347 - 4359
• Main Authors: Gao, Wenlong, Zhi, Minqian, Ke, Yongsong, Wang, Xiaolong, Zhuo, Yun, Liu, Anping, Yang, Yi
• Format: Journal Article
• Language: English
• Published: London, England SAGE Publications 14.02.2024; Sage Publications Ltd
• Subjects: Bayesian analysis; Crossovers; Evolutionary computation; Machine learning; Mutation; Particle swarm optimization; Search algorithms; DE algorithm; HC algorithm; Keywords. Bayesian network; structure learning; PSO algorithm
• ISSN: 1064-1246; 1875-8967
• DOI: 10.3233/JIFS-236454

Structure learning is the core of graph model Bayesian Network learning, and the current mainstream single search algorithm has problems such as poor learning effect, fuzzy initial network, and easy falling into local optimum. In this paper, we propose a heuristic learning algorithm HC-PSO combining the HC (Hill Climbing) algorithm and PSO (Particle Swarm Optimization) algorithm, which firstly uses HC algorithm to search for locally optimal network structures, takes these networks as the initial networks, then introduces mutation operator and crossover operator, and uses PSO algorithm for global search. Meanwhile, we use the DE (Differential Evolution) strategy to select the mutation operator and crossover operator. Finally, experiments are conducted in four different datasets to calculate BIC (Bayesian Information Criterion) and HD (Hamming Distance), and comparative analysis is made with other algorithms, the structure shows that the HC-PSO algorithm is superior in feasibility and accuracy.