Machine-learning based risk prediction of in-hospital outcomes following STEMI: the STEMI-ML score

Background: Traditional prognostic models for ST-segment elevation myocardial infarction (STEMI) have limitations in statistical methods and usability. Objective: We aimed to develop a machine-learning (ML) based risk score to predict in-hospital mortality, intensive care unit (ICU) admission, and left ventricular ejection fraction less than 40% (LVEF < 40%) in STEMI patients. Methods: We reviewed 1,863 consecutive STEMI patients undergoing primary percutaneous coronary intervention (pPCI) or rescue PCI. Eight supervised ML methods [LASSO, ridge, elastic net (EN), decision tree, support vector machine, random forest, AdaBoost and gradient boosting] were trained and validated. A feature selection method was used to establish more informative and nonredundant variables, which were then considered in groups of 5/10/15/20/25/30(all). Final models were chosen to optimise area under the curve (AUC) score while ensuring interpretability. Results: Overall, 128 (6.9%) patients died in hospital, with 292 (15.7%) patients requiring ICU admission and 373 (20.0%) patients with LVEF < 40%. The best-performing model with 5 included variables, EN, achieved an AUC of 0.79 for in-hospital mortality, 0.78 for ICU admission, and 0.74 for LVEF < 40%. The included variables were age, pre-hospital cardiac arrest, robust collateral recruitment (Rentrop grade 2 or 3), family history of coronary disease, initial systolic blood pressure, initial heart rate, hypercholesterolemia, culprit vessel, smoking status and TIMI flow pre-PCI. We developed a user-friendly web application for real-world use, yielding risk scores as a percentage. Conclusions: The STEMI-ML score effectively predicts in-hospital outcomes in STEMI patients and may assist with risk stratification and individualising patient management.