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.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Acute kidney injury (AKI) and severe AKI that requires renal replacement therapy (RRT) initiation accounts for 50% and 5%–15% of intensive care units (ICU) patients, respectively1,2. AKI morbidity and mortality increase with increase in its severity and is highest among patients with AKI requiring dialysis (AKI-D)3. Outcomes for AKI-D can vary from successful recovery without the need for RRT to dialysis dependent, or even mortality4. Nonrecovery of renal function is a vital morbid event with long-term implications for both patients and healthcare system, including longer ICU stay, higher mortality, and economic burden4.
Over the past two decades, several studies have investigated various variables5,6, including conventional biochemical markers used as surrogates of kidney function7,8, urine output changes9, and novel kidney biomarkers10,11, to predict successful dialysis discontinuation. However, parameters assessed at the initiation or early stages of dialysis have shown relatively poor predictive values. Early and accurate stratification of renal prognosis is essential for closely monitoring renal function, avoiding nephrotoxic agents, applying kidney protective measures, adjusting dialysis parameters, and facilitating early and informed conversations regarding patient care goals12. Therefore, developing a precise prediction model shortly after the initiation of dialysis may enhance decision-making and care for patients with AKI-D in ICUs.
Predicting outcomes related to AKI holds significant clinical interest, yet challenging due to the complex and high-dimensional nature of ICU data. Machine learning prediction models have become vital tools, employed to forecast a variety of kidney-related outcomes. These models effectively identify patterns that not only predict the development and progression of AKI but also the potential need for acute dialysis and the long-term consequences following AKI12,13,14,15. However, to explore good machine-learning models for predicting successful dialysis liberation in patients with AKI-D remains an area of research16. To date, no widely accepted tool for early prediction of kidney recovery in patients with AKI-D exist11. Moreover, the absence of patient mortality as a competing risk remains a main concern in studies focusing on RRT liberation12. This study aimed to develop and validate clinically applicable machine-learning models for the early prediction of successful RRT discontinuation at discharge in patients with AKI-D admitted to ICU using routinely collected parameters obtained within the first 3 days after RRT initiation, as well as explore early predictors.
Methods
Study population
Cases of AKI-D were defined as patients who experienced AKI and required RRT, excluding those with pre-existing end-stage renal disease (ESRD). We herein retrospectively identified adult patients who underwent their first dialysis due to AKI during the index ICU admission at Taichung Veterans General Hospital (TCVGH), a tertiary care teaching hospital in Taiwan with five adult ICUs and a computerized electronic medical record (EMR) system, between January 2015 and December 2020. The use of RRT was always reviewed and approved by nephrologists before application. We excluded patients who had been in the ICU for < 48 h, those with a history of chronic dialysis or renal transplant for ESRD, individuals with a baseline creatinine (Cr) ≥ 4 mg/dL, or those with > 20% missing data (Fig. 1A). This cohort was established from two databases: the clinical data warehouse at TCVGH and cause-of-death data from the National Health Insurance Research Database (NHIRD) in Taiwan to determine the date of death of the patients until the end of 2021.
Data collection and feature selection
We identified candidate predictors from routinely collected data within 3 days after initiating dialysis: EMR data and time series of vital signs, which were categorized as follows: ‘Demographics and severity at admission’, variables such as the length of hospitalization before ICU admission and severity scores (the acute physiology and chronic health evaluation II [APACHE-II] score and the sequential organ failure assessment [SOFA]) were considered; ‘Comorbidities at admission’, determined using International Classification of Diseases, 9th Revision, Clinical Modification (ICD-9-CM) and 10th Revision (ICD-10) codes within the past year17; ‘Vital signs’, including vital sign measurements taken within 72 h after dialysis initiation; ‘Laboratory values’, including laboratory results obtained within the initial 72 h of dialysis; ‘Medications’, considering the use of potential nephrotoxic drugs over a 1-week period before the prediction time; ‘Inputs and outputs’, for fluid status data throughout the first 72 h of dialysis; and ‘AKI and dialysis parameters’, including the days from ICU admission to dialysis, AKI defined by either Cr or urine output criteria as outlined in the Kidney Disease: Improving Global Outcomes (KDIGO) clinical practice guidelines for AKI18, and dialysis mode. We assessed whether patients met the AKI diagnosis criteria before initiating dialysis: an increase in serum Cr of more than 0.3 mg/dL from baseline (label_Cr) and/or a decrease in urine output to less than 0.5 mL/kg/hour for more than 6 h (label_Ur). Furthermore, vital signs are recorded at least every two hours and inputs and outputs data every eight hours in the ICU. For instance, within a 72 h feature window, there are at least 36 entries of vital signs. Given the complexity of the vital signs measured over time, we calculated the mean, variance, and trend of these signs to capture the dynamics of patient physiology. Besides, each entry of diet, intravenous fluid, and urine output was standardized to an hourly rate to facilitate the calculation of trends. Data preprocessing details can be found in the supplemental material (Supplemental Table S1, S2, and Figure S1). Table 1 presents the final set of 90 predictors used in our models.
Hyperparameter optimization
For the XGBoost model, we experimented with various hyperparameters, including the number of estimators (500, 1000), learning rates (0.01, 0.02, 0.1), and maximum depths (3, 5, 7, 10). Using the grid search method, we found that the optimal configuration was 500 estimators, a learning rate of 0.01, and a maximum depth of 5, which yielded the best performance results. Similarly, for the Random Forest model, we conducted grid search experiments with different numbers of trees (50, 100, 150, 200) and criteria (Gini, entropy) to determine the most effective parameters. The grid search identified 'Gini' as the best criterion and 100 trees as the optimal number of estimators for achieving superior performance in the Random Forest classifier.
Outcome measure: renal recovery and patient survival
We annotated the outcome as recovery for patients with AKI-D who were both free from dialysis and survived beyond 30 days after discontinuing dialysis, defined as RRT cessation for at least 5 days before hospital discharge (Fig. 1B). To minimize potential misclassification of patients who discontinued dialysis due to withdrawal of care, the deceased date from NHIRD were also gathered to ensure that patients were alive for > 30 days after dialysis discontinuation.
For patients who still required dialysis but were alive upon hospital discharge, their status were checked based on the presence of dialysis-dependent catastrophic illness certificates, which confirmed true nonrecovery; these certificates are issued after a review by at least two nephrologists who carefully examine medical records19. Furthermore, we compared the all-cause mortality on the 90th day and 1 year after hospital discharge.
Data analysis and model development
Patients were divided into two groups: the training group (those admitted between 2015 and 2019) and testing cohort (those admitted in 2020) (Supplemental Figure S1). The prediction time point was set at 72 h post the first dialysis commencement (Fig. 1B). During initial model development, the training dataset was randomly split into training and validation sets in an 80:20 ratio. A fivefold cross-validation was performed to avoid overfitting and considered a robust method for model evaluation prior to temporal testing. Dialysis cessation after AKI-D was predicted using several machine-learning algorithms: extreme gradient boosting (XGBoost), random forest (RF), and logistic regression (LR)20,21,22.
The area under receiver operating characteristic (AUROC) statistics along with sensitivity, specificity, Brier Score, accuracy, precision, recall, and F-1 score was used for the training and testing groups. In the testing datasets, the receiver operating characteristic (ROC) curve, precision-recall curve (PRC), calibration curve, and decision curve were referred to evaluate the predictive machine-learning models’ discrimination, accuracy, and clinical applicability.
Sensitivity analysis
Additional analyses were conducted to validate the robustness and support the clinical utility of our study. First, we made predictions at various time points post-dialysis along with 72 h baseline models. Models were generated using data obtained during the first 24- or 48-h window after first dialysis (observation time window), respectively, to estimate the renal and patient outcomes earlier. The prediction of recovery was assessed immediately after the end of this feature window. Second, to construct more simple models, we applied Least Absolute Shrinkage and Selection Operator (LASSO) regression selection to generate a 24-variable out of 90-variable full model. Third, the number of true negative outcomes was higher than that of true positive outcomes, thus requiring class weighting to address data imbalance. The different thresholds were presented to examine the model prediction performance to determine an optimal threshold.
Feature interpretation and statistical analysis
For the best-performing model (XGBoost), Shapley additive explanation (SHAP) values were employed to evaluate feature importance and their relationship with the outcome in the test set23. Partial dependence plots (PDP) were generated for the most influential variables, and individual feature interactions were assessed with the study outcome24. Additionally, we illustrated SHAP values and Local Interpretable Model-agnostic Explanations (LIME) to provide visual insights into the clinical utility of selected patients. Baseline characteristics between groups with different renal recovery statuses were examined for statistical significance using a threshold of p < 0.05. All analyses were performed using Python (version 3.7.6) and R programming (version 3.4.0).
Ethical approval and consent to participate
The study was approved by the Institutional Review Board of Taichung Veterans General Hospital with a waiver of informed consent since this was a retrospective analysis of anonymous data (TCVGH IRB number: SE20249B and SE21098B).
Results
Clinical characteristics of the study participants
Overall, 26,593 adults were admitted to TCVGH ICUs during the study period. Among the eligible patients, 1,381 patients (61.9% male) experienced AKI-D during their ICU stays (Fig. 1A). The included patients had a median age of 68.0 (interquartile range, 58.0–79.0), with 89.4% requiring mechanical ventilation. Overall, 378 patients (27.3%) achieved renal recovery, whereas 1,003 patients experienced nonrecovery of renal function. Further, 660 patients (47.8%) passed away before dialysis discontinuation or after withdrawing dialysis, whereas 343 patients (24.8%) remained dependent on dialysis upon discharge. Table 2 summarizes baseline and clinical characteristics of patients with AKI-D, stratified by renal recovery status at hospital discharge. Of the recruited patients with AKI-D, 1,135 patients admitted between 2015 and 2019 constituted the training cohort, whereas 246 patients admitted in 2020 comprised the temporal testing cohort. Supplemental Table S3 provides the patient characteristics, stratified based on the train/test split. The train/test cohorts showed comparable age, Charlson comorbidity index (CCI), and patient outcomes.
Performance of the models
When trained and temporal tested with all 90 variables, the XGBoost, RF, and LR predictors achieved AUROC values of 0.85 (95% CI, 0.81–0.88), 0.83 (95% CI, 0.80–0.87), and 0.82 (95% CI, 0.79–0.85), respectively, for predicting renal recovery at 72 h windows in the testing cohort (Fig. 2A). XGBoost exhibited the highest accuracy among the three machine-learning models, and Delong’s test confirmed its significant outperformance of LR (Supplemental Table S4). Table 3 presents the performance statistics for the models with a proposed recovery threshold of 50%. Given this threshold examining the XGBoost model in the testing cohort, only 57% of the patients who were predicted to recover truly did recover (sensitivity), while 93% of the patients who were predicted not to recover indeed did not recover (specificity). Furthermore, the trade-off between sensitivity and specificity was evaluated at predicted renal recovery thresholds in increments of 10% for the XGBoost model (Supplemental Table S5). The F1 score, a good measure of test accuracy, was highest at a cut-off threshold of 0.3, while maintaining good values across the range of 0.2 to 0.5. Supplemental Table S6 demonstrates the performance matrix of a cut-off threshold of 0.3. At this threshold, the sensitivity increases to 73%, while the specificity is 81%. Additionally, The PRC for the models shown in Fig. 2B provided additional clarify on the model’s performance, especially for imbalance datasets, as in our study. Moreover, the temporal testing cohort had good calibration based on the disparity between the predicted and actual outcomes (Fig. 2C). Furthermore, the decision curve analysis illustrated favorable net benefits throughout a wide range of threshold probabilities (Fig. 2D).
Independent predictors for renal recovery
Figure 3A displays top 20 predictors of dialysis liberation and their XGBoost model rankings based on gain. Positive SHAP value (in red) suggests the feature boosts the predicted probability of dialysis liberation, while negative SHAP value (in blue) indicates a higher probability of dialysis dependency or death. The most influential feature was urine volume, followed by CCI, peripheral capillary oxygen saturation (SpO2), lactate, respiratory rate trend, renal disease, BUN, arterial PH, systolic blood pressure (BP) variation, days to ICU admission, white blood cell, and diet (Fig. 3). Figure 4 presents PDPs for those highest-ranking variables. Figure 3B demonstrates SHAP feature importance categorized by the main clinical domains in predicting dialysis liberation, highlighting the significant impact of early vital signs and inputs/outputs domains on the renal recovery probability.
Sensitivity analyses
We conducted several additional analyses (Supplemental Table S7–S9). To address data imbalance (27.3% renal recovery vs. 72.7% nonrecovery), class weights were incorporated into the machine-learning models, resulting in consistent performance. Moreover, LASSO was used to select a reduced set of parameters for model building, retaining the top 24 most influential variables, with a c-statistic of 0.84 in the XGBoost model. Furthermore, dialysis liberation was predicted using different time windows: the original prediction at 72 h post-dialysis (Table 3) and earlier predictions at 48 and 24 h post-dialysis (Supplemental Table S9). These predictions yielded AUROC values of 0.84, 0.82, and 0.82, respectively, in the XGBoost model. Supplemental Figure S2 summarizes the ten most influential features in the model at 24, 48, and 72 h post-dialysis, along with their temporal evolution.
Discussion
We constructed a machine-learning algorithm to forecast the early likelihood of dialysis liberation in critically ill patients with AKI, incorporating both the variability and trends of dynamic parameters from routine data within the first 72 h of dialysis. These models were developed, cross-validated, and temporal tested, thus demonstrating good discrimination in predicting renal recovery at hospital discharge. Furthermore, we applied class weighting to address data imbalance, used LASSO to develop models with few variables, and predicted short time windows (the first 24 or 48 h), all showing good discrimination. These results suggest the potential clinical utility of integration into EMR for clinical decision-support systems. Finally, using SHAP value and PDP, we identified critical features influencing the predictions of the model, with early vital signs and inputs/outputs domains being the vital drivers of the model. Explainable machine learning-based prediction for AKI-D recovery using existing EMR data hold potential for improving risk stratification and gaining insights into patient outcomes.
Nonrecovery of renal function after AKI-D is associated with increased morbidity and mortality and high health care cost25,26. Consistent with the findings of previous epidemiology studies1,26, the present study showed higher short-term (3-month) and long-term (1-year) post-discharge mortality rates for patients dependent on dialysis compared to those liberated from acute dialysis (Supplemental Figure S3). Early prediction of recovery from AKI-D in critically ill patients has significant implications regarding patient-centered care27. Currently, the prediction solely relies on clinical experience. The most commonly used dialysis cessation indicator is the increase in urinary output5. However, urinary output’s accuracy in predicting successful RRT discontinuation remains controversial, with reported AUROCs ranging from 0.63 to 0.91 and varying cut-off values5,28. Additionally, urinary output is typically employed as an indicator of renal recovery later upon RRT discontinuation rather than a marker in the early stages. Traditional functional biomarkers (serum/urine Cr or cystatin C) and novel biomarkers (kidney injury molecule-1, neutrophil gelatinase-associated lipocalin, osteopontin, tissue inhibitor of metalloprotease-2/insulin-like growth factor binding protein-7, proenkephalin A 119–159, etc.) have been explored as predictors of AKI-D recovery5,8,11,27,29. Current biomarkers for renal function recovery after AKI-D, which require additional samples and have limited conclusive evidence, have not been used widely to identify patients with a high probability of renal recovery in the early stages. The urgent need for precision guide to liberate from RRT was also recognized in the recent Acute Dialysis Quality Initiative (ADQI) consensus conference30. The experts emphasized on the integration of big data analysis and single case EMR evaluation to allow personalized RRT for every single individual. To address this gap, there is a clinical unmet need to integrate EMR to assess their predictive value for RRT discontinuation and prognosis in AKI-D.
Machine-learning models developed in critical nephrology can harness the data collected in EMR for important renal outcome predictions13,31,32. As data accumulates, these models will also offer the additional advantage of early prediction or enhanced accuracy. However, validated machine-learning models for predicting acute dialysis discontinuation in critical setting are less studied. To our knowledge, one prior research has employed a machine-learning approach to predict freedom from RRT in patients with AKI-D. Pattharanitima et al. utilized the Medical Information Mart for Intensive Care (MIMIC-III) database to predict RRT-free survival in critically ill patients with AKI requiring continuous renal replacement therapy (CRRT)16. Out of 684 patients, 30% had stopped from RRT successfully. Models using 81 features extracted between hospital admission and CRRT initiation yielded AUROC values ranging 0.43–0.7. In our study including 1,381 AKI-D individuals, we used 90 variables from the initial three days post-dialysis, including all vital signs and input–output records. Thus, variability and trends across multiple time points of these data were incorporated into our models. The prediction models in our study exhibited good performance, with AUROC of 0.77–0.81 in the development cohort and 0.82–0.85 in the temporal testing cohort. Aside from candidate predictors, the differences in model performance between our study and the prior study may also be due to differences in the study populations character, number of participants, and different feature window. The first three days are considered as acute phase of ICU patients, as exemplified by septic shock, where shock reversal often occurs within the first 3 days33. Importantly, providing additional prognostic information after initial intensive treatment period can aid in subsequent medical decisions, including the consideration of clinical trials for high-risk groups, or the potential withdrawal of life-sustaining medical care. Moreover, we trained models at 24 and 48 h in addition to the 72 h model, both maintaining good predictive performance.
As shown in Table 3 and Table S6, a proposed threshold of 0.5 for predicting renal recovery provided good specificity, whereas a threshold of 0.3 enhanced sensitivity. Decision curve analysis revealed the net benefits of using these models in clinical decision-making by considering the trade-offs between sensitivity and specificity at various threshold probabilities. The model’s use would yield more benefit than harm at both threshold of 0.5 and 0.3. Consequently, a lower threshold, such as 0.3, allowed for the identification of a broader subset of patients likely to recover renal function. Meanwhile, a threshold of 0.5 resulted in fewer false positives and would reduce alarm fatigue, a major concern in ICU alarm systems34,35. Therefore, the selection of a threshold in practical applications should be based on whether a healthcare provider requires assistance in accurately identifying patients who can recover or cannot recover after AKI-D, while effectively managing resources.
Using the interpretable machine-learning algorithm, nonsurprisingly, the single most influential variables in renal function recovery after initiating dialysis for AKI was urine output. Patients who successfully liberated from RRT demonstrated significantly higher urine output. According to PDP (Fig. 4), patients with urine volume > 1570 ml over the 72 h period post-dialysis were more likely to achieve dialysis independence at discharge. Figure 4 demonstrates the PDP of top predictors by SHAP value and the cut-off value in favor of renal recovery. In our study, the top 20 variables include previously well-studies factors for renal recovery, including urine volume or BUN5,36,37, along with less-explored variables (ex: enteral diet intake within the first 3 days after dialysis). Moreover, we categorized the top 20 variables identified by the XGBoost model by clinical domains, including comorbidity, vital signs, laboratory data, Inputs/outputs domain, and others. Besides urine volume, most of the early predictors were related to the vital signs domain (Fig. 3B). A general consensus exists that hemodynamic instability caused by excessive fluid removal during dialysis hinders renal recovery38. However, traditional prediction models for renal recovery have often overlooked vital signs due to their complexity and dynamic nature. Bellomo et al. conducted a retrospective study of critically ill patients with shock and found that higher levels of relative hypotension during the first few hours of vasopressor support were significantly associated with an increased risk of adverse kidney-related outcomes39. In line with the current evidence, our data suggests that early vital signs, not only the variance of systolic BP, but also SpO2, trend of respiratory rate, were significantly associated with the renal prognosis of critically ill patients. Additionally, the use of LASSO model with more limited variables and the incorporation of routinely collected laboratory data offer a practical means of rapid integration into EMR (Supplemental Table S8). An illustrate the interpretability of the models and the evolving of the key features over time using two separate individuals is presented in Supplemental Figure S4. Altogether, explainable machine-learning models can be deconvoluted to unveil new insights of how ICU patient features at the early stage interact with patient future events.
Strengths of our study include its size, comprising 1,381 patients with AKI-D among 26,593 ICU admissions. We also have complete data on vital signs and inputs/outputs with very low missing rates (< 1%). Furthermore, we linked the NHIRD cause-of-death data to mitigate withdrawal bias risk. This is especially important in ICU studies, as 40%–60% of critically ill patients with AKI-D have their treatment discontinued due to life support withdrawal or death4.
Our study has several limitations. First, recovery status was determined at hospital discharge; however, we recognized that dialysis liberation may proceed further. Nevertheless, the median hospital stay of critically ill patients with AKI-D was long (25.0 days for the entire cohort and 30.5 days for the recovery group), with dialysis-dependent catastrophic illness certificates verified before discharge by nephrologist among AKI-D nonrecovers. Thus, the kidney prognosis is clinically relevant. Second, our models made one-time early predictions of renal recovery on the basis of data obtained within 3 days of dialysis initiation. Events after that may drive the patient outcome away from the predictions. Though, we developed additional models at various time horizons (1 and 2 days), a continuously updating prediction is more appropriate for such cases. Third, limitations of our retrospective database include lack of other predictors of interest, including the degree of urine proteinuria, timed creatinine clearance, and novel kidney biomarkers, which may influence renal and patient recovery. Last, we used temporal testing, which is considered as an in-between validation of internal and external validation40. Although the recovery and mortality rates in this cohort was comparable to those reported in the literature4,26, the results should be further validated in other settings.
Conlusions
To conclude, we developed, validated, and temporal tested applicable machine-learning prediction models using routinely collected clinical and laboratory data during the initial 72 h of dialysis to identify successful RRT discontinuation on hospital discharge in critically ill patients with AKI-D. Early vital signs and urine output are vital factors influencing the model. The interpretable machine learning-based algorithm can be used to facilitate patient-centered care in case of a complex scenario and vulnerable populations such as those with AKI-D. Models should be further validated and implemented in critically ill patients.
Data availability
The data is available from the corresponding author on reasonable request.
Abbreviations
- ADQI:
-
Acute Dialysis Quality Initiative
- AKI:
-
Acute kidney injury
- AKI-D:
-
AKI requiring dialysis
- APACHE-II:
-
Acute physiology and chronic health evaluation II
- AUPRC:
-
Area under precision-recall curve
- AUROC:
-
Area under receiver operating characteristic
- BP:
-
Blood pressure
- CCI:
-
Charlson comorbidity index
- Cr:
-
Creatinine
- CRRT:
-
Continuous renal replacement therapy
- EMR:
-
Electronic medical record
- ESRD:
-
End-stage renal disease
- ICU:
-
Intensive care unit
- KDIGO:
-
The Kidney Disease: Improving Global Outcomes
- LASSO:
-
Least Absolute Shrinkage and Selection Operator
- LIME:
-
Local Interpretable Model-agnostic Explanations
- LR:
-
Logistic regression
- MIMIC-III:
-
Medical Information Mart for Intensive Care
- NHIRD:
-
National Health Insurance Research Database
- PDP:
-
Partial dependence plots
- PRC:
-
Precision-recall curve
- RF:
-
Random forest
- RRT:
-
Renal replacement therapy
- SHAP:
-
Shapley additive explanation
- SOFA:
-
Sequential organ failure assessment
- SpO2:
-
Saturation of peripheral oxygen
- TCVGH:
-
Taichung Veterans General Hospital
- XGBoost:
-
Extreme gradient boosting
References
Hoste, E. A. et al. Epidemiology of acute kidney injury in critically ill patients: the multinational AKI-EPI study. Intensive Care Med. 41, 1411–1423. https://doi.org/10.1007/s00134-015-3934-7 (2015).
Shiao, C. C. et al. Association between regional economic status and renal recovery of dialysis-requiring acute kidney injury among critically ill patients. Sci. Rep. 10, 14573. https://doi.org/10.1038/s41598-020-71540-7 (2020).
Rewa, O. & Bagshaw, S. M. Acute kidney injury-epidemiology, outcomes and economics. Nat. Rev. Nephrol. 10, 193–207. https://doi.org/10.1038/nrneph.2013.282 (2014).
Klouche, K., Gibney, R. T. N. & Forni, L. G. Can this patient be safely weaned from RRT?. Intens. Care Med. 44, 639–642. https://doi.org/10.1007/s00134-017-4948-0 (2018).
Katulka, R. J. et al. Determining the optimal time for liberation from renal replacement therapy in critically ill patients: A systematic review and meta-analysis (DOnE RRT). Crit. Care 24, 50. https://doi.org/10.1186/s13054-020-2751-8 (2020).
Hsu, C. M. et al. Kidney recovery and death in critically ill patients with COVID-19-associated acute kidney injury treated with dialysis: The STOP-COVID cohort study. Am. J. Kidney Dis. 79, 404–416. https://doi.org/10.1053/j.ajkd.2021.11.004 (2022).
Aniort, J. et al. Daily urinary urea excretion to guide intermittent hemodialysis weaning in critically ill patients. Crit. Care 20, 43. https://doi.org/10.1186/s13054-016-1225-5 (2016).
Viallet, N. et al. Daily urinary creatinine predicts the weaning of renal replacement therapy in ICU acute kidney injury patients. Ann. Intens. Care 6, 71. https://doi.org/10.1186/s13613-016-0176-y (2016).
Raurich, J. M. et al. Successful weaning from continuous renal replacement therapy Associated risk factors. J. Crit. Care 45, 144–148. https://doi.org/10.1016/j.jcrc.2018.02.009 (2018).
Yang, T. et al. Biomarkers upon discontinuation of renal replacement therapy predict 60-day survival and renal recovery in critically ill patients with acute kidney injury. Hemodial. Int. 22, 56–65. https://doi.org/10.1111/hdi.12532 (2018).
von Groote, T. et al. Proenkephalin A 119–159 predicts early and successful liberation from renal replacement therapy in critically ill patients with acute kidney injury: a post hoc analysis of the ELAIN trial. Crit. Care 26, 333. https://doi.org/10.1186/s13054-022-04217-4 (2022).
Vaid, A. et al. Predictive Approaches for Acute Dialysis Requirement and Death in COVID-19. Clin J Am Soc Nephrol 16, 1158–1168. https://doi.org/10.2215/CJN.17311120 (2021).
Tomasev, N. et al. A clinically applicable approach to continuous prediction of future acute kidney injury. Nature 572, 116–119. https://doi.org/10.1038/s41586-019-1390-1 (2019).
Wei, C., Zhang, L., Feng, Y., Ma, A. & Kang, Y. Machine learning model for predicting acute kidney injury progression in critically ill patients. BMC Med. Inform. Decis. Mak. 22, 17. https://doi.org/10.1186/s12911-021-01740-2 (2022).
Nateghi Haredasht, F., Viaene, L., Pottel, H., De Corte, W. & Vens, C. Predicting outcomes of acute kidney injury in critically ill patients using machine learning. Sci. Rep. 13, 9864. https://doi.org/10.1038/s41598-023-36782-1 (2023).
Pattharanitima, P. et al. Comparison of approaches for prediction of renal replacement therapy-free survival in patients with acute kidney injury. Blood Purif. 1, 1–7. https://doi.org/10.1159/000513700 (2021).
Quan, H. et al. Coding algorithms for defining comorbidities in ICD-9-CM and ICD-10 administrative data. Med. Care 43, 1130–1139. https://doi.org/10.1097/01.mlr.0000182534.19832.83 (2005).
Khwaja, A. et al. KDIGO clinical practice guidelines for acute kidney injury. Kidney Int. Suppl. 2, 19–36. https://doi.org/10.1038/kisup.2011.32 (2012).
Wu, H. H. et al. Family aggregation and heritability of ESRD in Taiwan: A population-based study. Am. J. Kidney Dis. 70, 619–626. https://doi.org/10.1053/j.ajkd.2017.05.007 (2017).
Chen, T. & Guestrin, C. in Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining 785–794 (2016).
Dreiseitl, S. & Ohno-Machado, L. Logistic regression and artificial neural network classification models: A methodology review. J Biomed Inform 35, 352–359. https://doi.org/10.1016/s1532-0464(03)00034-0 (2002).
Breiman, L. Random forests. Mach. Learn. 45, 5–32. https://doi.org/10.1023/A:1010933404324 (2001).
Štrumbelj, E. & Kononenko, I. Explaining prediction models and individual predictions with feature contributions. Knowl. Inf. Syst. 41, 647–665 (2014).
Nielsen, A. B. et al. Survival prediction in intensive-care units based on aggregation of long-term disease history and acute physiology: a retrospective study of the Danish National Patient Registry and electronic patient records. Lancet Digit. Health 1, e78–e89. https://doi.org/10.1016/s2589-7500(19)30024-x (2019).
Noble, R. A., Lucas, B. J. & Selby, N. M. Long-term outcomes in patients with acute kidney injury. Clin. J. Am. Soc. Nephrol. 15, 423–429. https://doi.org/10.2215/CJN.10410919 (2020).
Shah, S. et al. Mortality and recovery associated with kidney failure due to acute kidney injury. Clin. J. Am. Soc. Nephrol. 15, 995–1006. https://doi.org/10.2215/CJN.11200919 (2020).
Vijayan, A. et al. Recovery after critical illness and acute kidney injury. Clin J Am Soc Nephrol 16, 1601–1609. https://doi.org/10.2215/CJN.19601220 (2021).
Mendu, M. L. et al. A decision-making algorithm for initiation and discontinuation of RRT in severe AKI. Clin. J. Am. Soc. Nephrol. 12, 228–236. https://doi.org/10.2215/CJN.07170716 (2017).
von Groote, T. et al. Evaluation of Proenkephalin A 119–159 for liberation from renal replacement therapy: An external, multicenter pilot study in critically ill patients with acute kidney injury. Crit. Care 27, 276. https://doi.org/10.1186/s13054-023-04556-w (2023).
Zhao, X. et al. Predicting renal function recovery and short-term reversibility among acute kidney injury patients in the ICU: Comparison of machine learning methods and conventional regression. Ren. Fail 44, 1326–1337. https://doi.org/10.1080/0886022X.2022.2107542 (2022).
Kang, M. W. et al. Machine learning algorithm to predict mortality in patients undergoing continuous renal replacement therapy. Crit. Care 24, 42. https://doi.org/10.1186/s13054-020-2752-7 (2020).
Neyra, J. A. et al. Prediction of mortality and major adverse kidney events in critically ill patients with acute kidney injury. Am. J. Kidney Dis. 81, 36–47. https://doi.org/10.1053/j.ajkd.2022.06.004 (2023).
Sakr, Y. et al. Higher fluid balance increases the risk of death from sepsis: Results from a large international audit*. Crit. Care Med. 45, 386–394. https://doi.org/10.1097/ccm.0000000000002189 (2017).
Escobar, G. J. et al. Automated identification of adults at risk for in-hospital clinical deterioration. N. Engl. J. Med. 383, 1951–1960. https://doi.org/10.1056/NEJMsa2001090 (2020).
Hyland, S. L. et al. Early prediction of circulatory failure in the intensive care unit using machine learning. Nat. Med. 26, 364–373. https://doi.org/10.1038/s41591-020-0789-4 (2020).
Uchino, S. et al. Discontinuation of continuous renal replacement therapy: A post hoc analysis of a prospective multicenter observational study. Crit. Care Med. 37, 2576–2582. https://doi.org/10.1097/CCM.0b013e3181a38241 (2009).
Lee, B. J. et al. Pre-admission proteinuria impacts risk of non-recovery after dialysis-requiring acute kidney injury. Kidney Int. 93, 968–976. https://doi.org/10.1016/j.kint.2017.10.017 (2018).
Chawla, L. S. et al. Acute kidney disease and renal recovery: consensus report of the Acute Disease Quality Initiative (ADQI) 16 Workgroup. Nat. Rev. Nephrol. 13, 241–257. https://doi.org/10.1038/nrneph.2017.2 (2017).
Panwar, R. et al. Relative hypotension and adverse kidney-related outcomes among critically ill patients with shock: A multicenter, prospective cohort study. Am. J. Respir. Crit. Care Med. 202, 1407–1418. https://doi.org/10.1164/rccm.201912-2316OC (2020).
Ohbe, H., Goto, T., Nakamura, K., Matsui, H. & Yasunaga, H. Development and validation of early prediction models for new-onset functional impairment at hospital discharge of ICU admission. Intens. Care Med. https://doi.org/10.1007/s00134-022-06688-z (2022).
Funding
This study was supported by Taichung Veterans General Hospital (TCVGH-1124402B, TCVGH-1134402B) and Ministry of Science and Technology Taiwan (NSTC 112–2321-B-075A-001). The funders had no role in the study design, data collection and analysis, decision to publish, or preparation of the manuscript.
Author information
Authors and Affiliations
Contributions
Study concept and design: T.J.W. and K.C.P. Study coordination: C.L.W., C.H.C., and W.C.C. Acquisition of data: C.L.W. and K.C.P. Statistical analysis: M.S.W. and K.C.P. Interpretation of data: T.J.W., C.T.H., Y.C.H., and K.C.P. Drafting the manuscript: T.J.W. and K.C.P. All authors reviewed and approved the final manuscript.
Corresponding author
Ethics declarations
Competing interests
The authors declare no competing interests.
Additional information
Publisher's note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Supplementary Information
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/.
About this article
Cite this article
Wang, TJ., Huang, CT., Wu, CL. et al. Predictive approach for liberation from acute dialysis in ICU patients using interpretable machine learning. Sci Rep 14, 13142 (2024). https://doi.org/10.1038/s41598-024-63992-y
Received:
Accepted:
Published:
DOI: https://doi.org/10.1038/s41598-024-63992-y
- Springer Nature Limited