Abstract:The difficulty of accurately identifying patients who would benefit from promising treatments makes it challenging to prove the efficacy of novel treatments for traumatic brain injury (TBI). Although machine learning is being increasingly applied to this task, existing binary outcome prediction models are insufficient for the effective stratification of TBI patients. The aim of this study was to develop an accurate 3-class outcome prediction model to enable appropriate patient stratification. To this end, retrospective balanced data of 1200 blunt TBI patients admitted to six Japanese hospitals from January 2018 onwards (200 consecutive cases at each institution) were used for model training and validation. We incorporated 21 predictors obtained in the emergency department, including age, sex, six clinical findings, four laboratory parameters, eight computed tomography findings, and an emergency craniotomy. We developed two machine learning models (XGBoost and dense neural network) and logistic regression models to predict 3-class outcomes based on the Glasgow Outcome Scale-Extended (GOSE) at discharge. The prediction models were developed using a training dataset with n = 1000, and their prediction performances were evaluated over two validation rounds on a validation dataset (n = 80) and a test dataset (n = 120) using the bootstrap method. Of the 1200 patients in aggregate, the median patient age was 71 years, 199 (16.7%) exhibited severe TBI, and emergency craniotomy was performed on 104 patients (8.7%). The median length of stay was 13.0 days. The 3-class outcomes were good recovery/moderate disability for 709 patients (59.1%), severe disability/vegetative state in 416 patients (34.7%), and death in 75 patients (6.2%). XGBoost model performed well with 69.5% sensitivity, 82.5% accuracy, and an area under the receiver operating characteristic curve of 0.901 in the final validation. In terms of the receiver operating characteristic curve analysis, the XGBoost outperformed the neural network-based and logistic regression models slightly. In particular, XGBoost outperformed the logistic regression model significantly in predicting severe disability/vegetative state. Although each model predicted favorable outcomes accurately, they tended to miss the mortality prediction. The proposed machine learning model was demonstrated to be capable of accurate prediction of in-hospital outcomes following TBI, even with the three GOSE-based categories. As a result, it is expected to be more impactful in the development of appropriate patient stratification methods in future TBI studies than conventional binary prognostic models. Further, outcomes were predicted based on only clinical data obtained from the emergency department. However, developing a robust model with consistent performance in diverse scenarios remains challenging, and further efforts are needed to improve generalization performance.

Development of a Multimodal Machine Learning-Based Prognostication Model for Traumatic Brain Injury Using Clinical Data and Computed Tomography Scans: A CENTER-TBI and CINTER-TBI Study

Enhancing Hospital Course and Outcome Prediction in Patients with Traumatic Brain Injury: A Machine Learning Study.

Evaluation of models that predict short-term outcome after traumatic brain injury

Validation of a Deep Learning Model for Traumatic Brain Injury Detection and NIRIS Grading on Non-Contrast CT: a Multi-Reader Study with Promising Results and Opportunities for Improvement.

Machine Learning to Predict Three Types of Outcomes After Traumatic Brain Injury Using Data at Admission: A Multi-Center Study for Development and Validation

Prognosticating outcome using magnetic resonance imaging in patients with moderate to severe traumatic brain injury: a machine learning approach

Predictive utility of an adapted Marshall head CT classification scheme after traumatic brain injury

Machine Learning Applications in Traumatic Brain Injury: A Spotlight on Mild TBI

Automatic lesion analysis for increased efficiency in outcome prediction of traumatic brain injury

445 Performance of the IMPACT and CRASH Prognostic Models for Traumatic Brain Injury in a Contemporary Multicenter Cohort: A TRACK-TBI Study

Automated Measurement of Cerebral Hemorrhagic Contusions and Outcomes After Traumatic Brain Injury in the TRACK-TBI Study

A Machine Learning Model Based on CT Imaging Metrics and Clinical Features to Predict the Risk of Hospital-Acquired Pneumonia After Traumatic Brain Injury

Machine learning models for predicting in-hospital outcomes after non-surgical treatment among patients with moderate-to-severe traumatic brain injury

Mixture Model Framework for Traumatic Brain Injury Prognosis Using Heterogeneous Clinical and Outcome Data

A Machine Learning-Based Approach to Predict Prognosis and Length of Hospital Stay in Adults and Children with Traumatic Brain Injury: Retrospective Cohort Study

Dynamic prediction of mortality after traumatic brain injury using a machine learning algorithm

Predicting Mortality and Functional Status Scores of Traumatic Brain Injury Patients using Supervised Machine Learning

Machine Learning Algorithm Predicts Mortality Risk in Intensive Care Unit for Patients with Traumatic Brain Injury

Development and validation of a nomogram for predicting mortality in patients with acute severe traumatic brain injury: A retrospective analysis

Mortality Prediction in Severe Traumatic Brain Injury Using Traditional and Machine Learning Algorithms.

A Machine Learning Approach for the Prediction of Traumatic Brain Injury Induced Coagulopathy