Bayesian network optimized random forest imputation algorithm
10.16781/j.CN31-2187/R.20230333
- VernacularTitle:基于贝叶斯网络的随机森林优化填补算法
- Author:
Xinyu DONG
1
;
Qi CHEN
;
Zhiyu YANG
;
Jia HE
Author Information
1. 同济大学医学院,上海 200092
- Keywords:
prediction model;
missing data imputation;
random forest;
bayesian network
- From:
Academic Journal of Naval Medical University
2025;46(2):253-257
- CountryChina
- Language:Chinese
-
Abstract:
Objective To evaluate and improve missing data imputation methods to enhance the performance of binary classification prediction model.Methods By simulating data missing scenes,the effects of direct elimination,mean imputation,random forest(RF)imputation algorithm,and multiple imputation-random forest(MI-RF)on the performance of the prediction model were jointly evaluated by receiver operating characteristic area under curve(AUC)and root mean square error(RMSE).Bayesian Network was introduced for the random forest imputation algorithm to optimize the imputation method using the correlations between variables.Results Under different missing proportions,both AUC and RMSE indicated that Bayesian network optimized random forest(BN-RF)imputation algorithm was better.In addition,when the missing proportion was 10％-20％,various imputation methods had roughly the same improvement effect for the prediction model;when the proportion of missing data was 30％-40％,compared to the mean imputation,except for the BN-RF,RF was more effective and its effect was slightly better than MI-RF;however,when the proportion of missing data was close to 50％,even if the model performance was still appropriate,the imputation data gradually deviated from the true data features,resulting in a decrease in the usability of the model.Conclusion The overall effect of BN-RF is satisfactory,and it should be chosen when random missing was 30％-40％.
