Machine learning for the prediction of acute kidney injury in critical care patients with acute cerebrovascular disease

PURPOSE: Acute kidney injury (AKI) is a common complication and associated with a poor clinical outcome. In this study, we developed and validated a model for predicting the risk of AKI through machine learning methods in critical care patients with acute cerebrovascular disease. METHODS: This study was a retrospective study based on two different cohorts. Five machine learning methods were used to develop AKI risk prediction models. We used six popular metrics (AUROC, F2-Score, accuracy, sensitivity, specificity and precision) to evaluate the performance of these models. RESULTS: We identified 2935 patients in the MIMIC-III database and 499 patients in our local database to develop and validate the AKI risk prediction model. The incidence of AKI in these two different cohorts was 18.3% and 61.7%, respectively. Analysis showed that several laboratory parameters (serum creatinine, hemoglobin, white blood cell count, bicarbonate, blood urea nitrogen, sodium, albumin, and platelet count), age, and length of hospital stay, were the top ten important factors associated with AKI. The analysis demonstrated that the XGBoost had higher AUROC (0.880, 95%CI: 0.831–0.929), indicating that the XGBoost model was better at predicting AKI risk in patients with acute cerebrovascular disease than other models. CONCLUSIONS: This study developed machine learning methods to identify critically ill patients with acute cerebrovascular disease who are at a high risk of developing AKI. This result suggested that machine learning techniques had the potential to improve the prediction of AKI risk models in critical care.