Ensemble Learning-Based Mortality Prediction After Acute Myocardial Infarction

• Published in: Shanghai jiao tong da xue xue bao Vol. 30; no. 1; pp. 153 - 165
• Main Authors: Yan, Mingxuan, Miao, Yutong, Sheng, Shuqian, Gan, Xiaoying, He, Ben, Shen, Lan
• Format: Journal Article
• Language: English
• Published: Shanghai Shanghai Jiaotong University Press 01.02.2025; Springer Nature B.V
• Subjects: Algorithms; Architecture; Computer Science; Correlation coefficient; Correlation coefficients; Creatinine; Death; Decision trees; Electrical Engineering; Engineering; Ensemble learning; Heart attacks; Hemoglobin; Life Sciences; Machine learning; Materials Science; Mortality; Myocardial infarction; Patients; Prediction models; Recall; ensemble learning; A; 集成学习; 急性心肌梗死; 机器学习; 特征工程; feature engineering; acute myocardial infarction (AMI); machine learning; R 542.2+2; acute myocardial infarction(AMI)
• ISSN: 1007-1172; 1674-8115; 1995-8188
• DOI: 10.1007/s12204-023-2611-1

A mortality prediction model based on small acute myocardial infarction (AMI) patients coherent with low death rate is established. In total, 1 639 AMI patients are selected as research objects who received treatment in seven tertiary and secondary hospitals in Shanghai between January 1, 2016 and January 1, 2018. Among them, 72 patients deceased during the two-year follow-up. Models are established with ensemble learning framework and machine learning algorithms based on 51 physiological indicators of the patient. Shapley additive explanations algorithm and univariate test with point-biserial and phi correlation coefficients are employed to determine significant features and rank feature importance. Based on 5-fold cross validation experiment and external validation, prediction model with self-paced ensemble framework and random forest algorithm achieves the best performance with area under receiver operating characteristic curve (AUROC) score of 0.911 and recall of 0.864. Both feature ranking methods showed that ejection fractions, serum creatinine (admission), hemoglobin and Killip class are the most important features. With these top-ranked features, the simplified prediction model is capable of achieving a comparable result with AUROC score of 0.872 and recall of 0.818. This work proposes a new method to establish mortality prediction models for AMI patients based on self-paced ensemble framework, which allows models to achieve high performance with small scale of patients coherent with low death rate. It will assist in medical decision and prognosis as a new reference.