- 1Department of Gastroenterology, Peking University Third Hospital, Beijing, China
- 2Department of Internal Medicine, Wuhan University of Technology Hospital, Wuhan, China
Background: Acute kidney injury (AKI) is a common and important complication in patients with gastrointestinal bleeding who are admitted to the intensive care unit. The present study proposes an artificial intelligence solution for acute kidney injury prediction in patients with gastrointestinal bleeding admitted to the intensive care unit.
Methods: Data were collected from the eICU Collaborative Research Database (eICU-CRD) and Medical Information Mart for Intensive Care-IV (MIMIC-IV) database. The prediction model was developed using the extreme gradient boosting (XGBoost) model. The area under the receiver operating characteristic curve, accuracy, precision, area under the precision–recall curve (AUC-PR), and F1 score were used to evaluate the predictive performance of each model.
Results: Logistic regression, XGBoost, and XGBoost with severity scores were used to predict acute kidney injury risk using all features. The XGBoost-based acute kidney injury predictive models including XGBoost and XGBoost+severity scores model showed greater accuracy, recall, precision AUC, AUC-PR, and F1 score compared to logistic regression.
Conclusion: The XGBoost model obtained better risk prediction for acute kidney injury in patients with gastrointestinal bleeding admitted to the intensive care unit than the traditional logistic regression model, suggesting that machine learning (ML) techniques have the potential to improve the development and validation of predictive models in patients with gastrointestinal bleeding admitted to the intensive care unit.
Introduction
Acute kidney injury (AKI) is a common morbidity with a high incidence in patients admitted to the intensive care unit (ICU). It is associated with significant mortality, and a considerable proportion of patients develop AKI that progresses to chronic kidney disease (1–3). AKI has often been reported to occur in patients with gastrointestinal bleeding (GIB), especially those admitted to the ICU due to massive blood loss, leading to renal hypoperfusion secondary to intravascular volume depletion and eventually AKI (4, 5). AKI has been reported to develop in 1–11.4% of patients with acute GIB (6, 7). A systematic review aimed to explore the incidence and mortality of renal dysfunction in cirrhotic patients with acute GIB revealed that the pooled incidence of AKI was 25% (8).
For critically ill patients with GIB concomitant with AKI, hospitalization times may be prolonged, and costs will greatly increase, bringing a heavy burden to the medical system (9–11). Approximately 20% of patients with severe GIB and new-onset AKI can restore normal renal function if appropriate and effective interventions are performed on time (12). However, the lack of early prediction tools for AKI is a major challenge for ICU clinicians. Early recognition, risk assessment, and care for AKI can improve clinical outcomes and reduce the high healthcare costs of these patients. To assist physicians with risk assessment of AKI, various prediction models have been developed across various patient populations with varying degrees of predictive accuracy.
Models being built using machine learning (ML), which are mathematical models to make decisions and predictions based on datasets, have become popular, and ML techniques have been widely used clinically for prognosis prediction, including AKI (13). ML has shown better performance and low error rates in predicting clinical outcomes compared to traditional prediction tools such as logistic regression and Cox regression analysis. Moreover, ML has been widely used clinically to predict survival (14). Extreme gradient boosting (XGBoost) is recognized as a more advanced ML algorithm with much higher prediction accuracy and operation efficiency and has been widely applied for diagnosis and prognostic prediction (15). Recently, the use of ML models for AKI prediction has been rapidly growing in different clinical settings. Yue et al. reported that the XGBoost model had the best predictive performance for AKI in critically ill patients with sepsis (16). Zhang et al. evaluated five machine learning methods including XGBoost, adaptive boosting, random forest, logistic regression, and multi-layer perception to develop AKI risk prediction models in critical care patients with acute cerebrovascular disease and found that the XGBoost model was better at predicting AKI risk in patients with acute cerebrovascular disease than other models (17). However, the efficacy of XGBoost in predicting AKI in critically ill patients with GIB remains unclear.
This study aimed to use XGBoost to construct a predictive model to evaluate AKI risk in critically ill patients with GIB and use the publicly available eICU Collaborative Research Database (eICU-CRD) as a data source for the training cohort and the Medical Information Mart for Intensive Care-IV (MIMIC-IV) database as a data source for the validation cohort. This study explored the accuracy of XGBoost for the construction of AKI prediction models and the extraction of important features. Furthermore, a shapely additive explanation (SHAP) analysis was used to reveal the influence of the major factors and provide comprehensive explanations of their quantitative impacts on output. In addition, the XGBoost model was compared with the traditional logistic model and score systems commonly used in the ICU, including the Oxford Acute Severity of Illness Score (OASIS), sequential organ failure assessment (SOFA), and acute physiology score III (APS III). The present study would provide a reference for an XGBoost-based clinical decision support system to aid the early prediction of AKI in patients with GIB admitted to the ICU setting.
Methods
Data source
All data were extracted from the eICU-CRD (18) and MIMIC-IV version 1.0 databases (19). The MIMIC-IV contains comprehensive and high-quality data of 524,520 admissions (including 257,366 patients) admitted to intensive care units (ICUs) at the Beth Israel Deaconess Medical Center during 2008–2019. The eICU-CRD covered 200,859 ICU admissions (including 139,367 patients) between 2014 and 2015 at 208 hospitals in the United States. The research use of these databases was approved by the institutional review board of the Massachusetts Institute of Technology. All procedures were performed in accordance with the ethical standards of the Declaration of Helsinki and its later amendments or comparable ethical standards. We obtained permission to extract data from the MIMIC-IV database and eICU-CRD database.
Cohort selection
GIB was defined according to the European Society of Gastrointestinal Endoscopy guideline (20, 21); AKI was diagnosed according to the KDIGO-AKI criteria based on serum creatinine in the first 48 h of ICU admission (22).
Patients with one of the following conditions were excluded: (1) an age of <18 years at first admission to the ICU, (2) a hospital stay of <48 h, (3) >70% of personal data missing, (4) repeated ICU admissions, and (5) a history of end-stage renal disease (ESRD). Finally, 6,679 patients with eICU-CRD and 2,968 patients with MIMIC-IV were included in this study. Moreover, patients from the eICU-CRD were randomly divided into training (n = 5,679) and internal validation cohorts (n = 1,000) at a ratio of 7:3. Patients from MIMIC-IV (n = 2,968) were used as an external validation set. A detailed flowchart is shown in Figure 1.
Data collection and outcomes
Baseline characteristics and admission information, including age, sex, body mass index, bleeding site, comorbidities, severity score, and drug usage, were recorded. Initial vital signs and laboratory results were also measured during the first 24 h of ICU admission (19).
The primary outcome was AKI based on the KDIGO guidelines for serum creatinine within 48 h.
Statistical analysis
For all continuous covariates, the mean values and standard deviations were reported, and categorical data were expressed as frequency (percentage). The chi-square test or Fisher's exact test was performed to compare differences between groups. Baseline characteristics were reported as training and validation cohorts. Baseline characteristics were compared using R software version 4.1.0. A P-value of < 0.05 was considered statistically significant. Modeling was performed using Python 3.6.4.
AKI prediction model
Logistic regression, XGBoost, and XGBoost+severity scores (SOFA, OASIS, and APS III) were applied to build the prediction models. The XGBoost model was used as previously reported (23, 24). Moreover, all the machine-learning algorithms were implemented using the “sklearn” machine-learning library of Python programming software. The detailed XGBoost parameters are shown in Supplementary Table 1. The framework of the prediction models is illustrated in Figure 2.
Performance evaluation
To assess and compare the predictive accuracy of XGBoost, XGBoost+severity scores, and logistic regression models, each model was assessed according to precision, recall, accuracy, F1 score, area under the receiver operating characteristic (AUC) curve, and area under precision–recall curve (AUC-PR) (25).
Shapely additive explanation (SHAP) analysis
To further analyze the positive and negative effects of the important features identified for AKI prediction and investigate the relationship between them, a SHAP analysis was performed using Python 3.7.0. The SHAP value is the assigned predicted value for each feature of the data (26).
Results
Baseline characteristics
The incidence rate of AKI was 15.0% in the training cohort, 13.7% in the internal validation cohort, and 30.7% in the external validation cohort. Table 1 shows the baseline characteristics of all patients in the training, internal validation, and external validation cohorts, as classified by NAKI and AKI.
Variable selection
The importance matrix plot for the XGBoost model is shown in Figure 3, revealing the top 15 most important variables that contribute to the model. Bilirubin (max) was the most important predictor variable for all prediction horizons, followed closely by bicarbonate (min), renal replacement therapy (RRT), mechanical ventilation, and bilirubin (first time).
![www.frontiersin.org](https://www.frontiersin.org/files/Articles/1221602/fmed-10-1221602-HTML/image_m/fmed-10-1221602-g003.jpg)
Figure 3. The top 15 features derived from the XGBoost model. Bilirubin_max, maximum serum bilirubin; bicarbonate_min, minimum bicarbonate; bilirubin_min, minimum bilirubin; RRT, renal replacement therapy; mechvent, mechanical ventilation; MAP, mean arterial pressure; CKD, chronic kidney disease.
Model performance
Three models, logistic regression, XGBoost, and XGBoost+severity scores, were used to predict AKI risk using all features. The accuracy, recall, precision, F1 score, AUC-PR, and AUC of XGBoost were higher than those of the logistic regression model. When XGBoost+severity scores (SOFA, OASIS, and APS III) were used, this model exhibited the best predictive ability compared to the XGBoost model only, as well as the logistic regression model with the highest accuracy, recall, precision, F1 score, AUC-PR, and AUC in the training cohort. The results in the internal and external validation cohorts were similar to the results in the training cohort (Table 2). Furthermore, ROC analysis was also performed to further check the performance of the three models, as shown in Figures 4A–C. The XGBoost+severity score model exhibited the largest AUC, followed by the XGBoost model, in all training, internal validation, and external validation cohorts.
![www.frontiersin.org](https://www.frontiersin.org/files/Articles/1221602/fmed-10-1221602-HTML/image_m/fmed-10-1221602-g004.jpg)
Figure 4. ROC curves of the prediction models using all features as well as three common severity scores for predicting AKI in the training set (A) and in the internal validation set (B) and in the external validation set (C).
The calibration curves for the predictive models (logistic regression model, XGBoost model, and XGBoost+severity scores model) all showed high agreement between the actual probability and predicted probability in the training, internal validation, and external validation sets (Figures 5A–C). Subsequently, a decision curve analysis (DCA) was performed to determine the net benefit and clinical utility of the predictive models. The DCA curve also indicated that the three predictive models were all clinically useful and that the benefit of using the XGBoost+severity score model was superior to that of using the XGBoost and logistic regression models in all sets (Figures 5D–F).
![www.frontiersin.org](https://www.frontiersin.org/files/Articles/1221602/fmed-10-1221602-HTML/image_m/fmed-10-1221602-g005.jpg)
Figure 5. The performance of the models of logistic, XGBoost, and XGBoost+severity scores for AKI. The calibration curves of the logistic, XGBoost, and XGBoost+severity scores for AKI in the training set (A), in the internal validation set (B), and in the external validation set (C). The decision curve analysis of the logistic, XGBoost, and XGBoost+severity scores for AKI in the training set (D), in the internal validation set (E), and in the external validation set (F).
SHAP analysis
To examine the influence of characteristics on the prediction results in more samples and analyze the similarities and differences in the important characteristics of patients with varying severities of AKI with different severities, a SHAP summary chart was used. As shown in Figure 6, bilirubin (max) ranked first in importance; the larger the bilirubin (max) in patients, the higher the probability of AKI development, suggesting that this indicator should be observed first in early prediction.
![www.frontiersin.org](https://www.frontiersin.org/files/Articles/1221602/fmed-10-1221602-HTML/image_m/fmed-10-1221602-g006.jpg)
Figure 6. SHAP summary plot of the features of the XGBoost model. The higher the SHAP value of a feature, the higher the probability of AKI development. A dot is created for each feature attribution value for the model of each patient, and thus one patient is allocated one dot on the line for each feature. Dots are colored according to the values of features for the respective patient and accumulate vertically to depict density. Red represents higher feature values, and blue represents lower feature values. Creatinine_max, maximum serum creatinine; first_time_creatinine, the first measurement of serum creatinine after their ICU admission; bilirubin_min, minimum bilirubin; mechvent, mechanical ventilation.
Using all features as an example in the XGBoost model, which has excellent performance for predicting AKI, as well as the SHAP analysis method, representative non-AKI and AKI patients were selected to illustrate the effect of features on prediction ability. As shown in Figure 7, for predicting non-AKI patients, mechanical ventilation (mechvent) played a major positive role in prediction results, sodium (min) played a major negative role in predicting outcomes, and the SHAP value of the final model predicted for this patient was −0.25, which is < 0 and, therefore, considered to have successfully predicted the absence of AKI. For predicting AKI patients, the bicarbonate plays a major positive role in prediction results, the bilirubin (max) plays a major negative role in predicting outcomes, and the SHAP value of the final model predicted for this patient was 1.23, which is considered to have successfully predicted AKI.
Discussion
Few studies have explored AKI-prediction models based on machine-learning techniques in critically ill GIB settings. The present study compared the predictive accuracy of the prediction of AKI in patients with GIB admitted to the ICU using the machine learning technique XGBoost, traditional statistical approach logistic regression analysis, and previous risk scoring models (SOFA, OASIS, and APS III). The results have shown that the XGBoost model had the largest AUC, accuracy, precision, and recall among all the techniques and risk scores. Moreover, the XGBoost+severity scores (SOFA, OASIS, and APS III) exhibited better AKI prediction performance than XGBoost. The XGBoost model-based prediction may induce a significant improvement in the prediction of AKI in patients with GIB admitted to the ICU. A risk estimator based on the XGBoost model was developed to determine the risk of AKI in high-risk patients with GIB.
Acute GIB is very common in patients in ICU (27). Mortality in patients with acute GIB is very high, approaching 48.5–65% (28, 29). According to previous reports, AKI occurs in ~25% of patients with acute GIB. Although AKI accounts for a small number of complications in critically ill patients with GIB, the mortality rate of critically ill patients with AKI is higher than that of patients with severe GIB without AKI. Xie et al. found that AKI occurred in 30% of patients with cirrhosis and that patients with cirrhosis and AKI had a worse prognosis (37 vs. 3%) (30). Moreover, a study by Kim et al. also showed that the 6-week mortality rate of cirrhotic patients with new-onset AKI was significantly higher than that of patients without AKI (31). The early identification of AKI can effectively prevent disease progression. However, there is currently a lack of reliable and effective predictive models for such patients, warranting researchers to develop a reliable AKI predictive model to identify high-risk critically ill patients with GIB.
With the advent of big data, ML has great potential in the field of AKI research owing to its unparalleled ability in data processing. Therefore, machine learning models may be powerful tools for AKI risk stratification and prediction (32). Several ML techniques have been used to predict AKI in different disease settings (33–35). However, the use of ML techniques to predict AKI in critically ill patients with GIB has not been investigated. As an ML technique, XGBoost is a highly efficient boosting ensemble learning model that originated in the decision tree model, using a tree classifier for better prediction results and higher operation efficiency (36). Several studies have found that XGBoost is superior to other machine learning techniques. Liu et al. reported that XGBoost exhibited the best performance in predicting mortality in patients with AKI in the ICU, with the highest AUC, F1 score, and accuracy compared with logistic regression, support vector machines, and random forest (37). Yue et al. aimed to establish and validate predictive models based on novel machine learning algorithms for AKI in critically ill patients with sepsis and found that the XGBoost model had the best predictive performance in terms of discrimination, calibration, and clinical application among all models, including logistic regression, SOFA, and the customized Simplified Acute Physiology Score (SAPS) II model (16). Qu et al. used support vector machine, random forest, classification and regression tree, and XGBoost models to predict AKI prediction, and compared to the predictive performance of the classic model using logistic regression, the results demonstrated that XGBoost performed best in predicting AKI among the machine learning models (34). Hence, the XGBoost algorithm was selected to structured and unstructured patient data from electronic medical records to develop an AKI prediction model in the present study. Consistent with previous reports (16, 34, 37), the XGBoost model was better than the traditional logistic regression model and previous risk scoring models (SOFA, OASIS, and APS III). The XGBoost+severity score model exhibited the highest accuracy, recall, precision, AUC, AUC-PR, and F1 score.
There are several studies in this respect. Although the studies have been conducted on different data and are not comparable, studies employed traditional ML techniques to predict AKI events, and XGBoost was the most commonly used algorithm. Using the MIMIC dataset for AKI prediction in the ICU setting, Zhang et al. (23) reported that XGBoost had a significantly greater ROC than the logistic regression model (0.86 vs. 0.728) in differentiating between volume-responsive and volume-unresponsive AKI. Zimmerman et al. (38) developed ML models to predict the new onset of AKI in critical care settings with a mean AUC of 0.783 by our all-feature, logistic-regression model. Sun et al. (39). used an ensemble learning algorithm for the early prediction of AKI with AUC 24 h ahead: 0.81, 48 h ahead: 0.78; MIMIC-III: AUC 24 h ahead: 0.95, and 48 h ahead: 0.95. In addition, Wang et al. (40) reported AUC above 0.83 with SVM as the best performer, and Qian et al. (41) reported that LightGBM had the best performance, with all evaluation indicators achieving the highest value (average AUC = 0.905, F1 = 0.897, recall = 0.836). Alfieri et al. (42) showed that AUC for deep learning is 0.907 and LR is 0.877. Shawwa et al. (43) indicated that a 30-feature model showed 0.690 in the Mayo Clinic cohort set and 0.656 in the MIMIC-III cohort. Because of different datasets (MIMIV-IV in our study and MIMIC-III in previous studies), comparing our results with other studies is difficult. However, in general, in comparison with the best results from previous studies, we also achieved a high AUC (XGBoost+severity scores: 0.89).
A SHAP analysis was used to determine the quantitative impact of each feature on AKI prediction based on SHAP values. The results of our study demonstrated that bilirubin and albumin were the most influential feature among all other physiological measurements A UK-wide study in acute medical units aimed to investigate patients who were at risk of developing AKI in hospitals and found that elevated serum bilirubin was independently associated with AKI development (44). Moreover, Wang et al. indicated that lower serum albumin levels were independently associated with a greater risk of contrast-induced AKI among patients who underwent percutaneous coronary intervention (45). Moreover, mechanical ventilation (mechvent), bicarbonate, and RRT also displayed strong predictive powers, which reflected their roles in AKI prediction in critically ill patients with GIB.
Nevertheless, this study has some limitations. First, the present study extracted data from two large public databases, and additional external clinical datasets may be needed to verify the results of this study. Second, we collected data during the first 24 h of ICU stay, and more dynamic time-point data are needed in future studies. Moreover, the variables we stated indicate that the predictive model's utilities are challenging, as they are at different time points (e.g., patients' first creatinine and highest bilirubin). Therefore, in reality, it would not be possible to use them to predict AKI risk until all time-point data were collected. Finally, the present study included an imbalanced dataset to check the performance of the machine learning and the predictive model developed using the machine learning algorithms could be biased and inaccurate. The results of this study should be further validated in the future.
Conclusion
This study utilized an XGBoost-based model to predict AKI in patients with GIB admitted to the ICU. The results demonstrated that it is feasible to apply the XGBoost-based prediction models for the management of critically ill patients with GIB and that this model has better predictive performance than that of classic logistic regression methods and severity score models. The XGBoost-based model in this study has not been verified by an external cohort, and further studies are needed to determine the clinical application of the XGBoost-based model and to perform prospective and large sample experiments to verify our conclusion.
Data availability statement
The raw data supporting the conclusions of this article will be made available by the authors, without undue reservation.
Ethics statement
Research use of MIMIC-IV and eICU-CRD were approved by the Institutional Review Board (IRB) of the Massachusetts Institute of Technology (MIT). All procedures were performed according to the ethical standards of the Helsinki Declaration and its later amendments or comparable ethical standards. The studies were conducted in accordance with the local legislation and institutional requirements. Written informed consent for participation was not required from the participants or the participants' legal guardians/next of kin in accordance with the national legislation and institutional requirements.
Author contributions
HS conceived and designed the study. YS extracted the data. HS and YS analyzed the data and drafted the manuscript. LL takes responsibility for the content of the manuscript including the data and analysis. All authors have approved the final version of the manuscript for submission and agree to be accountable for all aspects of the manuscript.
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.
Supplementary material
The Supplementary Material for this article can be found online at: https://www.frontiersin.org/articles/10.3389/fmed.2023.1221602/full#supplementary-material
Abbreviations
AKI, acute kidney injury; ICU, intensive care unit; GIB, gastrointestinal bleeding; ML, machine learning; eICU-CRD, eICU Collaborative Research Database; MIMIC-IV, Medical Information Mart for Intensive Care-IV; SOFA, sequential organ failure assessment; OASIS, Oxford Acute Severity of Illness Score; APSIII, acute physiology score III; AUC, area under the receiver operating characteristic; SHAP, shapely additive explanation; DCA, decision curve analysis; NB, Naïve Bayes; RF, random forest; SVM, support vector machine; LR, logistic regression; CNN, convolutional neural network; GB, gradient boosting; MLP, multi-layer perceptron; SBP, systolic blood pressure; DBP, diastolic blood pressure; INR, international normalized ratio.
References
1. Kellum JA, Prowle JR. Paradigms of acute kidney injury in the intensive care setting. Nat Rev Nephrol. (2018) 14:217–30. doi: 10.1038/nrneph.2017.184
2. Mehta RL, Pascual MT, Soroko S, Savage BR, Himmelfarb J, Ikizler TA, et al. Spectrum of acute renal failure in the intensive care unit: the PICARD experience. Kidney Int. (2004) 66:1613–21. doi: 10.1111/j.1523-1755.2004.00927.x
3. Koeze J, Keus F, Dieperink W, van der Horst IC, Zijlstra JG, van Meurs M. Incidence, timing and outcome of AKI in critically ill patients varies with the definition used and the addition of urine output criteria. BMC Nephrol. (2017) 18:70. doi: 10.1186/s12882-017-0487-8
4. Cardenas A, Gines P, Uriz J, Bessa X, Salmerón JM, Mas A, et al. Renal failure after upper gastrointestinal bleeding in cirrhosis: incidence, clinical course, predictive factors, and short-term prognosis. Hepatology. (2001) 34:671–6. doi: 10.1016/S0168-8278(01)80229-4
5. Belcher JM, Parikh CR, Garcia-Tsao G. Acute kidney injury in patients with cirrhosis: perils and promise. Clin Gastroenterol Hepatol. (2013) 11:1550–8. doi: 10.1016/j.cgh.2013.03.018
6. del Olmo JA, Pena A, Serra MA, Wassel AH, Benages A, Rodrigo JM. Predictors of morbidity and mortality after the first episode of upper gastrointestinal bleeding in liver cirrhosis. J Hepatol. (2000) 32:19–24. doi: 10.1016/S0168-8278(01)68827-5
7. Kang JY. The gastrointestinal tract in uremia. Dig Dis Sci. (1993) 38:257–68. doi: 10.1007/BF01307542
8. Bai Z, Primignani M, Guo X, Zheng K, Li H, Qi X. Incidence and mortality of renal dysfunction in cirrhotic patients with acute gastrointestinal bleeding: a systematic review and meta-analysis. Expert Rev Gastroenterol Hepatol. (2019) 13:1181–8. doi: 10.1080/17474124.2019.1694904
9. Hobson C, Ozrazgat-Baslanti T, Kuxhausen A, Thottakkara P, Efron PA, Moore FA, et al. Cost and mortality associated with postoperative acute kidney injury. Ann Surg. (2015) 261:1207–14. doi: 10.1097/SLA.0000000000000732
10. Collister D, Pannu N, Ye F, James M, Hemmelgarn B, Chui B, et al. Health care costs associated with AKI. Clin J Am Soc Nephrol. (2017) 12:1733–43. doi: 10.2215/CJN.00950117
11. Xu X, Liu B, Lin S, Li B, Wu Y, Li Y, et al. Terlipressin may decrease in-hospital mortality of cirrhotic patients with acute gastrointestinal bleeding and renal dysfunction: a retrospective multicenter observational study. Adv Ther. (2020) 37:4396–413. doi: 10.1007/s12325-020-01466-z
12. Mayor S. UK report into acute kidney injury deaths urges electrolyte checks in all emergency admissions. Br Med J. (2009) 338:1407. doi: 10.1136/bmj.b1710
13. Mo X, Chen X, Li H, Li J, Zeng F, Chen Y, et al. Early and accurate prediction of clinical response to methotrexate treatment in juvenile idiopathic arthritis using machine learning. Front Pharmacol. (2019) 10:1155. doi: 10.3389/fphar.2019.01155
14. Song X, Liu X, Liu F, Wang C. Comparison of machine learning and logistic regression models in predicting acute kidney injury: a systematic review and meta-analysis. Int J Med Inform. (2021) 151:104484. doi: 10.1016/j.ijmedinf.2021.104484
15. Sheridan RP, Wang WM, Liaw A, Ma J, Gifford EM. Extreme gradient boosting as a method for quantitative structure-activity relationships. J Chem Inf Model. (2016) 56:2353–60. doi: 10.1021/acs.jcim.6b00591
16. Yue S, Li S, Huang X, Liu J, Hou X, Zhao Y, et al. Machine learning for the prediction of acute kidney injury in patients with sepsis. J Transl Med. (2022) 20:215. doi: 10.1186/s12967-022-03364-0
17. Zhang X, Chen S, Lai K, Chen Z, Wan J, Xu Y. Machine learning for the prediction of acute kidney injury in critical care patients with acute cerebrovascular disease. Ren Fail. (2022) 44:43–53. doi: 10.1080/0886022X.2022.2036619
18. Pollard TJ, Johnson AEW, Raffa JD, Celi LA, Mark RG, Badawi O. The eICU Collaborative Research Database, a freely available multi-center database for critical care research. Sci Data. (2018) 5:180178. doi: 10.1038/sdata.2018.178
19. Liu T, Zhao Q, Du B. Effects of high-flow oxygen therapy on patients with hypoxemia after extubation and predictors of reintubation: a retrospective study based on the MIMIC-IV database. BMC Pulm Med. (2021) 21:160. doi: 10.1186/s12890-021-01526-2
20. Gralnek IM, Stanley AJ, Morris AJ, Camus M, Lau J, Lanas A, et al. Endoscopic diagnosis and management of nonvariceal upper gastrointestinal hemorrhage (NVUGIH): European Society of Gastrointestinal Endoscopy (ESGE) Guideline- Update 2021. Endoscopy. (2021) 53:300–32. doi: 10.1055/a-1369-5274
21. Triantafyllou K, Gkolfakis P, Gralnek IM, Oakland K, Manes G, Radaelli F, et al. Diagnosis and management of acute lower gastrointestinal bleeding: European Society of Gastrointestinal Endoscopy (ESGE) Guideline. Endoscopy. (2021) 53:850–68. doi: 10.1055/a-1496-8969
22. Kellum JA, Lameire N. Diagnosis, evaluation, and management of acute kidney injury: a KDIGO summary (part 1). Crit Care. (2013) 17:204. doi: 10.1186/cc11454
23. 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. doi: 10.1186/s13054-019-2411-z
24. Tian X, Chong Y, Huang Y, Guo P, Li M, Zhang W, et al. Using machine learning algorithms to predict hepatitis B surface antigen seroclearance. Comput Math Methods Med. (2019) 2019:6915850. doi: 10.1155/2019/6915850
25. Kumar Y, Koul A, Singla R, Ijaz MF. Artificial intelligence in disease diagnosis: a systematic literature review, synthesizing framework and future research agenda. J Ambient Intell Humaniz Comput. (2023) 14:8459–86. doi: 10.1007/s12652-021-03612-z
26. Gao W, Wang J, Zhou L, Luo Q, Lao Y, Lyu H, et al. Prediction of acute kidney injury in ICU with gradient boosting decision tree algorithms. Comput Biol Med. (2021) 140:105097. doi: 10.1016/j.compbiomed.2021.105097
27. Kumar S, Ramos C, Garcia-Carrasquillo RJ, Green PH, Lebwohl B. Incidence and risk factors for gastrointestinal bleeding among patients admitted to medical intensive care units. Frontline Gastroenterol. (2017) 8:167–73. doi: 10.1136/flgastro-2016-100722
28. Ye Z, Reintam Blaser A, Lytvyn L, Wang Y, Guyatt GH, Mikita JS, et al. Gastrointestinal bleeding prophylaxis for critically ill patients: a clinical practice guideline. Br Med J. (2020) 368:l6722. doi: 10.1136/bmj.l6722
29. Cook DJ, Fuller HD, Guyatt GH, Marshall JC, Leasa D, Hall R, et al. Risk factors for gastrointestinal bleeding in critically ill patients. Can Crit Care Trials Group N Engl J Med. (1994) 330:377–81. doi: 10.1056/NEJM199402103300601
30. Hsieh YC, Lee KC, Chen PH, Su CW, Hou MC, Lin HC. Acute kidney injury predicts mortality in cirrhotic patients with gastric variceal bleeding. J Gastroenterol Hepatol. (2017) 32:1859–66. doi: 10.1111/jgh.13777
31. Kim JH, Im CB, Lee SS, Jeon H, Choi JW, Kim HJ, et al. Impact of acute kidney injury on mortality in patients with acute variceal bleeding. BMC Gastroenterol. (2021) 21:290. doi: 10.1186/s12876-021-01862-x
32. Sutherland SM, Goldstein SL, Bagshaw SM. Acute kidney injury and big data. Contrib Nephrol. (2018) 193:55–67. doi: 10.1159/000484963
33. Tseng PY, Chen YT, Wang CH, Chiu KM, Peng YS, Hsu SP, et al. Prediction of the development of acute kidney injury following cardiac surgery by machine learning. Crit Care. (2020) 24:478. doi: 10.1186/s13054-020-03179-9
34. Qu C, Gao L, Yu XQ, Wei M, Fang GQ, He J, et al. Machine learning models of acute kidney injury prediction in acute pancreatitis patients. Gastroenterol Res Pract. (2020) 2020:3431290. doi: 10.1155/2020/3431290
35. Lee Y, Ryu J, Kang MW, Seo KH, Kim J, Suh J, et al. Machine learning-based prediction of acute kidney injury after nephrectomy in patients with renal cell carcinoma. Sci Rep. (2021) 11:15704. doi: 10.1038/s41598-021-95019-1
36. Hsu CN, Liu CL, Tain YL, Kuo CY, Lin YC. Machine learning model for risk prediction of community acquired acute kidney injury hospitalization from electronic health records: development and validation study. J Med Internet Res. (2020) 22:e16903. doi: 10.2196/16903
37. Liu J, Wu J, Liu S, Li M, Hu K, Li K. Predicting mortality of patients with acute kidney injury in the ICU using XGBoost model. PLoS ONE. (2021) 16:e0246306. doi: 10.1371/journal.pone.0246306
38. Zimmerman LP, Reyfman PA, Smith ADR, Zeng Z, Kho A, Sanchez-Pinto LN, 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. doi: 10.1186/s12911-019-0733-z
39. Sun M, Baron J, Dighe A, Szolovits P, Wunderink RG, Isakova T, et al. Early prediction of acute kidney injury in critical care setting using clinical notes and structured multivariate physiological measurements. Stud Health Technol Inform. (2019) 264:368–72. doi: 10.3233/SHTI190245
40. Wang Y, Wei Y, Yang H, Li J, Zhou Y, Wu Q. Utilizing imbalanced electronic health records to predict acute kidney injury by ensemble learning and time series model. BMC Med Inform Decis Mak. (2020) 20:238. doi: 10.1186/s12911-020-01245-4
41. Qian Q, Wu J, Wang J, Sun H, Yang L. Prediction models for AKI in ICU: a comparative study. Int J Gen Med. (2021) 14:623–32. doi: 10.2196/preprints.18257
42. Alfieri F, Ancona A, Tripepi G, Randazzo V, Paviglianiti A, Pasero E, et al. External validation of a deep-learning model to predict severe acute kidney injury based on urine output changes in critically ill patients. J Nephrol. (2022) 35:2047–56. doi: 10.1007/s40620-022-01335-8
43. Shawwa K, Ghosh E, Lanius S, Schwager E, Eshelman L, Kashani KB. Predicting acute kidney injury in critically ill patients using comorbid conditions utilizing machine learning. Clin Kidney J. (2020) 14:1428–35. doi: 10.1093/ckj/sfaa145
44. Risk Investigators. Risk prediction for acute kidney injury in acute medical admissions in the UK. QJM. (2019) 112:197–205. doi: 10.1093/qjmed/hcy277
Keywords: XGBoost, acute kidney injury, gastrointestinal bleeding, intensive care unit, machine learning
Citation: Shi H, Shen Y and Li L (2023) Early prediction of acute kidney injury in patients with gastrointestinal bleeding admitted to the intensive care unit based on extreme gradient boosting. Front. Med. 10:1221602. doi: 10.3389/fmed.2023.1221602
Received: 12 May 2023; Accepted: 07 August 2023;
Published: 31 August 2023.
Edited by:
Mathieu Jozwiak, Centre Hospitalier Universitaire de Nice, FranceReviewed by:
Romain Lombardi, Centre Hospitalier Universitaire de Nice, FranceMiodrag Zivkovic, Singidunum University, Serbia
Copyright © 2023 Shi, Shen and Li. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.
*Correspondence: Lu Li, luli307034153@126.com
†These authors have contributed equally to this work