A Machine Learning Approach for the Prediction of Severe Acute Kidney Injury Following Traumatic Brain Injury

Background Acute kidney injury (AKI), a prevalent non-neurological complication following traumatic brain injury (TBI), is a major clinical issue with an unfavorable prognosis. This study aimed to develop and validate machine learning models to predict severe AKI (stage 3 or greater) incidence in patients with TBI. Methods A retrospective cohort study was conducted by using two public databases: the Medical Information Mart for Intensive Care IV (MIMIC)-IV and the eICU Collaborative Research Database (eICU-CRD). Recursive feature elimination was used to select candidate predictors obtained within 24 h of intensive care unit admission. The area under the curve and decision curve analysis curves were used to determine the discriminatory ability. On the other hand, the calibration curve was employed to evaluate the calibrated performance of the newly developed machine learning models. Results In the MIMIC-IV database, there were 808 patients diagnosed with moderate and severe TBI (msTBI) (msTBI is defined as Glasgow Coma Score < 12). Of these, 60 (7.43%) patients experienced severe AKI. External validation in the eICU-CRD indicated that the random forest (RF) model had the highest area under the curve of 0.819 (95% confidence interval 0.783-0.851). Furthermore, in the calibration curve, the RF model was well calibrated (P = 0.795). Conclusions In this study, the RF model demonstrated better discrimination in predicting severe AKI than other models. An online calculator could facilitate its application, potentially improving the early detection of severe AKI and subsequently improving the clinical outcomes among patients with msTBI.