Predictive approach for liberation from acute dialysis in ICU patients using interpretable machine learning

Abstract Renal recovery following dialysis-requiring acute kidney injury (AKI-D) is a vital clinical outcome in critical care, yet it remains an understudied area. This retrospective cohort study, conducted in a medical center in Taiwan from 2015 to 2020, enrolled patients with AKI-D during intensive care unit stays. We aimed to develop and temporally test models for predicting dialysis liberation before hospital discharge using machine learning algorithms and explore early predictors. The dataset comprised 90 routinely collected variables within the first three days of dialysis initiation. Out of 1,381 patients who received acute dialysis, 27.3% experienced renal recovery. The cohort was divided into the training group (N = 1135) and temporal testing group (N = 251). The models demonstrated good performance, with an area under the receiver operating characteristic curve of 0.85 (95% CI, 0.81–0.88) and an area under the precision-recall curve of 0.69 (95% CI, 0.62–0.76) for the XGBoost model. Key predictors included urine volume, Charlson comorbidity index, vital sign derivatives (trend of respiratory rate and SpO2), and lactate levels. We successfully developed early prediction models for renal recovery by integrating early changes in vital signs and inputs/outputs, which have the potential to aid clinical decision-making in the ICU.