A Bayesian network perspective on neonatal pneumonia in pregnant women with diabetes mellitus

Abstract

Objective: To predict the influencing factors of neonatal pneumonia in pregnant women with diabetes mellitus using a Bayesian network model. By examining the intricate network connections between the numerous variables given by Bayesian networks (BN), this study aims to compare the prediction effect of the Bayesian network model and to analyze the influencing factors directly associated to neonatal pneumonia.

Method: Through the structure learning algorithms of BN, Naive Bayesian (NB), Tree Augmented Naive Bayes (TAN), and k-Dependence Bayesian Classifier (KDB), complex networks connecting variables were presented and their predictive abilities were tested. The BN model and three machine learning models computed using the R bnlean package were also compared in the data set.

Results: In constraint-based algorithms, three algorithms had different presentation DAGs. KDB had a better prediction effect than NB and TAN, and it achieved higher AUC compared with TAN. Among three machine learning modes, Support Vector Machine showed a accuracy rate of 91.04% and 67.88% of precision, which was lower than TAN (92.70%; 72.10%).

Conclusion: KDB was applicable, and it can detect the dependencies between variables, identify more potential associations and track changes between variables and outcome.

Keywords: Bayesian networks; K-Dependence Bayesian Classifier; Naive Bayes network; Neonatal pneumonia; Tree Augmented Naive Bayes model.