Development of a machine learning model to predict the risk of late cardiogenic shock in patients with ST-segment

Zhixun Bai1,2,3, Shan Hu1,2, Yan Wang1,2

  • 1College of Medicine, Soochow University, Suzhou, China.

Insights

This study developed machine learning models to predict cardiogenic shock (CS) risk in ST-elevation myocardial infarction (STEMI) patients. The LASSO model demonstrated superior accuracy, offering a valuable tool for early risk assessment and improved patient outcomes.

Area of Science:

  • Cardiology
  • Medical Informatics
  • Machine Learning

Background:

  • Cardiogenic shock (CS) significantly increases in-hospital mortality for ST-segment elevation myocardial infarction (STEMI) patients, exceeding 50%.
  • Accurate and timely risk assessment is crucial for managing STEMI patients at risk of developing CS.
  • Previous risk models may not adequately account for the dynamic nature of CS development in STEMI.

Purpose of the Study:

  • To compare the performance of various machine learning models in predicting in-hospital CS risk among STEMI patients.
  • To develop and validate a nomogram for predicting late-onset CS risk in STEMI patients.
  • To identify the optimal model for accurate prognostic prediction of CS in STEMI.

Main Methods:

  • Utilized logistic regression (LR), LASSO, SVM, LightGBM, and XGBoost models for CS risk prediction in STEMI.
  • Trained and tested models on datasets comprising 1,598 and 684 STEMI patients, respectively.
  • Evaluated model performance using accuracy, AUC, recall, precision, Gini score, C-index, calibration plots, and decision curve analysis.

Main Results:

  • The LASSO and LR models exhibited the highest predictive power, with accuracy over 0.93 and AUC above 0.82.
  • The LASSO nomogram demonstrated good differentiation and calibration, achieving a C-index of 0.811 in the test set and 0.821 in internal validation.
  • Decision curve analysis confirmed the superior clinical relevance of the LASSO model compared to existing score-based models.

Conclusions:

  • Five machine learning models were developed for in-hospital CS prediction in STEMI patients.
  • The LASSO model emerged as the best-performing predictive tool.
  • The developed LASSO nomogram offers accurate prognostic prediction for CS risk in STEMI patients.
Abstract

Related Concept Videos