Parallel structural learning of Bayesian networks: Iterative divide and conquer algorithm based on structural fusion

• Published in: Knowledge-based systems Vol. 296; p. 111840
• Main Authors: Laborda, Jorge D., Torrijos, Pablo, Puerta, José M., Gámez, José A.
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 19.07.2024
• Subjects: Bayesian network fusion/aggregation; Bayesian network learning; Distributed machine learning; High-dimensional problems; Bayesian network fusion/aggregation; Distributed machine learning; Bayesian network learning; High-dimensional problems
• ISSN: 0950-7051; 1872-7409; 1872-7409
• DOI: 10.1016/j.knosys.2024.111840

Learning Bayesian Networks (BNs) from high-dimensional data is a complex and time-consuming task. Although the literature includes approaches based on horizontal (instances) or vertical (variables) partitioning, none can guarantee the same theoretical properties as the Greedy Equivalence Search (GES) algorithm, except those based on the GES algorithm itself. This paper proposes a distributed BN learning algorithm that uses GES as the local learning algorithm, ensuring the same theoretical properties as GES but requiring less CPU time. The two main novelties in our proposed method are (1) the distribution of the set of possible edges among local learning processes, which are constrained to only use its local edge set; and (2) the use of BN fusion to aggregate the networks learned constrained to local edge sets. The algorithm is iterative, and at each step, the last aggregated network is used as the starting point by each local BN process. After a comprehensive experimental evaluation, the results show that the proposed algorithm (pGES) obtains networks of equal or better quality than GES in less computational time. This improvement is especially noticeable in high-dimensional BNs.