Enhancing hospital course and outcome prediction in patients with traumatic brain injury: A machine learning study

Abstract

Purpose We aimed to use machine learning (ML) algorithms with clinical, lab, and imaging data as input to predict various outcomes in traumatic brain injury (TBI) patients. Methods In this retrospective study, blood samples were analyzed for glial fibrillary acidic protein (GFAP) and ubiquitin C-terminal hydrolase L1 (UCH-L1). The non-contrast head CTs were reviewed by two neuroradiologists for TBI common data elements (CDE). Three outcomes were designed to predict: discharged or admitted for further management (prediction 1), deceased or not deceased (prediction 2), and admission only, prolonged stay, or neurosurgery performed (prediction 3). Five ML models were trained. SHapley Additive exPlanations (SHAP) analyses were used to assess the relative significance of variables. Results Four hundred forty patients were used to predict predictions 1 and 2, while 271 patients were used in prediction 3. Due to Prediction 3’s hospitalization requirement, deceased and discharged patients could not be utilized. The Random Forest model achieved an average accuracy of 1.00 for prediction 1 and an accuracy of 0.99 for prediction 2. The Random Forest model achieved a mean accuracy of 0.93 for prediction 3. Key features were extracranial injury, hemorrhage, UCH-L1 for prediction 1; The Glasgow Coma Scale, age, GFAP for prediction 2; and GFAP, subdural hemorrhage volume, and pneumocephalus for prediction 3, per SHAP analysis. Conclusion Combining clinical and laboratory parameters with non-contrast CT CDEs allowed our ML models to accurately predict the designed outcomes of TBI patients. GFAP and UCH-L1 were among the significant predictor variables, demonstrating the importance of these biomarkers.