Improved heuristic equivalent search algorithm based on Maximal Information Coefficient for Bayesian Network Structure Learning

Yinghua Zhang, Wensheng Zhang, Yuan Xie

Research output: Contribution to journalArticlepeer-review

37 Scopus citations

Abstract

Greedy Equivalent Search (GES) is an effective algorithm for Bayesian network structure learning problem, which searches in the space of graph equivalence classes. However, original GES which takes greedy strategy into account may easily fall into local optimization trap because of the empty initial structure. In this paper, an improved GES method is proposed. It firstly designs a draft of the real network, based on conditional independence tests and Maximum Information Coefficient, which helps in finding more correct dependent relationship between variables. To ensure correctness, this draft is used as a seed structure of original GES algorithm. Numerical experiments on four standard networks show that SCo (the value of the BDeu score) and NEtoGS (the number of graph structure, which is equivalent to the Gold Standard network) have big improvement. Also, the total of learning time is greatly reduced. Therefore, our improved method can relatively quickly determine the structure with highest degree of data matching.

Original languageEnglish
Pages (from-to)186-195
Number of pages10
JournalNeurocomputing
Volume117
DOIs
StatePublished - 6 Oct 2013
Externally publishedYes

Keywords

  • Bayesian network
  • Heuristic search
  • Maximal information coefficient
  • Structure learning

Fingerprint

Dive into the research topics of 'Improved heuristic equivalent search algorithm based on Maximal Information Coefficient for Bayesian Network Structure Learning'. Together they form a unique fingerprint.

Cite this