Abstract:BACKGROUND:The treatment and care of adults and children with traumatic brain injury (TBI) constitute an intractable global health problem. Predicting the prognosis and length of hospital stay of patients with TBI may improve therapeutic effects and significantly reduce societal health care burden. Applying novel machine learning methods to the field of TBI may be valuable for determining the prognosis and cost-effectiveness of clinical treatment.OBJECTIVE:We aimed to combine multiple machine learning approaches to build hybrid models for predicting the prognosis and length of hospital stay for adults and children with TBI.METHODS:We collected relevant clinical information from patients treated at the Neurosurgery Center of the Second Affiliated Hospital of Anhui Medical University between May 2017 and May 2022, of which 80% was used for training the model and 20% for testing via screening and data splitting. We trained and tested the machine learning models using 5 cross-validations to avoid overfitting. In the machine learning models, 11 types of independent variables were used as input variables and Glasgow Outcome Scale score, used to evaluate patients' prognosis, and patient length of stay were used as output variables. Once the models were trained, we obtained and compared the errors of each machine learning model from 5 rounds of cross-validation to select the best predictive model. The model was then externally tested using clinical data of patients treated at the First Affiliated Hospital of Anhui Medical University from June 2021 to February 2022.RESULTS:The final convolutional neural network-support vector machine (CNN-SVM) model predicted Glasgow Outcome Scale score with an accuracy of 93% and 93.69% in the test and external validation sets, respectively, and an area under the curve of 94.68% and 94.32% in the test and external validation sets, respectively. The mean absolute percentage error of the final built convolutional neural network-support vector regression (CNN-SVR) model predicting inpatient time in the test set and external validation set was 10.72% and 10.44%, respectively. The coefficient of determination (R2) was 0.93 and 0.92 in the test set and external validation set, respectively. Compared with back-propagation neural network, CNN, and SVM models built separately, our hybrid model was identified to be optimal and had high confidence.CONCLUSIONS:This study demonstrates the clinical utility of 2 hybrid models built by combining multiple machine learning approaches to accurately predict the prognosis and length of stay in hospital for adults and children with TBI. Application of these models may reduce the burden on physicians when assessing TBI and assist clinicians in the medical decision-making process.

Predicting the severity and discharge prognosis of traumatic brain injury based on intracranial pressure data using machine learning algorithms

Predicting the Outcomes of Shunt Implantation in Patients with Post-Traumatic Hydrocephalus and Severe Conscious Disturbance: a Scoring System Based on Clinical Characteristics.

The Value of Machine Learning Models in Predicting Factors Associated with the Need for Permanent Shunting in Patients with Intracerebral Hemorrhage Requiring Emergency Cerebrospinal Fluid Diversion

Machine Learning Based Prediction of Imminent ICP Insults During Neurocritical Care of Traumatic Brain Injury

Random forest-based prediction of intracranial hypertension in patients with traumatic brain injury

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

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

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

Machine learning models for predicting early hemorrhage progression in traumatic brain injury

Machine Learning-Based Continuous Intracranial Pressure Prediction for Traumatic Injury Patients

A Computer-Assisted System for Early Mortality Risk Prediction in Patients with Traumatic Brain Injury Using Artificial Intelligence Algorithms in Emergency Room Triage

Predictors of 30-Day Mortality and Morbidity Following Craniotomy for Traumatic Brain Injury: An ACS NSQIP Database Analysis

Development of predictive model for the neurological deterioration among mild traumatic brain injury patients using machine learning algorithms

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 Model for Predicting Acute Respiratory Failure in Individuals With Moderate-to-Severe Traumatic Brain Injury

Development and validation of a prehospital-stage prediction tool for traumatic brain injury: a multicentre retrospective cohort study in Korea

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

Automated Hematoma Detection and Outcome Prediction in Patients With Traumatic Brain Injury

The establishment and validation of a prediction model for traumatic intracranial injury patients: a reliable nomogram

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

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