An In-Hospital Mortality Risk Model for Patients Undergoing Coronary Artery Bypass Grafting Based on Machine

Kun Zhu1, Wenyuan Lu1, Shui Liu2

  • 1Cardiac Surgery Centre, Fuwai Hospital, Chinese Academy of Medical Sciences and Peking Union Medical College, Beijing, China.

Insights

A new machine learning model using Extreme Gradient Boosting (XGBoost) accurately predicts in-hospital mortality after coronary artery bypass grafting (CABG). This AI tool outperforms existing risk scores, offering better risk stratification for cardiac surgery patients.

Area of Science:

  • Cardiovascular Surgery
  • Medical Informatics
  • Machine Learning in Healthcare

Background:

  • Ischemic heart disease is a leading global cause of death, with coronary artery bypass grafting (CABG) as the primary surgical intervention.
  • Existing models for predicting postoperative mortality after CABG lack sufficient accuracy and broad applicability.

Purpose of the Study:

  • To develop and validate a machine learning-based system for predicting in-hospital mortality in patients undergoing CABG.
  • To compare the performance of the developed model against established risk assessment tools: EuroSCORE II and SinoSCORE.

Main Methods:

  • Utilized data from 21,443 patients in the Chinese Cardiac Surgery Registry (2017-2020), randomly divided into training and testing cohorts.
  • Employed machine learning algorithms, including Extreme Gradient Boosting (XGBoost), addressing class imbalance with SMOTE and hyperparameter optimization.
  • Evaluated model performance using AUC, Brier score, and decision curve analysis, comparing against EuroSCORE II and SinoSCORE.

Main Results:

  • The XGBoost model demonstrated superior predictive performance with an AUC of 0.782 in the independent test cohort, significantly outperforming EuroSCORE II (AUC=0.722) and SinoSCORE (AUC=0.726).
  • The model showed excellent discrimination and calibration (P<.05) compared to traditional risk scores.
  • Overall in-hospital mortality in the study cohort was 2.1%.

Conclusions:

  • A machine learning model, specifically XGBoost, provides a more accurate prediction of in-hospital mortality post-CABG in a Chinese population.
  • Locally calibrated models like XGBoost may offer improved risk stratification for specific patient demographics compared to generalized scores.
  • A 7-variable web calculator derived from the model can aid bedside risk stratification, warranting further prospective validation for clinical decision-making.
Abstract

Related Concept Videos