A hybrid cost-sensitive ensemble for heart disease prediction

BMC Medical Informatics and Decision Making

Qi Zhenya¹  Zuoru Zhang² 

[1] College of Management and Economics, Tianjin University, Nankai District, 300072, Tianjin, People’s Republic of China;School of Mathematical Science, Hebei Normal University, Yuhua District, 050024, Shijiazhuang, People’s Republic of China;

关键词: Cost-sensitive;    Ensemble;    Heart disease;   

DOI  :  10.1186/s12911-021-01436-7

来源: Springer

【 摘 要 】

BackgroundHeart disease is the primary cause of morbidity and mortality in the world. It includes numerous problems and symptoms. The diagnosis of heart disease is difficult because there are too many factors to analyze. What’s more, the misclassification cost could be very high.MethodsA cost-sensitive ensemble method was proposed to improve the efficiency of diagnosis and reduce the misclassification cost. The proposed method contains five heterogeneous classifiers: random forest, logistic regression, support vector machine, extreme learning machine and k-nearest neighbor. T-test was used to investigate if the performance of the ensemble was better than individual classifiers and the contribution of Relief algorithm.ResultsThe best performance was achieved by the proposed method according to ten-fold cross validation. The statistical tests demonstrated that the performance of the proposed ensemble was significantly superior to individual classifiers, and the efficiency of classification was distinctively improved by Relief algorithm.ConclusionsThe proposed ensemble gained significantly better results compared with individual classifiers and previous studies, which implies that it can be used as a promising alternative tool in medical decision making for heart disease diagnosis.

【 授权许可】

CC BY