Machine learning based model for risk prediction after ST-Elevation myocardial infarction: Insights from the North

Manu Kumar Shetty1, Shekhar Kunal2, M P Girish2

  • 1Department of Clinical Pharmacology, Maulana Azad Medical College, Delhi, India.

Insights

Machine learning models improve mortality prediction for ST-Elevation Myocardial Infarction (STEMI) patients in resource-limited settings. Key predictors include delayed revascularization and heart failure, guiding intensive patient monitoring.

Area of Science:

  • Cardiology
  • Artificial Intelligence
  • Health Informatics

Background:

  • Accurate mortality risk prediction for ST-Elevation Myocardial Infarction (STEMI) is crucial in resource-limited countries.
  • Identifying high-risk patients enables targeted, intensive management strategies.

Purpose of the Study:

  • To develop and validate machine learning (ML) models for predicting 30-day mortality in STEMI patients.
  • To identify key clinical predictors of mortality using interpretable AI.

Main Methods:

  • Utilized data from the North India ST-Elevation Myocardial Infarction (NORIN-STEMI) registry (3191 patients).
  • Trained and validated various ML models using 31 clinical characteristics.
  • Employed Shapley Additive exPlanations (ShAP) for model interpretability.

Main Results:

  • The Extra Tree ML model demonstrated strong predictive performance (AUC: 79.7%, Accuracy: 75%) on the validation set.
  • Identified significant mortality predictors: delayed revascularization, cardiogenic shock, low ejection fraction, elevated creatinine, heart failure, female sex, and mitral regurgitation.
  • Overall 30-day mortality in the study cohort was 7.7%.

Conclusions:

  • ML models offer enhanced mortality prediction for STEMI patients.
  • Interpretable AI (ShAP) effectively highlights critical risk factors, aiding in the identification of individuals requiring intensified monitoring and care.
Abstract