A Dynamic Programming Bayesian Network Structure Learning Algorithm Based on Mutual Information
Zhigang Lv, Ye Li, Ruohai Di, Hongxi Wang, Liangliang Li, Peng Wang, Xiaoyan Li · Journal of Circuits Systems and Computers · 2022
Existing Bayesian network (BN) structure learning algorithms based on dynamic programming have high computational complexity and are difficult to apply to large-scale networks. Therefore, this paper proposes a Dynamic Programming BN structure learning algorithm based on Mutual Information, the MIDP (Dynamic Programming Based on Mutual Information) algorithm. The algorithm uses mutual information to build the maximum spanning tree and [Formula: see text]-order matrix, and introduces a penalty coefficient d based on the matrix-based node removal strategy, so as to reduce the number of scoring calculations and time consumption of the algorithm. Simulation results show that, compared with DP, SMDP and MEDP algorithms, the MIDP algorithm can reduce the calculation times and time consumption of algorithm scores while maintaining the accuracy of the algorithm when selecting the appropriate [Formula: see text] value.