In BMC medical informatics and decision making ; h5-index 38.0
BACKGROUND : Clinical diagnosis based on machine learning usually uses case samples as training samples, and uses machine learning to construct disease prediction models characterized by descriptive texts of clinical manifestations. However, the problem of sample imbalance often exists in the medical field, which leads to a decrease in classification performance of the machine learning.
METHODS : To solve the problem of sample imbalance in medical dataset, we propose a hybrid sampling algorithm combining synthetic minority over-sampling technique (SMOTE) and edited nearest neighbor (ENN). Firstly, the SMOTE is used to over-sampling missed abortion and diabetes datasets, so that the number of samples of the two classes is balanced. Then, ENN is used to under-sampling the over-sampled dataset to delete the "noisy sample" in the majority. Finally, Random forest is used to model and predict the sampled missed abortion and diabetes datasets to achieve an accurate clinical diagnosis.
RESULTS : Experimental results show that Random forest has the best classification performance on missed abortion and diabetes datasets after SMOTE-ENN sampled, and the MCC index is 95.6% and 90.0%, respectively. In addition, the results of pairwise comparison and multiple comparisons show that the SMOTE-ENN is significantly better than other sampling algorithms.
CONCLUSION : Random forest has significantly improved all indexes on the missed abortion dataset after SMOTE-ENN sampled.
Yang Fangyuan, Wang Kang, Sun Lisha, Zhai Mengjiao, Song Jiejie, Wang Hong
2022-Dec-29
Data sampling, Decision tree, Ensemble algorithm, Imbalanced medical data
