Development and validation of an interpretable machine learning model for predicting 5-year major adverse

Zhongxing Jiang1, Haofeng Zhou2,3, Yindu Liu4

  • 1Department of Cardiology, Guangdong Provincial People's Hospital, Guangdong Academy of Medical Sciences, Southern Medical University, 106 Zhongshan Second Road, Yuexiu District, Guangzhou, Guangdong Province, China.

Insights

Machine learning models can predict 5-year major adverse cardiovascular events (MACE) in hospitalized coronary artery disease (CAD) patients. The random survival forest model shows promise for personalized risk stratification and secondary prevention strategies.

Area of Science:

  • Cardiology
  • Machine Learning
  • Predictive Analytics

Background:

  • Coronary artery disease (CAD) is a leading cause of cardiovascular mortality globally.
  • Accurate prognosis for CAD patients is crucial for effective clinical management.
  • This study focuses on developing interpretable machine learning (ML) models for predicting 5-year MACE in hospitalized CAD patients.

Purpose of the Study:

  • To develop and validate interpretable ML models for predicting 5-year MACE in hospitalized CAD patients.
  • To identify key predictors of MACE using LASSO regression.
  • To assess the clinical utility and interpretability of ML models for risk stratification.

Main Methods:

  • A prospective cohort of 705 CAD patients was used, divided into training and validation sets.
  • Least Absolute Shrinkage and Selection Operator (LASSO) regression was employed for predictor selection.
  • Four survival-based ML models were developed and evaluated using discrimination, calibration, and decision curve analysis, with SHAP analysis for interpretability.

Main Results:

  • The study included 705 hospitalized CAD patients; 31.3% experienced MACE within 5 years.
  • Eleven key predictors were identified, including LVEF, NT-proBNP, nitrate use, CAD duration, depressive symptoms, and age.
  • The Random Survival Forest (RSF) model demonstrated strong performance (C-index 0.804 training, 0.710 validation) and clinical utility, with LVEF, age, and diseased vessels being most impactful.

Conclusions:

  • The RSF model shows favorable discrimination, calibration, and clinical utility for predicting 5-year MACE in hospitalized CAD patients.
  • Interpretable ML models, like RSF, can aid in individualized risk stratification for CAD patients.
  • These ML approaches may enhance secondary prevention strategies for improved patient outcomes.
Abstract

Related Concept Videos