Next Article in Journal
Non-linear and Interaction Analyses of Biomarkers for Organ Dysfunctions as Predictive Markers for Sepsis: A Nationwide Retrospective Study
Previous Article in Journal
COL-3-Induced Molecular and Ultrastructural Alterations in K562 Cells
 
 
Search for Articles:
Title / Keyword
Author / Affiliation / Email
Journal
Article Type
 
 
Advanced Search
 
Section
Special Issue
Volume
Issue
Number
Page
 
Journals
JPM
Volume 12
Issue 1
10.3390/jpm12010043
jpm-logo
Submit to this Journal Review for this Journal Propose a Special Issue
► Article Menu

Article Menu

share Share announcement Help format_quote Cite question_answer Discuss in SciProfiles thumb_up
...
Endorse textsms
...
Comment
first_page
settings
Order Article Reprints
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Artificial Intelligence for Risk Prediction of Rehospitalization with Acute Kidney Injury in Sepsis Survivors

by
Shuo-Ming Ou
1,2,3,4,
Kuo-Hua Lee
1,2,3,4,
Ming-Tsun Tsai
1,2,3,4,
Wei-Cheng Tseng
1,2,3,4,
Yuan-Chia Chu
5,6,7,* and
Der-Cherng Tarng
1,2,3,4,8,*
1
Department of Medicine, Division of Nephrology, Taipei Veterans General Hospital, Taipei 11217, Taiwan
2
School of Medicine, National Yang Ming Chiao Tung University, Taipei 11221, Taiwan
3
Institute of Clinical Medicine, National Yang Ming Chiao Tung University, Taipei 11221, Taiwan
4
Center for Intelligent Drug Systems and Smart Bio-Devices (IDS2B), National Yang Ming Chiao Tung University, Hsinchu 30010, Taiwan
5
Information Management Office, Taipei Veterans General Hospital, Taipei 11217, Taiwan
6
Big Data Center, Taipei Veterans General Hospital, Taipei 11217, Taiwan
7
Department of Information Management, National Taipei University of Nursing and Health Sciences, Taipei 11219, Taiwan
8
Department and Institute of Physiology, National Yang Ming Chiao Tung University, Taipei 11221, Taiwan
*
Authors to whom correspondence should be addressed.
J. Pers. Med. 2022, 12(1), 43; https://doi.org/10.3390/jpm12010043
Submission received: 18 November 2021 / Revised: 26 December 2021 / Accepted: 29 December 2021 / Published: 4 January 2022
Browse Figures
Versions Notes

Abstract

:
Sepsis survivors have a higher risk of long-term complications. Acute kidney injury (AKI) may still be common among sepsis survivors after discharge from sepsis. Therefore, our study utilized an artificial-intelligence-based machine learning approach to predict future risks of rehospitalization with AKI between 1 January 2008 and 31 December 2018. We included a total of 23,761 patients aged ≥ 20 years who were admitted due to sepsis and survived to discharge. We adopted a machine learning method by using models based on logistic regression, random forest, extra tree classifier, gradient boosting decision tree (GBDT), extreme gradient boosting, and light gradient boosting machine (LGBM). The LGBM model exhibited the highest area under the receiver operating characteristic curves (AUCs) of 0.816 to predict rehospitalization with AKI in sepsis survivors and followed by the GBDT model with AUCs of 0.813. The top five most important features in the LGBM model were C-reactive protein, white blood cell counts, use of inotropes, blood urea nitrogen and use of diuretics. We established machine learning models for the prediction of the risk of rehospitalization with AKI in sepsis survivors, and the machine learning model may set the stage for the broader use of clinical features in healthcare.

1. Introduction

Sepsis is estimated to affect 19.4 million patients, with an annual sepsis-related mortality of approximately 5.3 million cases [1]. Therefore, sepsis is a major public health concern due to the life-threatening organ dysfunction and the dysregulated host response to infection, and sepsis is a common cause of death in hospitalized patients [2,3]. As there has been significant medical progress in decreasing mortality and morbidity after sepsis, attention to the complications after discharge in sepsis survivors has become more important [4,5,6,7].
Acute kidney injury (AKI) frequently occurs with sepsis due to pathologic interactions of multiple organ dysfunction, systemic hypotension, inflammatory cytokine storms and nephrotoxic drugs, which all indirectly and directly contributing to renal injury [8,9,10]. Previous studies have found that 40% to 50% of patients with AKI had sepsis [8,11], and approximately 11% to 42% of patients with sepsis developed AKI [12,13,14]. Unplanned rehospitalization is associated with worsening patient outcomes and increased treatment costs [15,16]. Although AKI is a common complication in sepsis, the risks of rehospitalization with AKI in sepsis survivors remains unknown. Therefore, the development of a prediction model for rehospitalization with AKI has become an important therapeutic goal in the management of sepsis survivors.
To appropriately manage rehospitalization with AKI in sepsis survivors, a precise prediction model for identifying high-risk patients is required to optimize the treatment strategy. This predictive model is important not only to allow a more comprehensive prognostication of patients’ well-being but also to reduce the healthcare financial burdens. Machine learning models have already been applied in many fields, such as outcome prediction [17,18,19], and these models may potentially be used to identify high-risk patients. Machine learning models have been mostly described to predict episodes of the occurrence of AKI during sepsis [20,21,22]. However, there is no study to evaluate their effects on rehospitalization with AKI after patients who survived to discharge from sepsis. To resolve this important issue, we conducted a large-scale cohort study of sepsis survivors, and the predictive ability of the machine learning model was compared to select the optimal machine learning model.

2. Materials and Methods

2.1. Study Design and Data Source

We established a database including the detailed information of sepsis survivors extracted from the Big Data Center of Taipei Veterans General Hospitals between 1 January 2008 and 31 December 2018, which included the comprehensive medical records from the inpatient, outpatient, and emergent departmental records [23]. The detailed patient demographic, clinical, diagnostic/procedural information, drug prescriptions, procedural codes, and laboratory data were included in our analysis. To identify the sepsis survivors, we included all patients with discharge codes based on the International Classification of Diseases, Ninth and Tenth Edition, Clinical Modification (ICD-9-CM and ICD-10-CM) codes for sepsis (ICD code 038, 995.91, A40 and A41), severe sepsis (ICD code 995.92 and R65.20) or septic shock (ICD code 785.52 and R65.21) during the study period who were discharged alive [24]. We excluded patients who had pre-existing end-stage kidney disease maintained with dialysis or kidney transplant before discharge, were younger than 20 years old, or who died during admission.

2.2. Class Definition

The class was labeled as 1 if there was rehospitalization with AKI during the follow-up periods; otherwise, the class was labeled as 0 if there was no rehospitalization with AKI. The diagnosis of AKI was defined as a 0.3 mg/dL within 48 h or 50% increase within 7 days from the baseline creatinine based on the Kidney Disease Improving Global Outcomes classification (KDIGO) definition [25]. We included the first-time rehospitalization with AKI because multiple admissions may introduce a bias favoring survivors.

2.3. Machine Learning Algorithm and Statistical Analysis

Continuous data are presented as the median (interquartile ranges (IQRs)) and categorical data are presented as numbers (proportions). Before the machine learning processes, the missing values of the clinical variables were imputed using the k-nearest neighbors (KNN) algorithm [26,27]. The whole dataset was then randomly split into a training dataset and a validation dataset at a ratio of 70:30%, respectively. In our study, we used several machine learning methods, including logistic regression, a random forest [28], an extra tree classifier [29], an extreme gradient boosting (XGBoost) [30], a light gradient boosting machine (LGBM) [31], and a gradient boosting decision tree (GBDT) [32], to predict risks of rehospitalization with AKI. The prediction abilities of various machine learning models were examined based on the area under the curve of receiver operating characteristics (AUCs) and precision-recall curves of each model. As the methods of prediction in machine learning models are often unclear, we used SHapley Additive exPlanation (SHAP) values to provide accurate attribution values for each clinical feature in our prediction model [33,34,35]. The data were analyzed by using Python (Python Software Foundation version 3.7.6, available at http://www.python.org, accessed on 1 November 2021). All tests were two-tailed, and a p value < 0.05 was statistically significant.

3. Results

3.1. Study Population

In the 10-year study period, 23,761 sepsis survivors were included in our final cohort, and the detailed patient demographic data are presented in Table 1. Sepsis survivors were predominantly female, and 55.7% of the patients had hypertension, 32.8% had diabetes mellites, and 44.3% used CCBs. The patients had a baseline creatinine level of 1.1 mg/dL. We further divided the sepsis survivors randomly into the two groups and allocated 70% of them to the training set and the remaining 30% to the test set. Among these patients, 8756 (36.9%) sepsis survivors had episodes of rehospitalization with AKI in sepsis survivors with the median intervals from discharge to rehospitalization of 8.6 months. In addition, there were 6076 (36.5%) and 2680 (37.6%) episodes of rehospitalization with AKI in the training and testing datasets, respectively.

3.2. Model Prediction Ability

The 84 features, including the demographic characteristics, underlying comorbidities, laboratory data and concomitant medications, that were used in our machine learning models are listed in Table 2. We utilized the following machine learning methods with all the clinical features as input variables: logistic regression, random forest, extra tree classifier, XGBoost, GBDT, and LGBM (Figure 1). Regarding the predictive ability of the models for outcome prediction, the LGBM exhibited the largest AUC of 0.816, and the GBDT model had the second highest AUC of 0.813. The logistic regression exhibited the smallest AUC of 0.683.

3.3. Ranks of Feature Importance and SHAP Value in the Machine Learning Models

To identify important features in the LGBM model, we performed a feature importance plot by using SHAP values and listed the features in descending order. The top five important features were C-reactive protein, white blood cell counts, use of inotropes, blood urea nitrogen, and use of diuretics, which contribute to higher predictive powers than the bottom features (Figure 2A). The local bar plot of a sepsis survivor showed how the SHAP values of features affected the model prediction (Figure 2B). Red SHAP values increased the prediction, and blue values decreased it. The SHAP heatmap plot were shown in Figure 3A, and features with higher SHAP values were highlighted in redder boxes. The dependent plots revealed the interaction effects between C-reactive protein (which is the top-most important feature), white blood cell counts, and blood urea nitrogen in our LGBM model (Figure 3B–D).

4. Discussion

In our cohort study, 23,761 sepsis survivors suffered from rehospitalization with AKI after discharge. We developed machine learning algorithms using 84 clinical features to predict rehospitalization with AKI and compared the AUCs of the different machine learning models. We found that the LGBM model had the highest AUC of 0.816 compared to the other machine learning models. Our study suggests that AKI might still be an unrecognized outcome after discharge from sepsis, and the use of machine learning models may help to predict rehospitalization with AKI.
AKI is frequently observed in patients with sepsis, and a study including 2871 patients from the critical care database developed risk-prediction nomogram for AKI with C-index of 0.75 [20]. Another study including 15,726 patients with sepsis from the same critical care database established a prediction model by using logistic regression with a C-index of 0.71 [21]. The prediction models established by these studies were limited by only using the logistic regression method rather than other machine learning models to improve the predictive ability. Moreover, a study including 5984 septic patients with AKI established five prediction models, including logistic regression, random forest, support vector machine, artificial neural network, and extreme gradient boosting to predict persistent AKI [22]. The artificial neural network and logistic regression models achieved the highest AUC of 0.76. However, none of the studies carried out so far have considered whether sepsis survivors are still at greater risks of AKI after discharge.
Sepsis survivors were found to be the highest risks for short- and long-term outcomes after discharge from sepsis [7,36,37,38]. However, the rates of rehospitalization with AKI have never been explored, and an understanding of such complication is required for physicians to initiate early treatment and follow-up strategies. In our study, the incidence of rehospitalization with AKI is still high, with approximately 36.9% of sepsis survivors after discharge. Our study is the first in the literature to use a machine learning approach to predict risks of rehospitalization with AKI, and the optimal AUC was achieved to 0.816 in the LGBM model. The performance of LGBM was higher than that of traditional logistic regression model (AUC: 0.683) for the prediction of rehospitalization with AKI.
In addition, the feature importance plot using SHAP value in our LGBM found some important predictors for risks of rehospitalization with AKI, some of which were consistent with the traditional factors. The important predictors for AKI, such as C-reactive protein, white blood cell counts, and the use of inotropes may be associated with the infectious status before discharge. In addition, blood urea nitrogen levels and the use of diuretics may reflex the fluid status and were traditionally associated with future risks of rehospitalization with AKI. Therefore, our machine learning model may help identify high-risk sepsis survivors who are prone to rehospitalization with AKI after considering clinical features related to their infection conditions or fluid status.
Our study has several strengths. First, compared to previous studies, our study is the first to predict the risks of rehospitalization with AKI after discharge from sepsis for a large number of sepsis survivors. Second, our study evaluated the laboratory data, and we included sepsis survivors who had more than two serum creatinine measurements. Therefore, we had the ability to discriminate the sepsis survivors’ outcomes, including the rehospitalization with AKI, which may reduce the possible underreporting or misclassifications of AKI by using the International Statistical Classification of Diseases and Related Health Problems (ICD) coding, compared to other studies that extracted data from administrative datasets. Finally, we established predictive models of machine learning algorithms that might be important to apply in clinical practice.
Our study may have several limitations that should be noted. First, because of the nature of observational studies, the causal inference of rehospitalization with AKI might be confounded by unmeasured factors. Second, as our analysis was based on a single tertiary medical center’s data, some age and disease group particularities regarding old age (median, 76.4 years) and higher cancer incidence (48.8%), which are factors that may induce some bias to the renal function or rehospitalization with AKI in the analyzed subjects. Third, the machine learning algorithm learned from the input clinical features, and some hidden relationships may be unknown if the features were not included by the physicians.

5. Conclusions

Our study established a machine learning algorithm for the detection and prediction of rehospitalization with AKI. Therefore, our findings support the implementation of a useful machine learning algorithm for risks of rehospitalization with AKI. Due to the age distribution, disease particularities, and single-center-based character of our study, external validation is required to evaluate the generalizability.

Author Contributions

Conceptualization, Y.-C.C. and D.-C.T.; data curation, S.-M.O., Y.-C.C. and D.-C.T.; data analyses, S.-M.O., Y.-C.C. and D.-C.T.; funding acquisition, S.-M.O., Y.-C.C. and D.-C.T.; investigation, S.-M.O., K.-H.L., M.-T.T., W.-C.T., Y.-C.C. and D.-C.T.; methodology, S.-M.O., Y.-C.C. and D.-C.T.; project administration, S.-M.O.; supervision, Y.-C.C. and D.-C.T.; validation, K.-H.L., M.-T.T. and W.-C.T.; visualization, S.-M.O. and Y.-C.C.; writing—original draft, S.-M.O.; writing—review and editing, S.-M.O., Y.-C.C. and D.-C.T. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported in part by the Ministry of Science and Technology, Taiwan (MOST 106-2314-B-010-039-MY3, MOST 107-2314-B-075-052, MOST 108-2314-B-075-008, MOST 109-2314-B-075 -067-MY3, MOST 109-2320-B-075-006, MOST109-2314-B-075-097-MY3, MOST110-2312-B-075-002, MOST 110-2320-B-075-004-MY3); Taipei Veterans General Hospital (V107B-027, V108B-023, V108C-103, V108D42-004-MY3-2, V109B-022, V109C-114, V109D50-001-MY3-1, V109D50-001-MY3-2, V109D50-001-MY3-3,V109D50-002-MY3-3, V109E-008-5(110),V110C-152, V110E-003-2,V110E-003-2, V111E-002-3,V111C-171,V111C-151,V111D60-004-MY3-1); Taipei Veterans General Hospital-National Yang-Ming University Excellent Physician Scientists Cultivation Program (No.104-V-B-044). Taipei, Taichung, Kaohsiung Veterans General Hospital, Tri-Service General Hospital, Academia Sinica Joint Research Program (VTA110-V1-3-1) and Foundation for Poison Control (FPC-109-002). The funders did not play any role in the study design, data collection or analysis, decision to publish, or preparation of the manuscript.

Institutional Review Board Statement

This study was carried out in accordance with the Declaration of Helsinki, with appropriate approvals obtained from the institutional review board of Taipei Veterans General Hospital (2017-09-002BC).

Informed Consent Statement

This study was approved waiving the informed consent requirement because of de-identified data.

Data Availability Statement

The data analyzed in this study are not publicly available because individual privacy may be compromised. Interested groups could contact Shuo-Ming Ou at [email protected] to request permission to access these datasets.

Acknowledgments

We thank the Big Data Center, Taipei Veterans General Hospital (BDC, TPEVGH). The interpretations and conclusions contained herein do not represent the position of Taipei Veterans General Hospital.

Conflicts of Interest

All listed authors have no conflict of interest to declare.

References

  1. Fleischmann, C.; Scherag, A.; Adhikari, N.K.; Hartog, C.S.; Tsaganos, T.; Schlattmann, P.; Angus, D.C.; Reinhart, K. Assessment of Global Incidence and Mortality of Hospital-treated Sepsis. Current Estimates and Limitations. Am. J. Respir. Crit. Care Med. 2016, 193, 259–272. [Google Scholar] [CrossRef]
  2. Rudd, K.E.; Johnson, S.C.; Agesa, K.M.; Shackelford, K.A.; Tsoi, D.; Kievlan, D.R.; Colombara, D.V.; Ikuta, K.S.; Kissoon, N.; Finfer, S.; et al. Global, regional, and national sepsis incidence and mortality, 1990–2017: Analysis for the Global Burden of Disease Study. Lancet 2020, 395, 200–211. [Google Scholar] [CrossRef] [Green Version]
  3. Singer, M.; Deutschman, C.S.; Seymour, C.W.; Shankar-Hari, M.; Annane, D.; Bauer, M.; Bellomo, R.; Bernard, G.R.; Chiche, J.-D.; Coopersmith, C.M.; et al. The Third International Consensus Definitions for Sepsis and Septic Shock (Sepsis-3). J. Am. Med. Assoc. 2016, 315, 801–810. [Google Scholar] [CrossRef]
  4. Wu, M.H.; Tsou, P.Y.; Wang, Y.H.; Lee, M.G.; Chao, C.C.T.; Lee, W.C.; Lee, S.-H.; Hu, J.-R.; Wu, J.-Y.; Chang, S.-S.; et al. Impact of post-sepsis cardiovascular complications on mortality in sepsis survivors: A population-based study. Crit. Care 2019, 23, 293. [Google Scholar] [CrossRef] [Green Version]
  5. Ou, L.; Chen, J.; Hillman, K.; Flabouris, A.; Parr, M.; Assareh, H.; Bellomo, R. The impact of post-operative sepsis on mortality after hospital dis-charge among elective surgical patients: A population-based cohort study. Crit. Care 2017, 21, 34. [Google Scholar] [CrossRef] [Green Version]
  6. Prescott, H.C.; Langa, K.M.; Liu, V.; Escobar, G.J.; Iwashyna, T.J. Increased 1-Year Healthcare Use in Survivors of Severe Sepsis. Am. J. Respir. Crit. Care Med. 2014, 190, 62–69. [Google Scholar] [CrossRef] [Green Version]
  7. Ou, S.M.; Chu, H.; Chao, P.W.; Lee, Y.J.; Kuo, S.C.; Chen, T.J.; Tseng, C.-M.; Shih, C.-J.; Chen, Y.-T. Long-Term Mortality and Major Adverse Cardiovascular Events in Sepsis Survivors. A Nationwide Population-based Study. Am. J. Respir. Crit. Care Med. 2016, 194, 209–217. [Google Scholar] [CrossRef]
  8. Peerapornratana, S.; Manrique-Caballero, C.L.; Gómez, H.; Kellum, J.A. Acute kidney injury from sepsis: Current concepts, epidemiology, pathophysiology, prevention and treatment. Kidney Int. 2019, 96, 1083–1099. [Google Scholar] [CrossRef]
  9. Bellomo, R.; Kellum, J.A.; Ronco, C.; Wald, R.; Martensson, J.; Maiden, M.; Bagshaw, S.M.; Glassford, N.J.; Yugeesh, L.; Vaara, S.T.; et al. Acute kidney injury in sepsis. Intensive Care Med. 2017, 43, 816–828. [Google Scholar] [CrossRef] [Green Version]
  10. Zarjou, A.; Agarwal, A. Sepsis and acute kidney injury. J. Am. Soc. Nephrol. 2011, 22, 999–1006. [Google Scholar] [CrossRef] [Green Version]
  11. Murugan, R.; Karajala-Subramanyam, V.; Lee, M.; Yende, S.; Kong, L.; Carter, M.; Angus, D.C.; Kellum, J.A. Acute kidney injury in non-severe pneumonia is associated with an increased immune response and lower survival. Kidney Int. 2010, 77, 527–535. [Google Scholar] [CrossRef] [Green Version]
  12. Godin, M.; Murray, P.; Mehta, R.L. Clinical Approach to the Patient with AKI and Sepsis. Semin. Nephrol. 2015, 35, 12–22. [Google Scholar] [CrossRef] [Green Version]
  13. Hoste, E.A.; Lameire, N.H.; Vanholder, R.C.; Benoit, D.D.; Decruyenaere, J.M.; Colardyn, F.A. Acute Renal Failure in Patients with Sepsis in a Surgical ICU: Predictive Factors, Incidence, Comorbidity, and Outcome. J. Am. Soc. Nephrol. 2003, 14, 1022–1030. [Google Scholar] [CrossRef] [Green Version]
  14. Bagshaw, S.M.; George, C.; Bellomo, R. Early acute kidney injury and sepsis: A multicentre evaluation. Crit. Care 2008, 12, R47. [Google Scholar] [CrossRef] [Green Version]
  15. Donnelly, J.P.; Hohmann, S.F.; Wang, H.E. Unplanned Readmissions After Hospitalization for Severe Sepsis at Academic Medical Center-Affiliated Hospitals. Crit. Care Med. 2015, 43, 1916–1927. [Google Scholar] [CrossRef]
  16. Singh, A.; Bhagat, M.; George, S.V.; Gorthi, R.; Chaturvedula, C. Factors Associated with 30-day Unplanned Readmissions of Sepsis Patients: A Retrospective Analysis of Patients Admitted with Sepsis at a Community Hospital. Cureus 2019, 11, e5118. [Google Scholar] [CrossRef] [Green Version]
  17. Shamout, F.; Zhu, T.; Clifton, D.A. Machine Learning for Clinical Outcome Prediction. IEEE Rev. Biomed. Eng. 2021, 14, 116–126. [Google Scholar] [CrossRef]
  18. Heo, J.; Yoon, J.G.; Park, H.; Kim, Y.D.; Nam, H.S.; Heo, J.H. Machine Learning—Based Model for Prediction of Outcomes in Acute Stroke. Stroke 2019, 50, 1263–1265. [Google Scholar] [CrossRef]
  19. Lai, C.C.; Huang, W.H.; Chang, B.C.; Hwang, L.C. Development of Machine Learning Models for Prediction of Smoking Cessation Outcome. Int. J. Environ. Res. Public Health 2021, 18, 2584. [Google Scholar] [CrossRef]
  20. Yang, S.; Su, T.; Huang, L.; Feng, L.H.; Liao, T. A novel risk-predicted nomogram for sepsis associated-acute kidney injury among critically ill patients. BMC Nephrol. 2021, 22, 173. [Google Scholar] [CrossRef]
  21. Fan, C.; Ding, X.; Song, Y. A new prediction model for acute kidney injury in patients with sepsis. Ann. Palliat. Med. 2021, 10, 1772–1778. [Google Scholar] [CrossRef]
  22. Luo, X.Q.; Yan, P.; Zhang, N.Y.; Luo, B.; Wang, M.; Deng, Y.H.; Wu, T.; Wu, X.; Liu, Q.; Wang, H.-S.; et al. Machine learning for early discrimination between transient and persistent acute kidney injury in critically ill patients with sepsis. Sci. Rep. 2021, 11, 20269. [Google Scholar] [CrossRef]
  23. Kuan, A.S.; Chen, T.J. Healthcare data research: The inception of the Taipei Veterans General Hospital Big Data Center. J. Chin. Med. Assoc. 2019, 82, 679. [Google Scholar] [CrossRef]
  24. Jolley, R.J.; Sawka, K.J.; Yergens, D.W.; Quan, H.; Jetté, N.; Doig, C.J. Validity of administrative data in recording sepsis: A systematic review. Crit. Care. 2015, 19, 139. [Google Scholar] [CrossRef] [Green Version]
  25. Acosta-Ochoa, I.; Bustamante-Munguira, J.; Mendiluce-Herrero, A.; Bustamante-Bustamante, J.; Coca-Rojo, A. Impact on Outcomes across KDIGO-2012 AKI Criteria According to Baseline Renal Function. J. Clin. Med. 2019, 8, 1323. [Google Scholar] [CrossRef] [Green Version]
  26. Stevens, J.R.; Suyundikov, A.; Slattery, M.L. Accounting for Missing Data in Clinical Research. JAMA 2016, 315, 517–518. [Google Scholar] [CrossRef]
  27. Liao, S.G.; Lin, Y.; Kang, D.D.; Chandra, D.; Bon, J.; Kaminski, N.; Sciurba, F.C.; Tseng, G.C. Missing value imputation in high-dimensional phenomic data: Imputable or not, and how? BMC Bioinform. 2014, 15, 346. [Google Scholar] [CrossRef] [Green Version]
  28. Rigatti, S.J. Random Forest. J. Insurance Med. 2017, 47, 31–39. [Google Scholar] [CrossRef] [Green Version]
  29. Geurts, P.; Ernstr, D.; Wehenkel, L. Extremely Randomized Trees. Mach. Learn. 2006, 63, 3–42. [Google Scholar] [CrossRef] [Green Version]
  30. Mustapha, I.B.; Saeed, F. Bioactive Molecule Prediction Using Extreme Gradient Boosting. Molecules 2016, 21, 983. [Google Scholar] [CrossRef] [Green Version]
  31. Rahman, S.; Irfan, M.; Raza, M.; Ghori, K.; Yaqoob, S.; Awais, M. Performance Analysis of Boosting Classifiers in Recognizing Activities of Daily Living. Int. J. Environ. Res. Public Health 2020, 17, 1082. [Google Scholar] [CrossRef] [Green Version]
  32. Natekin, A.; Knoll, A. Gradient boosting machines, a tutorial. Front. Neurorobot. 2013, 7, 21. [Google Scholar] [CrossRef] [Green Version]
  33. Rodríguez-Pérez, R.; Bajorath, J. Interpretation of machine learning models using shapley values: Application to compound potency and multi-target activity predictions. J. Comput. Mol. Des. 2020, 34, 1013–1026. [Google Scholar] [CrossRef]
  34. Parsa, A.B.; Movahedi, A.; Taghipour, H.; Derrible, S.; Mohammadian, A.K. Toward safer highways, application of XGBoost and SHAP for real-time accident detection and feature analysis. Accid. Anal. Prev. 2020, 136, 105405. [Google Scholar] [CrossRef]
  35. Bloch, L.; Friedrich, C.M. Data analysis with Shapley values for automatic subject selection in Alzheimer’s disease data sets using interpretable machine learning. Alzheimer’s Res. Ther. 2021, 13, 155. [Google Scholar] [CrossRef]
  36. Shankar-Hari, M.; Rubenfeld, G.D. Understanding Long-Term Outcomes Following Sepsis: Implications and Challenges. Curr. Infect. Dis. Rep. 2016, 18, 37. [Google Scholar] [CrossRef] [Green Version]
  37. Shankar-Hari, M.; Harrison, D.A.; Ferrando-Vivas, P.; Rubenfeld, G.D.; Rowan, K. Risk Factors at Index Hospitalization Associated with Longer-term Mortality in Adult Sepsis Survivors. JAMA Netw. Open 2019, 2, e194900. [Google Scholar] [CrossRef] [Green Version]
  38. Meurer, W.J.; Losman, E.D.; Smith, B.L.; Malani, P.N.; Younger, J.G. Short-term functional decline of older adults admitted for suspected sepsis. Am. J. Emerg. Med. 2011, 29, 936–942. [Google Scholar] [CrossRef]
Jpm 12 00043 g001 550
Figure 1. Comparison of (A) ROC curve, (B) PR curve and (C) predictive ability among different machine learning models. Abbreviations: ROC, receiver operating characteristic curve; PR, precision-recall; GBDT, gradient boosting decision tree; XGBoost, extreme gradient boosting; LGBM, light gradient boosting machine LGBM. Abbreviations: ROC, receiver operating characteristic curve; PR, precision-recall; AUC, area under the curve of receiver operating characteristic curve; LGBM, light gradient boosting machine; GBDT, Gradient Boosting Decision Tree; XGBoost, extreme gradient boosting.
Figure 1. Comparison of (A) ROC curve, (B) PR curve and (C) predictive ability among different machine learning models. Abbreviations: ROC, receiver operating characteristic curve; PR, precision-recall; GBDT, gradient boosting decision tree; XGBoost, extreme gradient boosting; LGBM, light gradient boosting machine LGBM. Abbreviations: ROC, receiver operating characteristic curve; PR, precision-recall; AUC, area under the curve of receiver operating characteristic curve; LGBM, light gradient boosting machine; GBDT, Gradient Boosting Decision Tree; XGBoost, extreme gradient boosting.
Jpm 12 00043 g001
Jpm 12 00043 g002 550
Figure 2. (A) The feature importance plot using SHAP value reveals the important features in the final LGBM model (B)The local bar plots of feature importance reveal the SHAP values for features in the model. The feature values are show in gray to the left of the feature names. Abbreviations: SHAP, SHapley Additive exPlanation; LGBM, light gradient boosting machine; CRP, C-reactive protein; WBC, white blood cell counts; BUN, blood urea nitrogen; CCB, calcium channel blocker; INR, international normalized ratio; Hgb, hemoglobin; Na, sodium; NSAID, non-steroidal anti-inflammatory drug; aPTT, activated partial thromboplastin time; AST, aspartate aminotransferase; INR, international normalized ratio.
Figure 2. (A) The feature importance plot using SHAP value reveals the important features in the final LGBM model (B)The local bar plots of feature importance reveal the SHAP values for features in the model. The feature values are show in gray to the left of the feature names. Abbreviations: SHAP, SHapley Additive exPlanation; LGBM, light gradient boosting machine; CRP, C-reactive protein; WBC, white blood cell counts; BUN, blood urea nitrogen; CCB, calcium channel blocker; INR, international normalized ratio; Hgb, hemoglobin; Na, sodium; NSAID, non-steroidal anti-inflammatory drug; aPTT, activated partial thromboplastin time; AST, aspartate aminotransferase; INR, international normalized ratio.
Jpm 12 00043 g002
Jpm 12 00043 g003 550
Figure 3. (A) The SHAP heatmap plot with the model’s predictions for important features. The heatmap plot function creates a plot with the features on the y-axis, the model inputs on the x-axis, and the SHAP values encoded on a color scale with a higher value being redder, and a lower value being bluer. The SHAP dependence plots showed the interaction effects between (B) CRP and WBC, (C) CRP and BUN and (D) WBC and BUN in the LGBM model. Abbreviations: SHAP, SHapley Additive exPlanation; CRP, C-reactive protein; WBC, white blood cell counts; BUN, blood urea nitrogen; LGBM, light gradient boosting machine.
Figure 3. (A) The SHAP heatmap plot with the model’s predictions for important features. The heatmap plot function creates a plot with the features on the y-axis, the model inputs on the x-axis, and the SHAP values encoded on a color scale with a higher value being redder, and a lower value being bluer. The SHAP dependence plots showed the interaction effects between (B) CRP and WBC, (C) CRP and BUN and (D) WBC and BUN in the LGBM model. Abbreviations: SHAP, SHapley Additive exPlanation; CRP, C-reactive protein; WBC, white blood cell counts; BUN, blood urea nitrogen; LGBM, light gradient boosting machine.
Jpm 12 00043 g003
Table
Table 1. Clinical features and classes in sepsis survivors divided into training and testing dataset.
Table 1. Clinical features and classes in sepsis survivors divided into training and testing dataset.
All PatientsTraining SetTesting Set
(n = 23,761)(n = 16,632)(n = 7129)
Demographic and clinical characteristics
Age, years76.4 (61.4, 85.2)76.4 (61.2, 85.2)76.4 (61.9, 85.2)
Male sex, n (%)8557 (36.0)5995 (36.0)2562 (35.9)
Smoking, n (%)5373 (22.6)3744 (22.5)1629 (22.9)
Alcohol consumption, n (%)3945 (16.6)2739 (16.5)1206 (16.9)
Underlying Comorbidities
Hypertension, n (%)13,238 (55.7)9271 (55.7)3967 (55.6)
Transient ischemic attack, n (%)579 (2.4)399 (2.4)180 (2.5)
Ischemic stroke, n (%)3343 (14.1)2358 (14.2)985 (13.8)
Hemorrhagic stroke, n (%)1084 (4.6)774 (4.7)310 (4.3)
Dementia, n (%)3190 (13.4)2218 (13.3)972 (13.6)
Diabetes mellitus, n (%)7803 (32.8)5432 (32.7)2371 (33.3)
Gout, n (%)2443 (10.3)1737 (10.4)706 (9.9)
Myocardial infarction, n (%)1852 (7.8)1272 (7.6)580 (8.1)
Coronary artery disease, n (%)6260 (26.3)4308 (25.9)1952 (27.4)
CHF, n (%)4759 (20.0)3282 (19.7)1477 (20.7)
Atrial fibrillation, n (%)2394 (10.1)1667 (10.0)727 (10.2)
Chronic liver disease, n (%)3875 (16.3)2729 (16.4)1146 (16.1)
Cirrhosis, n (%)1395 (5.9)996 (6.0)399 (5.6)
Peptic ulcer disease, n (%)5632 (23.7)3957 (23.8)1675 (23.5)
COPD, n (%)4469 (18.8)3110 (18.7)1359 (19.1)
Asthma, n (%)1192 (5.0)835 (5.0)357 (5.0)
PAOD, n (%)192 (0.8)130 (0.8)62 (0.9)
Autoimmune disease, n (%)821 (3.5)591 (3.6)230 (3.2)
Cancer, n (%)11,592 (48.8)8145 (49.0)3447 (48.4)
Valvular heart disease, n (%)1303 (5.5)908 (5.5)395 (5.5)
Critical conditions
ICU admission, n (%)12,962 (54.6)9041 (54.4)3921 (55.0)
Use of mechanical ventilators, n (%)8740 (36.8)6083 (36.6)2657 (37.3)
Use of inotropes, n (%)11,343 (47.7)7933 (47.7)3410 (47.8)
Laboratory data
Blood urea nitrogen, mg/dL24.0 (14.0, 51.0)24.0 (14.0, 51.0)24.0 (14.0, 50.0)
Creatinine, mg/dL1.1 (0.7, 2.1)1.1 (0.7, 2.2)1.1 (0.7, 2.1)
White blood cells, /mm38100 (5700, 11,900)8100 (5700, 11,900)8100 (5700, 12,000)
Hemoglobin, g/dL10.1 (8.9, 11.5)10.1 (8.9, 11.5)10.1 (9.0, 11.6)
Sodium, mmol/L139.0 (135.0, 142.0)139.0 (135.0, 142.0)139.0 (135.0, 142.0)
Potassium, mmol/L4.1 (3.6, 4.6)4.1 (3.6, 4.6)4.1 (3.6, 4.6)
Chloride, mmol/L103.0 (98.0, 106.0)103.0 (98.0, 106.0)103.0 (98.0, 106.0)
Calcium, mg/dL8.5 (8.0, 9.0)8.5 (8.0, 9.0)8.5 (8.0, 9.0)
Phosphate, mg/dL3.3 (2.6, 4.0)3.3 (2.6, 4.0)3.3 (2.7, 4.1)
HCO3, mmol/L23.7 (19.3, 28.0)23.7 (19.3, 28.0)23.8 (19.4, 28.0)
C-reactive protein, mg/dL3.4 (1.2, 9.0)3.4 (1.2, 9.1)3.3 (1.1, 8.7)
Albumin, mg/dL3.0 (2.6, 3.4)3.0 (2.6, 3.4)3.0 (2.6, 3.4)
Alanine transaminase, U/L25.0 (15.0, 44.0)25.0 (15.0, 45.0)25.0 (15.0, 44.0)
Aspartate transaminase, U/L29.0 (20.0, 51.0)29.0 (20.0, 51.0)29.0 (20.0, 50.0)
Alkaline phosphatase, U/L95.0 (70.0, 147.0)95.0 (69.0, 147.0)94.0 (70.0, 147.0)
Gamma-glutamyl transferase, U/L54.0 (25.0, 125.0)53.0 (25.0, 125.0)54.0 (24.0, 126.0)
Total bilirubin, mg/dL0.6 (0.4, 1.1)0.6 (0.4, 1.1)0.6 (0.4, 1.1)
HbA1c, %6.4 (5.8, 7.4)6.4 (5.8, 7.4)6.4 (5.8, 7.4)
Glucose, mg/dL116.0 (95.0, 156.0)116.0 (94.0, 155.0)117.0 (95.0, 157.0)
Uric acid, mg/dL5.5 (4.1, 7.1)5.5 (4.1, 7.1)5.6 (4.1, 7.1)
Cholesterol, mg/dL151.0 (122.0, 182.0)152.0 (122.0, 183.0)150.0 (121.0, 181.0)
LDL-C, mg/dL91.0 (70.0, 114.0)91.0 (70.0, 115.0)91.0 (69.0, 113.0)
HDL-C, mg/dL41.0 (32.0, 51.0)41.0 (32.0, 51.0)41.0 (32.0, 51.0)
INR1.1 (1.0, 1.2)1.1 (1.0, 1.2)1.1 (1.0, 1.2)
aPTT, seconds29.9 (27.1, 34.0)29.9 (27.2, 34.2)29.9 (27.1, 33.8)
D-Dimer, ug/mL3.6 (1.6, 8.1)3.6 (1.5, 7.7)3.9 (1.8, 9.3)
LDH, U/L253.0 (196.0, 361.0)252.0 (196.0, 361.0)255.0 (197.0, 361.0)
NT-proBNP, pg/mL3146.0 (836.5, 11,617.0)3142.0 (823.8, 11,648.5)3185.0 (856.8, 11,580.8)
Concomitant Medications
ACEI, n (%)2225 (9.4)1537 (9.2)688 (9.7)
ARB, n (%)6972 (29.3)4884 (29.4)2088 (29.3)
Alpha, blocker, n (%)6109 (25.7)4228 (25.4)1881 (26.4)
Beta blocker, n (%)8521 (35.9)5891 (35.4)2630 (36.9)
CCB, n (%)10,534 (44.3)7362 (44.3)3172 (44.5)
Warfarin, n (%)1263 (5.3)893 (5.4)370 (5.2)
DOAC, n (%)147 (0.6)106 (0.6)41 (0.6)
Aspirin, n (%)5445 (22.9)3743 (22.5)1702 (23.9)
Plavix, n (%)3267 (13.7)2242 (13.5)1025 (14.4)
Nitrate, n (%)6521 (27.4)4473 (26.9)2048 (28.7)
Statin, n (%)3446 (14.5)2387 (14.4)1059 (14.9)
Diuretic, n (%)14,714 (61.9)10,287 (61.9)4427 (62.1)
Spironolactone, n (%)4927 (20.7)3427 (20.6)1500 (21.0)
Metformin, n (%)3459 (14.6)2447 (14.7)1012 (14.2)
Sulfonylurea, n (%)2214 (9.3)1533 (9.2)681 (9.6)
Meglitinide, n (%)2150 (9.0)1495 (9.0)655 (9.2)
SGLT2 inhibitor, n (%)47 (0.2)33 (0.2)14 (0.2)
GLP1 receptor agonist, n (%)3 (0.0)3 (0.0)0 (0.0)
Dipeptidyl peptidase-4 inhibitor, n (%)2720 (11.4)1883 (11.3)837 (11.7)
Thiazolidinedione, n (%)283 (1.2)203 (1.2)80 (1.1)
Alpha-glucosidase inhibitor, n (%)1084 (4.6)744 (4.5)340 (4.8)
Insulin, n (%)11,163 (47.0)7810 (47.0)3353 (47.0)
NSAID, n (%)11,300 (47.6)7917 (47.6)3383 (47.5)
COX-2 inhibitor, n (%)3316 (14.0)2284 (13.7)1032 (14.5)
Proton pump inhibitor, n (%)13,642 (57.4)9506 (57.2)4136 (58.0)
Steroid, n (%)8227 (34.6)5781 (34.8)2446 (34.3)
Allopurinol, n (%)1583 (6.7)1110 (6.7)473 (6.6)
Febuxostat, n (%)1446 (6.1)1006 (6.0)440 (6.2)
Benzbromarone, n (%)1424 (6.0)1007 (6.1)417 (5.8)
Class/Outcome
Rehospitalization with AKI †8756 (36.9)6076 (36.5)2680 (37.6)
Data are presented as n (%) or median and interquartile range. † AKI was defined based on the criteria from Kidney Disease: Improving Global Outcomes. Abbreviations: CHF, congestive heart failure; COPD, chronic obstructive pulmonary disease; PAOD, peripheral arterial occlusive disease; ICU, intensive care unit; HCO3, bicarbonate; HbA1c, hemoglobin A1c; LDL-C, low-density lipoprotein cholesterol; HDL-C, high-density lipoprotein-cholesterol; INR, international normalized ratio; aPTT, activated partial thromboplastin time; LDH, lactate dehydrogenase; NT-proBNP, N-terminal pro-brain natriuretic peptide; ACEI, angiotensin converting enzyme inhibitors; ARB, angiotensin receptor blocker; CCB, calcium channel blocker; DOAC, direct oral anticoagulant; SGLT2, sodium-glucose cotransporter 2 inhibitor; GLP1, glucagon-like peptide-1; NSAID, nonsteroidal anti-inflammatory drug; COX-2, cyclooxygenase-2; AKI, acute kidney injury.
Table
Table 2. Clinical features selected in machine learning algorithm.
Table 2. Clinical features selected in machine learning algorithm.
DemographicsComorbiditiesLaboratory DataContainment Medications
AgeHypertensionBlood urea nitrogenACEI
GenderTransient ischemic attackCreatinineARB
SmokingIschemic strokeWhite blood cell countsAlpha blocker
Alcohol consumptionHemorrhagic strokeHemoglobinBeta blocker
DementiaSodiumCCB
Diabetes mellitusPotassiumWarfarin
GoutChlorideDOAC
Myocardial infarctionCalciumAspirin
Coronary artery diseasePhosphatePlavix
CHFHCO3Nitrate
Atrial fibrillationC-reactive proteinStatin
Chronic liver diseaseAlbuminDiuretic
CirrhosisAlanine transaminaseSpironolactone
Peptic ulcer diseaseAspartate transaminaseMetformin
COPDAlkaline phosphataseSulfonylurea
AsthmaGamma-glutamyl transferaseMeglitinide
PAODTotal bilirubinSGLT2 inhibitor
Autoimmune diseaseHbA1cGLP1 receptor agonist
CancerGlucoseDPP4 inhibitor
Valvular heart diseaseUric acidThiazolidinedione
ICU admissionCholesterolAlpha-glucosidase inhibitor
Use of mechanical ventilatorsLDL-CInsulin
Use of inotropesHDL-CNSAID
INRCOX-2 inhibitor
aPTTProton pump inhibitor
D-dimerSteroid
LDHAllopurinol
NT-proBNPFebuxostat
Benzbromarone
Abbreviations: CHF, congestive heart failure; COPD, chronic obstructive pulmonary disease; PAOD, peripheral arterial occlusive disease; ICU, intensive care unit; HCO3, bicarbonate; HbA1c, hemoglobin A1c; LDL-C, low-density lipoprotein cholesterol; HDL-C, high-density lipoprotein-cholesterol; INR, international normalized ratio; aPTT, activated partial thromboplastin time; LDH, lactate dehydrogenase; NT-proBNP, N-terminal pro-brain natriuretic peptide; ACEI, angiotensin converting enzyme inhibitors; ARB, angiotensin receptor blocker; CCB, calcium channel blocker; DOAC, direct oral anticoagulant; SGLT2, sodium-glucose cotransporter 2 inhibitor; GLP1, glucagon-like peptide-1; DPP4, Dipeptidyl peptidase-4; NSAID, nonsteroidal anti-inflammatory drug; COX-2, cyclooxygenase-2.
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Ou, S.-M.; Lee, K.-H.; Tsai, M.-T.; Tseng, W.-C.; Chu, Y.-C.; Tarng, D.-C. Artificial Intelligence for Risk Prediction of Rehospitalization with Acute Kidney Injury in Sepsis Survivors. J. Pers. Med. 2022, 12, 43. https://doi.org/10.3390/jpm12010043

AMA Style

Ou S-M, Lee K-H, Tsai M-T, Tseng W-C, Chu Y-C, Tarng D-C. Artificial Intelligence for Risk Prediction of Rehospitalization with Acute Kidney Injury in Sepsis Survivors. Journal of Personalized Medicine. 2022; 12(1):43. https://doi.org/10.3390/jpm12010043

Chicago/Turabian Style

Ou, Shuo-Ming, Kuo-Hua Lee, Ming-Tsun Tsai, Wei-Cheng Tseng, Yuan-Chia Chu, and Der-Cherng Tarng. 2022. "Artificial Intelligence for Risk Prediction of Rehospitalization with Acute Kidney Injury in Sepsis Survivors" Journal of Personalized Medicine 12, no. 1: 43. https://doi.org/10.3390/jpm12010043

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop