Construction and evaluation of a mortality prediction model for patients with acute kidney injury undergoing continuous renal replacement therapy based on machine learning algorithms

Abstract

Background To construct and evaluate a predictive model for in-hospital mortality among critically ill patients with acute kidney injury (AKI) undergoing continuous renal replacement therapy (CRRT), based on nine machine learning (ML) algorithm. Methods The study retrospectively included patients with AKI who underwent CRRT during their initial hospitalization in the United States using the medical information mart for intensive care (MIMIC) database IV (version 2.0), as well as in the intensive care unit (ICU) of Huzhou Central Hospital. Patients from the MIMIC database were used as the training cohort to construct the models (from 2008 to 2019, n = 1068). Patients from Huzhou Central Hospital were utilized as the external validation cohort to evaluate the models (from June 2019 to December 2022, n = 327). In the training cohort, least absolute shrinkage and selection operator (LASSO) regression with cross-validation was employed to select features for constructing the model and subsequently established nine ML predictive models. The performance of these nine models on the external validation cohort dataset was comprehensively evaluated based on the area under the receiver operating characteristic curve (AUROC) and the optimal model was selected. A static nomogram and a web-based dynamic nomogram were presented, with a comprehensive evaluation from the perspectives of discrimination (AUROC), calibration (calibration curve) and clinical practicability (DCA curves). Results Finally, 1395 eligible patients were enrolled, including 1068 patients in the training cohort and 327 patients in the external validation cohort. In the training cohort, LASSO regression with cross-validation was employed to select features and nine models were individually constructed. Compared to the other eight models, the Lasso regularized logistic regression (Lasso-LR) model exhibited the highest AUROC (0.756) and the optimal calibration curve. The DCA curve suggested a certain clinical utility in predicting in-hospital mortality among critically ill patients with AKI undergoing CRRT. Consequently, the Lasso-LR model was the optimal model and it was visualized as a common nomogram (static nomogram) and a web-based dynamic nomogram (https://chsyh2006.shinyapps.io/dynnomapp/). Discrimination, calibration and DCA curves were employed to assess the performance of the nomogram. The AUROC for the training and external validation cohorts in the nomogram model was 0.771 (95%CI: 0.743, 0.799) and 0.756 (95%CI: 0.702, 0.809), respectively. The calibration slope and Brier score for the training cohort were 1.000 and 0.195, while for the external validation cohort, they were 0.849 and 0.197, respectively. The DCA indicated that the model had a certain clinical application value. Conclusions Our study selected the optimal model and visualized it as a static and dynamic nomogram integrating clinical predictors, so that clinicians can personalized predict the in-hospital outcome of critically ill patients with AKI undergoing CRRT upon ICU admission.