Construction and validation of prognostic model for ICU mortality in cardiac arrest patients: an interpretable machine learning modeling approach
摘要
The incidence and mortality of cardiac arrest (CA) is high. We developed interpretable machine learning models for early prediction of ICU mortality risk in patients diagnosed with CA.
MethodsData from the Medical Information Mart for Intensive Care (MIMIC-IV, version 2.2) was randomized to training set (0.7) and internal validation set (0.3), and data from eICU(version 2.0.1) was used as external validation set. Five models including Logistic Regression (LR), Random Forest (RF), K Nearest Neighbor (KNN), Decision Tree (DT), and Extreme Gradient Boost (XGBoost) were developed. The model with the largest area under the Receiver Operating Characteristic (ROC) curve (AUC) and good performance in other features was defined as the best model, and Shapley Additive Explanations (SHAP) was used to improve the interpretability of the optimal model.
ResultsA total of 1088 patients from MIMIC-IV, and 3542 patients from eICU were included. Seven variables were selected to construct models by Least Absolute Shrinkage and Selection Operator (LASSO) regression. The RF model was the best predictive model with AUC and 95% CI at 0.83 (0.78–0.88) in internal validation set, and 0.71(0.68–0.74) in external validation set. SHAP analysis found that the variables that had a high impact on the risk of ICU death were minimal Glasgow Coma Scale (GCS), base excess, anion gap, and urine output.
ConclusionRF is the optimal model for predicting the risk of ICU death in CA patients. The development of this model is important for early identification and intervention of CA patients who are at risk of dying in the ICU.