Machine Learning Model for Risk Prediction of Prolonged Intensive Care Unit in Patients Receiving Intra-aortic Balloon Pump Therapy during Coronary Artery Bypass Graft Surgery.

Journal: Journal of cardiovascular translational research
PMID:

Abstract

This study aimed to construct machine learning models and predict prolonged intensive care units (ICU) stay in patients receiving perioperative intra-aortic balloon pump (IABP) therapy during cardiac surgery. 236 patients were divided into the normal (≤ 14 days) and prolonged (> 14 days) ICU groups based on the 75th percentile of ICU duration across the entire cohort. Seven machine learning models were trained and validated. The Shapley Additive explanations (SHAP) method was employed to illustrate the effects of the features. 94 patients (39.83%) experienced prolonged ICU stay. The XGBoost model outperformed other models in predictive performance, as evidenced by its highest area under the receiver operating characteristic curve (training: 0.92; validation: 0.73). The SHAP analysis identified tracheotomy, albumin, Sv1, and cardiac troponin T as the top four risk variables. The XGBoost model predicted risk variables for prolonged ICU stay in patients, possibly contributing to improving perioperative management and reducing ICU duration.

Authors

  • Changqing Yang
    Department of Emergency, The Sixth Affiliated Hospital of Nantong University, 2 Xinduxi Road, Yancheng, Jiangsu, 224000, China.
  • Peng Zheng
    Department of Neurology, The First Affiliated Hospital, Chongqing Medical University, Chongqing 400016, China.
  • Qian Zhang
    The Neonatal Intensive Care Unit, Peking Union Medical College Hospital, Peking, China.
  • Luo Li
    Department of Cardiovascular Surgery of the First Affiliated Hospital & Institute for Cardiovascular Science, Soochow University, Suzhou, Jiangsu, China.
  • Yajun Zhang
    Shanghai University, School of Computer Engineering and Science, Shanghai, China.
  • Quanye Li
    Department of Emergency, The Sixth Affiliated Hospital of Nantong University, 2 Xinduxi Road, Yancheng, Jiangsu, 224000, China. quanyeli1981@163.com.
  • Sheng Zhao
    Department of Ultrasound, Maternal and Child Health Hospital of Hubei Province, Wuhan 430070, China.
  • Zhan Shi
    School of Computer Engineering, Nanjing Institute of Technology, Nanjing 211167, China.