Related Experiment Video
Updated: Jan 18, 2026
Predicting Amputation using Local Circulating Mononuclear Progenitor Cells in Angioplasty-treated Patients with Critical Limb Ischemia
Published on: September 22, 2020
Enhancing one-year mortality prediction in STEMI patients post-PCI: an interpretable machine learning model with risk
Wenqiang Li1,2, Dongdong Yan1,3,4,5, Wei Hu1,3,4,5
1The First Clinical Medical School, Lanzhou University, Lanzhou, China.
Insights
Machine learning models can accurately predict one-year mortality in ST-elevation myocardial infarction (STEMI) patients after percutaneous coronary intervention (PCI). A simplified model with risk stratification achieved high predictive performance, aiding post-discharge management.
Area of Science:
- Cardiology
- Medical Informatics
- Machine Learning
Background:
- ST-elevation myocardial infarction (STEMI) is a major cause of mortality and disability worldwide.
- Percutaneous coronary intervention (PCI) has improved in-hospital survival, emphasizing the need for effective post-discharge care.
- Machine learning (ML) shows potential for predicting adverse outcomes in STEMI patients, but simple, accurate models are needed.
Purpose of the Study:
- To develop and validate a machine learning model for predicting one-year mortality in STEMI patients post-PCI.
- To identify key predictors of mortality and assess the impact of risk stratification on model performance.
- To create an accurate and interpretable model for improved long-term outcome prediction.
Main Methods:
- Retrospective analysis of 1,274 STEMI patients undergoing PCI.
- Application of data processing and ML algorithms (Random Forest) to predict one-year mortality.
- Evaluation of model performance using AUROC, AUPRC, accuracy, sensitivity, precision, and F1-score.
Main Results:
- The Random Forest model achieved an AUROC of 0.94 and AUPRC of 0.44.
- Key predictors identified: cardiogenic shock, creatinine, NT-proBNP, diastolic blood pressure, and left ventricular ejection fraction.
- Integrating risk stratification improved performance to AUROC 0.97 and AUPRC 0.74.
Conclusions:
- Accurate and interpretable ML models can be built using a minimal set of predictors for STEMI patients.
- Risk stratification enhances the predictive power of ML models for long-term outcomes.
- This approach facilitates improved post-discharge management and outcome prediction for STEMI survivors.
Background:
ST-elevation myocardial infarction (STEMI) poses a significant threat to global mortality and disability. Advances in percutaneous coronary intervention (PCI) have reduced in-hospital mortality, highlighting the importance of post-discharge management. Machine learning (ML) models have shown promise in predicting adverse clinical outcomes. However, a systematic approach that combines high predictive accuracy with model simplicity is still lacking.
Methods:
This retrospective study applied three data processing and ML algorithms to address class imbalance and support model development. ML models were trained to predict one-year mortality in STEMI patients post-PCI, with performance evaluated using accuracy, sensitivity, precision, F1-score, area under the receiver operating characteristic curve (AUROC), and the area under the precision-recall curve (AUPRC).
Results:
We analyzed data from 1,274 patients, incorporating 46 clinical and laboratory features. Using the Random Forest (RF) algorithm, we achieved an AUROC of 0.94 (95% confidence interval (CI): 0.90-0.98), an AUPRC of 0.44 (95% CI:0.15-0.76) in the internal validation set, identifying five key predictors: cardiogenic shock, creatinine, NT-proBNP, diastolic blood pressure, and left ventricular ejection fraction. By integrating risk stratification, the model's performance improved, achieving an AUROC of 0.97 (95% CI: 0.96-0.99) and an AUPRC of 0.74 (95% CI: 0.60-0.84).
Conclusion:
This study highlights the feasibility of constructing accurate and interpretable ML models using a minimal set of predictors, supplemented by risk stratification, to improve long-term outcome prediction in STEMI patients.
More Related Videos
06:19Constructing and Visualizing Models using Mime-based Machine-learning Framework
Published on: July 22, 2025
04:09Predicting Treatment Response to Image-Guided Therapies Using Machine Learning: An Example for Trans-Arterial Treatment of Hepatocellular Carcinoma
Published on: October 10, 2018