Ensemble Learning-Based Mortality Prediction After Acute Myocardial Infarction  

作　　者：YAN Mingruan MIAO Yutong SHENG Shuqian GAN Xiaoying HE Ben SHEN Lan 颜铭萱;苗雨桐;盛淑茜;甘小莺;何奔;沈兰

机构地区：[1]School of Electronic Information and Electrical Engineering,Shanghai Jiao Tong University,Shanghai,200240,China [2]Department of Cardiology,Shanghai Chest Hospital,Shanghai Jiao Tong University School of Medicine,Shanghai,200030,China [3]Clinical Research Center,Shanghai Chest Hospital,Shanghai Jiao Tong University School of Medicine,Shanghai,200030,China

出　　处：《Journal of Shanghai Jiaotong university(Science)》2025年第1期153-165,共13页上海交通大学学报(英文版)

基　　金：the National Natural Science Foundation of China(No.81900308)。

摘　　要：A mortality prediction model based on small acute myocardial infarction(AMI)patients coherent with low death rate is established.In total,1639 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.

关 键 词：acute myocardial infarction(AMI) ensemble learning machine learning feature engineering 

分 类 号：R542.2+2[医药卫生—心血管疾病]