Machine Learning for Predicting Discharge Disposition After Traumatic Brain Injury
Nihal Satyadev, Pranav Warman, Andreas Seas, Brad J. Kolls, Michael M. Haglund, Anthony T. Fuller, Timothy Dunn
Abstract
BACKGROUND: Current traumatic brain injury (TBI) prognostic calculators are commonly used to predict the mortality and Glasgow Outcome Scale, but these outcomes are most relevant for severe TBI. Because mild and moderate TBI rarely reaches severe outcomes, there is a need for novel prognostic endpoints. OBJECTIVE: To generate machine learning (ML) models with a strong predictive capacity for trichotomized discharge disposition, an outcome not previously used in TBI prognostic models. The outcome can serve as a proxy for patients' functional status, even in mild and moderate patients with TBI. METHODS: Using a large data set (n = 5292) of patients with TBI from a quaternary care center and 84 predictors, including vitals, demographics, mechanism of injury, initial Glasgow Coma Scale, and comorbidities, we trained 6 different ML algorithms using a nested-stratified-cross-validation protocol. After optimizing hyperparameters and performing model selection, isotonic regression was applied to calibrate models. RESULTS: When maximizing the microaveraged area under the receiver operating characteristic curve during hyperparameter optimization, a random forest model exhibited top performance. A random forest model was also selected when maximizing the microaveraged area under the precision-recall curve. For both models, the weighted average area under the receiver operating characteristic curves was 0.84 (95% CI 0.81-0.87) and the weighted average area under the precision-recall curves was 0.85 (95% CI 0.82-0.88). CONCLUSION: Our group presents high-performing ML models to predict trichotomized discharge disposition. These models can assist in optimization of patient triage and treatment, especially in cases of mild and moderate TBI.