Skip to main content
Frontiers in Medicine logoLink to Frontiers in Medicine
. 2021 Dec 10;8:792974. doi: 10.3389/fmed.2021.792974

Application of Machine Learning to Predict Acute Kidney Disease in Patients With Sepsis Associated Acute Kidney Injury

Jiawei He 1,, Jin Lin 1,, Meili Duan 1,*
PMCID: PMC8703139  PMID: 34957162

Abstract

Background: Sepsis-associated acute kidney injury (AKI) is frequent in patients admitted to intensive care units (ICU) and may contribute to adverse short-term and long-term outcomes. Acute kidney disease (AKD) reflects the adverse events developing after AKI. We aimed to develop and validate machine learning models to predict the occurrence of AKD in patients with sepsis-associated AKI.

Methods: Using clinical data from patients with sepsis in the ICU at Beijing Friendship Hospital (BFH), we studied whether the following three machine learning models could predict the occurrence of AKD using demographic, laboratory, and other related variables: Recurrent Neural Network-Long Short-Term Memory (RNN-LSTM), decision trees, and logistic regression. In addition, we externally validated the results in the Medical Information Mart for Intensive Care III (MIMIC III) database. The outcome was the diagnosis of AKD when defined as AKI prolonged for 7–90 days according to Acute Disease Quality Initiative-16.

Results: In this study, 209 patients from BFH were included, with 55.5% of them diagnosed as having AKD. Furthermore, 509 patients were included from the MIMIC III database, of which 46.4% were diagnosed as having AKD. Applying machine learning could successfully achieve very high accuracy (RNN-LSTM AUROC = 1; decision trees AUROC = 0.954; logistic regression AUROC = 0.728), with RNN-LSTM showing the best results. Further analyses revealed that the change of non-renal Sequential Organ Failure Assessment (SOFA) score between the 1st day and 3rd day (Δnon-renal SOFA) is instrumental in predicting the occurrence of AKD.

Conclusion: Our results showed that machine learning, particularly RNN-LSTM, can accurately predict AKD occurrence. In addition, Δ SOFAnon−renal plays an important role in predicting the occurrence of AKD.

Keywords: intensive care unit, sepsis, acute kidney injury, acute kidney disease, machine learning

Introduction

The prevalence of acute kidney injury (AKI) in patients admitted to intensive care units (ICU) is approximately 50%. Nearly half of all AKI cases are present with sepsis, which may further worsen the prognosis (1, 2). Previous studies have reported the mortality rate of ICU patients with septic AKI as 30–45%, with the survivors still associated with the increased risk of chronic kidney disease (CKD) and cardiovascular events (3).

Increased severity and higher duration of AKI are associated with poor prognosis. In line with several previous results, Kellum et al. reported poorer clinical outcomes in patients with AKI lasting longer than 7 days than in patients who had renal function recovered within 7 days (4). Similar results have been previously reported in other studies (5, 6). Furthermore, in patients who developed sepsis persistent AKI beyond 7 days was associated with adverse clinical outcomes (5, 6). Hence, Acute Disease Quality Initiative-16 (ADQI-16) workshop suggested defining acute kidney disease (AKD) as impaired kidney function lasting 7–90 days after AKI (7). Unlike AKI patients, whose renal function typically recovers within 7 days, AKD patients suffer from persistent renal impairment and often have poor clinical outcomes (8).

Recent studies have utilized machine learning techniques for predicting AKI. Using machine learning techniques such as logistic regression and extreme gradient boosting (XGBoost), Zhang et al. identified some important clinical factors associated with AKI such as age, urinary creatinine concentration, maximum blood urea nitrogen concentration, and albumin (9). Zimmerman et al. showed that comprehensive demographics and physiologic features can accurately predict max serum creatinine level during day 2 and day 3 and also predict new AKI onset by cross-validation on linear regression and multiple machine learning models (10). However, AKD prediction has not been reported.

The AKD phase is a time window for potentially initiating key interventions to alter the natural history of kidney disease (7), and thus, the early identification of patients at high risk of developing AKD is important. Previous studies have shown that age, hypertension, diabetes mellitus, the history of CKD, the severity of AKI, and the use of mechanical ventilators were associated with the onset of AKD (1117), however, machine learning methods have been seldom used to predict the occurrence of AKD. This study was aimed at using longitudinal data to predict the occurrence of AKD.

Materials and Methods

Data Source and Participants

Patients were recruited from the intensive care unit of Beijing Friendship Hospital (BFH), between January 1, 2015 and December 21, 2020. We obtained electronic healthcare data from Medical Information Mart for Intensive Care III (MIMIC III) (18). The inclusion criteria were as follows: (1) age ≥ 18 years old; (2) AKI caused by sepsis. The exclusion criteria were as follows: (1) AKI duration <48 h; (2) length of survival time <7 days; (3) CKD stage 5 or end-stage kidney disease defined as estimated glomerular filtration rate <15 ml/min/1.73 m2; (4) patients with missing important data (e.g., data on demographics and variables for calculating traditional severity scores). The study was reported according to the recommendations of the Transparent Reporting of a multivariable prediction model for Individual Prognosis or Diagnosis (TRIPOD) statement (19).

Data Extraction

We extracted the following data from BFH and the MIMIC III records upon admission to ICU (day 1): (1) demographic information; (2) ICU details, including vitals, laboratory data, mechanical ventilation requirement, and exposure to nephrotoxic drugs; (3) severity of illness was measured using Simplified Acute Physiology Score II (SAPS II), Acute Physiological Score III (APS III), and non-renal Sequential Organ Failure Assessment (SOFA) score. The data on non-renal SOFA, creatinine, and urine output were recorded daily until day 3. Delta non-renal SOFA, delta creatinine, and delta urine output was the difference between the value at day 3 and the admission value.

Outcomes and Definitions

The occurrence of AKD was the primary outcome. AKD was defined as the presentation of at least KDIGO Stage 1 criteria for >7 days after an AKI-initiating event, which agrees with the diagnostic criteria proposed by ADQI-16 in 2017 (7). The definition of sepsis was based on the diagnostic criteria of the Third International Consensus Definitions for Sepsis and Septic Shock (Sepsis-3), including a suspected infection and a SOFA score of ≥2 (20). The Kidney Disease: Improving Global Outcomes (KDIGO) classification according to both serum creatinine (SCr) and urine output (UO) criteria were used to define AKI (21). CKD was defined according to the Clinical Practice Guideline for the Evaluation and Management of Chronic Kidney Disease (22).

Sample Size

The sample size was defined as having at least 10 outcome events per variable per estimated parameter according to a previous study (23). Our sample and the number of AKD approached that determined by the calculated result.

Statistical Analysis

Values were presented as total numbers (percentages) for categorical variables and the means ± SDs or medians (interquartile ranges) for continuous variables. Comparisons were made using the Student's t-test or rank-sum test for continuous variables, and the Chi-square test or Fisher's exact test for categorical variables, as appropriate. All statistical tests were two-sided, and P-values of <0.05 were considered statistically significant.

Model Development and Validation

The included patients from BFH and MIMIC III comprised the training dataset and the validation dataset, respectively. We selected three models for comparison: Recurrent Neural Network-Long Short-Term Memory (RNN-LSTM), decision tree, and logistic regression. The discrimination performance of these models in the training dataset and the validation dataset was evaluated by area under the receiver operating characteristic (AUROC).

Recurrent Neural Network-Long Short-Term Memory

The RNN has been widely used to handle the longitudinal variables, LSTM is one type of RNN (24, 25). It can effectively process a large amount of sequential data. It comprises several modules, which can store the processed data from the previous stage. Unlike ordinary RNN, classic LSTM comprises several modules called cells. Data can be transferred from the previous cell to the next cell, including input gate, forget gate, and output gate. All data are added to the input gate, and the output gate displays the final data result. Unlike ordinary RNN which can have only one memory stacking method, LSTM can control the transmission state through the gating state, remember important information and forget unimportant information. The forget gate can enhance the ability of LSTM to process data and avoid the problem of data dependence.

Decision Tree

Decision tree/random forest can predict the classification (AKD or non-AKD) from the data, which can display the decision result more clearly (26). We can use the decision tree to interpret the prediction results. The process from the root to the leaf of the tree shows the prediction classification, according to the algorithm of the decision tree. Each step of the decision tree involves checking a piece of data. If the predictor satisfied a certain condition, it would follow the upper branch to indicate type 0, predicting that AKD will occur. Otherwise, it would follow the lower branch to indicate type 1, predicting that AKD will not occur. The decision trees were trained to create a model that could factor in multiple input variables and predict the value of the target variable. The division of the tree continues until the node contains the minimum number of training examples or reaches the maximum tree depth. The complexity parameter is used to indicate the prediction performance, which depends on how many classes are mixed in the two groups generated by the decision tree (27). We choose the number of leaves when the complexity parameter is the lowest to minimize the chance of making errors in the decision tree.

Logistic Regression

In the training dataset, we used the Least Absolute Shrinkage and Selection Operator (LASSO) method to select the most useful predictive variables (28). Continuous variables were made into dichotomous variables and were entered into a logistic regression with other variables. The nomogram predicting the occurrence of AKD was established using the LASSO method for the selected variables. The performance of the nomogram was evaluated by calibration curves. The calibration evaluation uses a calibration chart to show the relationship between the observed frequency and the predicted probability. The nomogram was verified in the validation dataset to evaluate the stability of the nomogram. In addition, decision curve analysis (DCA) was used to evaluate the clinical utility of the final nomogram (29). The net benefit is calculated by subtracting the proportion of false positives from true positives (30).

Moreover, the discrimination of three machine learning algorithms in predicting the occurrence of AKD patients was compared using Delong's method. The discrimination was validated externally by the AUROC in the MIMIC III database.

We performed all statistical analyzes using R software version 4.0.5 (R Foundation for Statistical Computing).

Results

Participants

As shown in Figure 1, a total of 5,629 patients were screened during the study period in the BFH. The initial research identified 23,620 ICU admissions from the MIMIC III database. In addition, 209 and 509 patients were assigned to the training dataset and validation dataset, respectively. Twenty-eight predictors were extracted from the database and included in the model. The occurrence of AKD rate was 55.5% (116 patients with AKD) in the training dataset and 46.4% (236 patients with AKD) in the validation dataset. A comparison of baseline characteristics between the AKD group and non-AKD group in BFH and MIMIC-III cohorts are recorded in Table 1. AKD patients were older and had higher Charlson score and delta non-renal SOFA; higher creatinine at day 3 and AKI stage; more medical history of hypertension, diabetes mellitus, and CKD; more application of diuretics and renal toxic drugs in the training dataset (p < 0.05), while they had a lower delta creatinine, urine output at day 3, and delta urine output (p < 0.05). Furthermore, comorbidities of CKD, higher AKI stage, and lower delta creatinine also showed similar results between AKD patients and non-AKD patients in the validation dataset (p < 0.05). Our study was reported according to the guidelines of the TRIPOD statement.

Figure 1.

Figure 1

Flow chart of patient selection.

Table 1.

Baseline characteristics of the Beijing Friendship Hospital (BFH) and Medical Information Mart for Intensive Care III (MIMIC III) cohorts.

BFH cohort MIMIC III cohort
Non-AKD (n = 93) AKD (n = 116) P-value Non-AKD (n = 273) AKD (n = 236) P-value
Age, mean (SD) 54.7 (20.7) 64.5 (14.7) <0.001 64.3 (16.5) 62.6 (18.1) 0.252
Male, (%) 58 (62.4) 74 (63.8) 0.832 160 (58.6) 117 (49.6) 0.041
BMI, kg/m2, median [Q1, Q3] 26.6 (22.5, 30.4) 26.3 (22.4, 29.4) 0.837 27.3 (23.5, 32.2) 27.3 (23.3, 31.2) 0.782
Heart failure, n (%) 19 (20.4) 30 (25.9) 0.357 66 (24.2) 50 (21.2) 0.423
Hypertension, n (%) 45 (48.4) 74 (63.8) 0.025 21 (7.7) 30 (12.7) 0.060
Chronic obstructive pulmonary disease, n (%) 12 (12.9) 19 (16.4) 0.482 62 (22.7) 61 (25.8) 0.410
Chronic liver disease, n (%) 3 (3.2) 6 (5.2) 0.491 30 (11.0) 27 (11.4) 0.872
Diabetes mellitus, n (%) 35 (37.6) 66 (56.9) 0.006 76 (27.8) 52 (22.0) 0.132
Chronic kidney disease, n (%) 37 (39.8) 26 (22.4) 0.007 36 (13.2) 18 (7.6) 0.042
Charlson score, median [Q1, Q3] 2 (1, 3) 2 (1, 4) 0.001 2 (1, 3) 2 (1, 3) 0.729
Emergency department, n (%) 59 (63.4) 75 (64.7) 0.856 33 (12.1) 36 (15.3) 0.298
Surgery, n (%) 25 (26.9) 20 (17.2) 0.092 99 (36.3) 56 (23.7) 0.002
APS III, median [Q1, Q3] 45 (32, 62) 44.5 (32, 63) 0.779 40 (29, 55) 36.5 (26, 48) 0.035
SAPS II, median [Q1, Q3] 35 (25, 46) 35 (27, 44) 0.779 31 (23, 44) 31.5 (23, 41) 0.461
Non-renal SOFA at day 1, median [Q1, Q3] 3 (1, 6) 3 (1, 6) 0.310 3 (1, 5) 2 (1, 4) 0.044
Non-renal SOFA at day 3, median [Q1, Q3] 3 (1, 6) 3 (1, 6) 0.375 2 (1, 4) 2 (1, 4) 0.226
Delta non-renal SOFA, median [Q1, Q3] 0 (0, 0) 0 (0, 1) <0.001 0 (0, 0) 0 (0, 0) 0.478
AKI stage, n (%) <0.001 0.008
1 13 (14.0) 3 (2.6) 132 (48.4) 145 (61.4)
2 33 (35.5) 20 (17.2) 91 (33.3) 55 (23.3)
3 47 (50.5) 93 (80.2) 50 (18.3) 36 (15.3)
Baseline creatinine, mg/dl, median [Q1, Q3] 0.7 (0.5,1.0) 0.8 (0.5, 1.1) 0.880 0.6 (0.5, 0.9) 0.60 (0.4, 0.9) 0.700
Creatinine at day 1, mg/dl, median [Q1, Q3] 1.3 (0.9, 2.2) 1.4 (0.9, 2.5) 0.857 1.1 (0.9, 1.5) 1.0 (0.8, 1.4) 0.050
Creatinine at day 3, mg/dl, median [Q1, Q3] 1.1 (0.9, 1.6) 1.2 (0.8, 2.0) 0.014 1.0 (0.8, 1.3) 0.9 (0.7, 1.4) 0.159
Delta creatinine, mg/dl, median [Q1, Q3] −0.1 (−0.7, 0.0) −0.1 (−0.40, 0.0) 0.001 −0.1 (−0.20, 0.0) 0.0 (−0.1, 0.0) <0.001
Urine output at day1, ml/kg/h, median [Q1, Q3] 0.9 (0.4, 2.9) 0.9 (0.4, 2.8) 0.457 1.9 (0.7, 3.9) 1.9 (0.8, 4.0) 0.324
Urine output at day3, ml/kg/h, median [Q1, Q3] 1.1 (0.7, 1.7) 0.9 (0.3, 1.5) <0.001 1.0 (0.6, 1.6) 1.1 (0.7, 2.1) 0.438
Delta urine output, ml/kg/h, median [Q1, Q3] 1.1 (0.7, 1.7) 0.9 (0.3, 1.5) <0.001 −0.3 (−1.7, 0.0) −0.2 (−1.0, 0.0) 0.219
Diuretics, n (%) 22 (23.7) 94 (81.0) <0.001 126 (46.2) 91 (38.6) 0.084
Mechanical ventilation, n (%) 61 (65.6) 74 (63.8) 0.787 137 (50.2) 103 (43.6) 0.141
Renal toxic drugs, n (%) 46 (49.5) 89 (76.7) <0.001 115 (42.1) 84 (35.6) 0.132

MIMIC III, Medical Information Mart for Intensive Care III; AKD, acute kidney disease; BMI, body mass index; APS III, Acute Physiological Score III; SPAS II, Simplified Acute Physiology Score II; SOFA, Sequential Organ Failure Assessment.

Model Development

In RNN-LSTM, as the validation loss was decreasing over time, the accuracy of the model increased (Figure 2). The LSTM has been trained up to 200 epochs to obtain the smallest loss and the greatest accuracy. Throughout the training process of 200 epochs, our training loss and validation loss had decreased and accuracy increased gradually, respectively. At the 200th epoch, the training loss and the validation loss are approximately the lowest, where the training accuracy and the validation accuracy reach 97.96 and 97.66%, respectively. We found that the training graph and the validation graph are quite similar. Thus, it can be concluded that the model is quite accurate. It is neither overfitting nor underfitting. The significance of the predictors in the RNN-LSTM model is presented in Figure 3. The feature variable importance showed that Δnon-renal SOFA had an important role. Other variables, such as creatinine on day 3, hypertension, and diuretics, also showed marked effects. As the decision trees algorithm has nodes that represent variables and conjunction that connects the nodes, the performance of this algorithm mainly depends on the number of nodes and tree size (31). We explored different ways to find the optimal performance of the decision trees algorithm by adjusting the number of nodes (Figure 4). We found that the optimal number of nodes that could minimize the decision trees' misclassification error rate was 10, where the complexity parameter was 0.018. Using this number of nodes, the decision trees' structure was pruned. Among these variables, Δnon-renal SOFA had a crucial role in the prediction of the occurrence of AKD. If Δnon-renal SOFA < 1, delta creatinine played an important role in the next decision. If Δnon-renal SOFA > 1, whether used diuretics or not was important. If Δnon-renal SOFA > 1 and patients did not receive diuretics, he/she was more likely to be diagnosed with AKD soon (Figure 5).

Figure 2.

Figure 2

Loss (A) and accuracy (B) vs. epoch graph (up to 200 epochs).

Figure 3.

Figure 3

Significance of the predictors in the Recurrent Neural Network-Long Short-Term Memory (RNN-LSTM) model. All 28 important features regarding the development of the final predictive model are depicted.

Figure 4.

Figure 4

Contribution of 28 variables in predicting the occurrence of patients with sepsis-associated AKD.

Figure 5.

Figure 5

Optimized decision tree for the classification of acute kidney disease (AKD)/non-AKD of patients.

In logistic regression, twenty-eight variables were included in the LASSO regression analysis and narrowed down to 10 features in the LASSO regression model (Figure 6). Next, a model integrating age, combined with hypertension, diabetes mellitus, CKD, delta non-renal SOFA, AKI stage, delta creatinine, delta urine output, diuretics, and nephrotoxic drugs was established using the training dataset. Based on this model, a nomogram was plotted to predict the probability of the occurrence of AKD patients (Figure 7). The calibration curve was described using the bootstrap method for both, the training and validation datasets (Figure 8A). The apparent line and a bias-corrected line only slightly deviated from the ideal line, indicating a good agreement between the prediction and reality. The DCA curve was plotted to perform a clinical application of this nomogram. In the training dataset, clinical intervention guided by this nomogram provided a greater net benefit when the threshold probability was within 0.01 and 0.71 (Figure 8B).

Figure 6.

Figure 6

Clinical feature selection using the Least Absolute Shrinkage and Selection Operator (LASSO) logistic regression. (A) Optimal parameter (lambda) selection in the LASSO logistic regression. The black vertical lines were drawn at the optimal values by using the minimum criteria and the one SE of the minimum criteria (the 1-SE criteria). (B) LASSO coefficient profiles of the 28 features. A coefficient profile plot was produced against the log (lambda) sequence.

Figure 7.

Figure 7

Nomogram developed based on the training dataset with the incorporation of age, combined with hypertension, diabetes mellitus, chronic kidney disease (CKD), delta non-renal Sequential Organ Failure Assessment (SOFA), acute kidney injury (AKI) stage, delta creatinine, delta urine output, diuretics, and renal toxic drugs.

Figure 8.

Figure 8

Calibration curves (A) and decision curve analysis (B) for nomogram.

Model Performance

In the training dataset, we evaluated the discrimination of three models. RNN-LSTM was well-discriminated in the external validation dataset (AUROC: 1), which was greater than decision trees and logistic regression (AUROC: decision trees 0.954, logistic regression 0.728; Figure 9A). In the validation dataset, among RNN-LSTM, decision trees, and logistic regression algorithms, the RNN-LSTM algorithm showed the highest performance with an AUROC of 1.000, followed by the decision trees with an AUROC of 0.872. Logistic regression had the least predictive accuracy, with an AUROC of 0.717. All machine learning models, except the logistic regression model, showed good discrimination ability in the training and validation datasets. In the training and validation datasets, the RNN-LSTM algorithm achieved the best performance among the four models (Figure 9B).

Figure 9.

Figure 9

The area under the receiver operating characteristic (AUROC) curve of the RNN-LSTM, decision trees, and logistic regression. (A) Training dataset; (B) Validation dataset.

Discussion

In the present study, a total of 209 patients from BFH were included, with 55.5% of them diagnosed as having AKD. Using the data from BFH and MIMIC III records, we successfully developed and validated machine learning models to predict the occurrence of AKD in patients with AKI.

Since the diagnostic criteria for AKD were released in ADQI-16, several investigations have been undertaken on the epidemiology of AKD. Kellum et al. reported the incidence rate of AKD as 36.2% in ICU patients (4). Federspiel et al. showed the incidence rate of sepsis-associated AKD as 32.4% in critically ill patients (5). Peerapornratana et al. reported the incidence rate of sepsis-associated AKD in patients dying within 7 days was 33.6% (161/479) from the first day of being diagnosed with AKI (11). Our studies showed an AKD diagnosis rate of 55.5%. This higher rate could be attributed to the exclusion of patients with an AKI duration of <3 days.

Ostermann et al. suggested that nephrotoxic drugs increase the risk of renal function impairment (32). Drugs are among the main causes of AKI. Its pathogenesis included acute tubular necrosis, tubular obstruction by crystals or casts, and interstitial nephritis induced by drugs and their metabolites (33). Our study shows that nephrotoxic drugs increase the incidence of AKD, possibly because they deteriorate renal function.

There has been a controversy about whether the application of diuretics can improve renal function in recent years. A Phase II Randomized Blinded Controlled Trial of the Effect of furoSemide in Critically Ill Patients With eARly Acute Kidney Injury (SPARK-RCT) study showed that diuretics improved neither the recovery rate of AKI nor the prognosis of the patients (34). The study of Zhao et al. reported that administering diuretics improved renal function in patients on the MIMIC III database (35). Our research shows that the use of diuretics may be related to the low incidence of AKD. The effective use of diuretics can reflect the recovery of the patients' renal function, but it may not change it. More research is needed to further clarify the role of diuretics in improving renal function.

There are some studies on the prediction of AKD in hospitalized patients with AKI. Zhao et al. used multivariable logistic regression analysis with the LASSO method to select features and build a nomogram (36). The model displayed good predictive power with an AUROC of 0.834 (95% CI:0.773–0.895) in the training dataset and an AUROC of 0.851 (95% CI:0.753–0949) in the validation dataset. Yan et al. also established a prediction model using multivariable logistic regression analysis (37). The 8-variable model showed good discrimination and calibration in predicting AKD stage 2–3 with the AUROC being 0.85 (95% CI:0.83–0.87). Xiao et al. established a prediction model using multivariable logistic regression analysis. This model showed a large AUROC (0.879 ± 0.009, 0.879 ± 0.011) and had stable sensitivity (81 and 82%) and specificity (81 and 80%) in derivation cohort and validation dataset, respectively (38). In our study, the AUROCs of the logistic regression model were 0.728 (training dataset) and 0.717 (validation dataset), which were lower than the above studies. This may be due to differences in the study population. A study by Tuan et al. studied sepsis-associated AKI patients, however, they predicted progression to chronic kidney disease rather than AKD (39). Therefore, to our knowledge, this is the first study to use longitudinal data to predict the occurrence of AKD with the application of machine learning.

To identify AKD patients, an important strength of our study was the use of new criteria of sepsis-associated AKI, and this method would overcome some inherent weaknesses of using hospital discharge data (40, 41). The delta non-renal SOFA contains only 5 simple variables recorded in clinical routines. Therefore, if implemented, the delta non-renal SOFA will not require manual input of additional variables as the model is based on variables routinely collected. In our study, for predicting the occurrence of AKD, the delta non-renal SOFA score had high discriminatory power. The delta non-renal SOFA is simple for calculation and easy to use and has robust discrimination and calibration. To predict the occurrence of AKD patients with sepsis, ICU physicians could use the delta non-renal SOFA and improve clinical decision-making at the bedside. Moreover, the predictor variables that we used were quite universally obtained in the emergency department. After further validation and recalibration, the delta non-renal SOFA appeared to have the potential to help emergency department clinicians triage decisions and ICU placement.

Limitations

The study has the following limitations. First, we chose to analyze the patients admitted to the ICU with sepsis. There were certainly patients who had been diagnosed with sepsis before or after the ICU admission, but we limited our study population to those who fulfilled sepsis-3 criteria during their 1st day in ICU. Second, we have a limited number of patients and a small sample size, but we conducted an external validation by using the data of 509 sepsis-associated AKI patients from the MIMIC III database, and the results indicated that the calibration of delta non-renal SOFA was relatively well with accordance of occurrence of AKD. Finally, we prepared our dataset from the retrospective database, and the outcomes of sepsis-associated AKI patients could have changed over time due to the update of treatment guidelines and advances in treatment and diagnostic technology.

Conclusion

Machine learning could be applied to the predictive AKD, and it is where the RNN-LSTM model works the best. The non-renal SOFA plays an important role in predicting the AKD.

Data Availability Statement

The raw data supporting the conclusions of this article will be made available by the authors, without undue reservation.

Ethics Statement

The studies involving human participants were reviewed and approved by institutional review boards of Beijing Friendship Hospital and the Massachudatasetts Institute of Technology. Written informed consent for participation was not required for this study in accordance with the national legislation and the institutional requirements.

Author Contributions

JH and MD conceived the idea, performed the analysis, and drafted the manuscript. JH and JL interpreted the results and helped to revise the manuscript. JL and MD helped to frame the idea of the study and helped to analyze the data. All authors read and approved the final manuscript.

Funding

This work was supported in part by grants from the Beijing Key Clinical Specialty Excellence Project.

Conflict of Interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

Publisher's Note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

References

  • 1.Hoste EA, Bagshaw SM, Bellomo R, Cely CM, Colman R, Cruz DN, et al. Epidemiology of acute kidney injury in critically ill patients: the multinational AKI-EPI study. Intensive Care Med. (2015) 41:1411–23. 10.1007/s00134-015-3934-7 [DOI] [PubMed] [Google Scholar]
  • 2.Bouchard J, Acharya A, Cerda J, Maccariello ER, Madarasu RC, Tolwani AJ, et al. A prospective international multicenter study of AKI in the intensive care unit. Clin J Am Soc Nephrol. (2015) 10:1324–31. 10.2215/CJN.04360514 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 3.See EJ, Jayasinghe K, Glassford N, Bailey M, Johnson DW, Polkinghorne KR, et al. Long-term risk of adverse outcomes after acute kidney injury: a systematic review and meta-analysis of cohort studies using consensus definitions of exposure. Kidney Int. (2019) 95:160–72. 10.1016/j.kint.2018.08.036 [DOI] [PubMed] [Google Scholar]
  • 4.Kellum JA, Sileanu FE, Bihorac A, Hoste EA, Chawla LS. Recovery after acute kidney injury. Am J Respir Crit Care Med. (2017) 195:784–91. 10.1164/rccm.201604-0799OC [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 5.Federspiel CK, Itenov TS, Mehta K, Hsu RK, Bestle MH, Liu KD. Duration of acute kidney injury in critically ill patients. Ann Intensive Care. (2018) 8:30. 10.1186/s13613-018-0374-x [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 6.Mehta S, Chauhan K, Patel A, Patel S, Pinotti R, Nadkarni GN, et al. The prognostic importance of duration of AKI: a systematic review and meta-analysis. BMC Nephrol. (2018) 19:91. 10.1186/s12882-018-0876-7 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 7.Chawla LS, Bellomo R, Bihorac A, Goldstein SL, Siew ED, Bagshaw SM, et al. Acute kidney disease and renal recovery: consensus report of the Acute Disease Quality Initiative (ADQI) 16 workgroup. Nat Rev Nephrol. (2017) 13:241–57. 10.1038/nrneph.2017.2 [DOI] [PubMed] [Google Scholar]
  • 8.Fujii T, Uchino S, Takinami M, Bellomo R. Subacute kidney injury in hospitalized patients. Clin J Am Soc Nephrol. (2014) 9:457–61. 10.2215/CJN.04120413 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 9.Zhang Z, Ho KM, Hong Y. Machine learning for the prediction of volume responsiveness in patients with oliguric acute kidney injury in critical care. Crit Care. (2019) 23:112. 10.1186/s13054-019-2411-z [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 10.Zimmerman LP, Reyfman PA, Smith ADR, et al. Early prediction of acute kidney injury following ICU admission using a multivariate panel of physiological measurements. BMC Med Inform Decis Mak. (2019) 19:16. 10.1186/s12911-019-0733-z [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 11.Peerapornratana S, Priyanka P, Wang S, Smith A, Singbartl K, Palevsky PM, et al. Sepsis-associated acute kidney disease. Kidney Int Rep. (2020) 5:839–50. 10.1016/j.ekir.2020.03.005 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 12.Hsu CK, Wu IW, Chen YT, Tsai TY, Tsai FC, Fang JT, et al. Acute kidney disease stage predicts outcome of patients on extracorporeal membrane oxygenation support. PLoS ONE. (2020) 15:e0231505. 10.1371/journal.pone.0231505 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 13.Chen YT, Jenq CC, Hsu CK, Yu YC, Chang CH, Fan PC, et al. Acute kidney disease and acute kidney injury biomarkers in coronary care unit patients. BMC Nephrol. (2020) 21:207. 10.1186/s12882-020-01872-z [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 14.Lee BJ, Hsu CY, Parikh R, McCulloch CE, Tan TC, Liu KD, et al. Predicting renal recovery after dialysis-requiring acute kidney injury. Kidney Int Rep. (2019) 4:571–81. 10.1016/j.ekir.2019.01.015 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 15.Fiorentino M, Tohme FA, Murugan R, Kellum JA. Plasma biomarkers in predicting renal recovery from acute kidney injury in critically Ill patients. Blood Purif. (2019) 48:253–61. 10.1159/000500423 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 16.Cho JS, Shim JK, Lee S, Song JW, Choi N, Lee S, et al. Chronic progression of cardiac surgery associated acute kidney injury: intermediary role of acute kidney disease. J Thorac Cardiovasc Surg. (2021) 161:681–8.e3. 10.1016/j.jtcvs.2019.10.101 [DOI] [PubMed] [Google Scholar]
  • 17.Hu P, Song L, Liang H, Chen Y, Wu Y, Zhang L, et al. Prospective model for predicting renal recovery in cardiac surgery patients with acute kidney injury requiring renal replacement therapy. Nephrology. (2021) 26:586–93. 10.1111/nep.13878 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 18.Johnson AE, Pollard TJ, Shen L, Lehman LW, Feng M, Ghassemi M, et al. MIMIC-III, a freely accessible critical care database. Sci Data. (2016) 3:160035. 10.1038/sdata.2016.35 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 19.Moons KG, Altman DG, Reitsma JB, Ioannidis JP, Macaskill P, et al. Transparent Reporting of a multivariable prediction model for Individual Prognosis or Diagnosis (TRIPOD): explanation and elaboration. Ann Intern Med. (2015) 162:W1–73. 10.7326/M14-0698 [DOI] [PubMed] [Google Scholar]
  • 20.Singer M, Deutschman CS, Seymour CW, Shankar-Hari M, Annane D, Bauer M, et al. The third international consensus definitions for sepsis and septic shock (Sepsis-3). JAMA. (2016) 315:801–10. 10.1001/jama.2016.0287 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 21.Kidney Disease: Improving Global Outcomes (KDIGO) Acute Kidney Injury Work Group. KDIGO Clinical Practice Guideline for Acute Kidney Injury. Kidney Inter. (2012) 2:1–138.24571801 [Google Scholar]
  • 22.Kidney Disease: Improving Global Outcomes (KDIGO) CKD Work Group. KDIGO 2012 Clinical Practice Guideline for the Evaluation and Management of Chronic Kidney Disease. Kidney inter. (2013) 3:1–150.23989362 [Google Scholar]
  • 23.Austin PC, Steyerberg EW. Events per variable (EPV) and the relative performance of different strategies for estimating the out-of-sample validity of logistic regression models. Stat Methods Med Res. (2017) 26:796–808. 10.1177/0962280214558972 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 24.Lipton ZC, Kale DC, Elkan C, Wetzel RJapa. Learning to diagnose with LSTM recurrent neural networks (2015). [Google Scholar]
  • 25.Khalil K, Eldash O, Kumar A, Bayoumi M. Economic LSTM approach for recurrent neural networks. IEEE Transac Circ Syst II: Expr Briefs. (2019) 66:1885–89. 10.1109/TCSII.2019.292466330979355 [DOI] [Google Scholar]
  • 26.Breiman L, Friedman JH, Olshen RA, Stone CJ. Classification and regression trees. (2017). 10.1201/9781315139470 [DOI] [Google Scholar]
  • 27.Drummond C, Holte RC. Exploiting the cost (In) sensitivity of decision tree splitting criteria. ICML. (2000) 1:8. [Google Scholar]
  • 28.Tibshirani R. Regression shrinkage and selection via the lasso. J R Stat Soc: Ser B. (1996) 58:267–88. 10.1111/j.2517-6161.1996.tb02080.x [DOI] [Google Scholar]
  • 29.Zhang Z, Rousson V, Lee WC, Ferdynus C, Chen M, Qian X, et al. Decision curve analysis: a technical note. Ann Transl Med. (2018) 6:308. 10.21037/atm.2018.07.02 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 30.Vickers AJ, Elkin EB. Decision curve analysis: a novel method for evaluating prediction models. Med Decis Making. (2006) 26:565–74. 10.1177/0272989X06295361 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 31.Bradford JP, Kunz C, Kohavi R, Brunk C, Brodley CE. Pruning decision trees with misclassification costs. In: Paper presented at: European Conference on Machine Learning. (1998). 10.1007/BFb002668226319694 [DOI] [Google Scholar]
  • 32.Ostermann M, Zarbock A, Goldstein S, Kashani K, Macedo E, Murugan R, et al. Recommendations on acute kidney injury biomarkers from the acute disease quality initiative consensus conference: a consensus statement. JAMA Netw Open. (2020) 3:e2019209. 10.1001/jamanetworkopen.2020.19209 [DOI] [PubMed] [Google Scholar]
  • 33.Kwiatkowska E, Domański L, Dziedziejko V, Kajdy A, Stefańska K, Kwiatkowski S. The mechanism of drug nephrotoxicity and the methods for preventing kidney damage. Int J Mol Sci. (2021) 22:6109. 10.3390/ijms22116109 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 34.Bagshaw SM, Gibney RTN, Kruger P, Hassan I, McAlister FA, Bellomo R. The effect of low-dose furosemide in critically ill patients with early acute kidney injury: A pilot randomized blinded controlled trial (the SPARK study). J Crit Care. (2017) 42:138–46. 10.1016/j.jcrc.2017.07.030 [DOI] [PubMed] [Google Scholar]
  • 35.Zhao GJ, Xu C, Ying JC, Lü WB, Hong GL Li MF, et al. Association between furosemide administration and outcomes in critically ill patients with acute kidney injury. Crit Care. (2020) 24:75. 10.1186/s13054-020-2798-6 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 36.Zhao H, Liang L, Pan S, Liu Z, Liang Y, Qiao Y, et al. Diabetes mellitus as a risk factor for progression from acute kidney injury to acute kidney disease: a specific prediction model. Diabetes Metab Syndr Obes. (2021) 14:2367–79. 10.2147/DMSO.S307776 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 37.Yan P, Duan XJ, Liu Y, Wu X, Zhang NY, Yuan F, et al. Acute kidney disease in hospitalized acute kidney injury patients. PeerJ. (2021) 9:e11400. 10.7717/peerj.11400 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 38.Xiao YQ, Cheng W, Wu X, Yan P, Feng LX, Zhang NY, et al. Novel risk models to predict acute kidney disease and its outcomes in a Chinese hospitalized population with acute kidney injury. Sci Rep. (2020) 10:15636. 10.1038/s41598-020-72651-x [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 39.Tuan PNH, Quyen DBQ, Van Khoa H, Loc ND, Van My P, Dung NH, et al. Serum and urine neutrophil gelatinase-associated lipocalin levels measured at admission predict progression to chronic kidney disease in sepsis-associated acute kidney injury patients. Dis Markers. (2020) 2020:8883404. 10.1155/2020/8883404 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 40.Ford DW, Goodwin AJ, Simpson AN, Johnson E, Nadig N, Simpson KN, et al. Severe sepsis mortality prediction model and score for use with administrative data. Crit Care Med. (2016) 44:319–27. 10.1097/CCM.0000000000001392 [DOI] [PMC free article] [PubMed] [Google Scholar]
  • 41.Johnson AEW, Aboab J, Raffa JD, Pollard TJ, Deliberato RO, Celi LA, et al. A comparative analysis of sepsis identification methods in an electronic database. Crit Care Med. (2018) 46:494–9. 10.1097/CCM.0000000000002965 [DOI] [PMC free article] [PubMed] [Google Scholar]

Associated Data

This section collects any data citations, data availability statements, or supplementary materials included in this article.

Data Availability Statement

The raw data supporting the conclusions of this article will be made available by the authors, without undue reservation.


Articles from Frontiers in Medicine are provided here courtesy of Frontiers Media SA

RESOURCES