Development of a System for Predicting Hospitalization Time for Patients With Traumatic Brain Injury Based on Machine Learning Algorithms: User-Centered Design Case Study.

Currently, the treatment and care of patients with traumatic brain injury (TBI) are intractable health problems worldwide and greatly increase the medical burden in society. However, machine learning-based algorithms and the use of a large amount of data accumulated in the clinic in the past can predict the hospitalization time of patients with brain injury in advance, so as to design a reasonable arrangement of resources and effectively reduce the medical burden of society. Especially in China, where medical resources are so tight, this method has important application value. We aimed to develop a system based on a machine learning model for predicting the length of hospitalization of patients with TBI, which is available to patients, nurses, and physicians. We collected information on 1128 patients who received treatment at the Neurosurgery Center of the Second Affiliated Hospital of Anhui Medical University from May 2017 to May 2022, and we trained and tested the machine learning model using 5 cross-validations to avoid overfitting; 28 types of independent variables were used as input variables in the machine learning model, and the length of hospitalization was used as the output variables. Once the models were trained, we obtained the error and goodness of fit (R2) of each machine learning model from the 5 rounds of cross-validation and compared them to select the best predictive model to be encapsulated in the developed system. In addition, we externally tested the models using clinical data related to patients treated at the First Affiliated Hospital of Anhui Medical University from June 2021 to February 2022. Six machine learning models were built, including support vector regression machine, convolutional neural network, back propagation neural network, random forest, logistic regression, and multilayer perceptron. Among them, the support vector regression has the smallest error of 10.22% on the test set, the highest goodness of fit of 90.4%, and all performances are the best among the 6 models. In addition, we used external datasets to verify the experimental results of these 6 models in order to avoid experimental chance, and the support vector regression machine eventually performed the best in the external datasets. Therefore, we chose to encapsulate the support vector regression machine into our system for predicting the length of stay of patients with traumatic brain trauma. Finally, we made the developed system available to patients, nurses, and physicians, and the satisfaction questionnaire showed that patients, nurses, and physicians agreed that the system was effective in providing clinical decisions to help patients, nurses, and physicians. This study shows that the support vector regression machine model developed using machine learning methods can accurately predict the length of hospitalization of patients with TBI, and the developed prediction system has strong clinical use.

Zhou H ，Fang C ，Pan Y 《-》

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

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. 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. 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. 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. 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.

Fang C ，Pan Y ，Zhao L ，Niu Z ，Guo Q ，Zhao B ... -  《-》

Construction of a predictive model based on MIV-SVR for prognosis and length of stay in patients with traumatic brain injury: Retrospective cohort study.

Pan Y ，Fang C ，Zhu X ，Wan J ... -  《Digital Health》

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

Mild traumatic brain injury (mTBI) comprises a majority of traumatic brain injury (TBI) cases. While some mTBI would suffer neurological deterioration (ND) and therefore have poorer prognosis. This study was designed to develop the predictive model for the ND among mTBI using machine learning algorithms. mTBI patients recorded in the Medical Information Mart for Intensive Care-III were selected for the study. The ND was defined as a drop of Glasgow Coma Scale ≥ 2 within the first 7 day after admission. Eight machine learning algorithms were trained and validated with 5-fold cross validation including extreme gradient boosting, logistic regression, light gradient boosting machine, random forest, adaptive boosting, decision tree, complement naïve Bayes, and support vector machine. The value of eight machine learning algorithms was compared by the area under the receiver operating characteristic curve (AUC). 361 mTBI patients suffered the ND with the incidence of 30.7%. The ND group had higher 30-day mortality (p = 0.001). In the training cohort of mTBI patients, the random forest performed the best on predicting the ND with the AUC of 1.000. The XGBoost and AdaBoost had an AUC of 0.827 and 0.815, respectively. The logistic regression performed the best on predicting the ND in the validation cohort with the AUC of 0.741. The XGBoost, random forest and AdaBoost had an AUC of 0.729, 0.735, 0.736 in the validation cohort, respectively. After adjusting confounding effects, the multivariate logistic regression found only two independent risk factors for the ND including Sequential Organ Failure Assessment (SOFA) (p < 0.001) and hypertension (p = 0.001). The logistic regression predictive model composed of SOFA and hypertension had an AUC of 0.741. SOFA score and complicated hypertension are two independent risk factors for the neurological deterioration among mTBI patients. The logistic regression predictive model incorporating SOFA and hypertension is helpful to identify mTBI patients with the high risk of ND.

Shan B ，Wang R ，Xu J 《-》

Predicting the Severity and Discharge Prognosis of Traumatic Brain Injury Based on Intracranial Pressure Data Using Machine Learning Algorithms.

This study aimed to explore the potential of employing machine learning algorithms based on intracranial pressure (ICP), ICP-derived parameters, and their complexity to predict the severity and short-term prognosis of traumatic brain injury (TBI). A single-center prospectively collected cohort of neurosurgical intensive care unit admissions was analyzed. We extracted ICP-related data within the first 6 hours and processed them using complex algorithms. To indicate TBI severity and short-term prognosis, Glasgow Coma Scale score on the first postoperative day and Glasgow Outcome Scale-Extended score at discharge were used as binary outcome variables. A univariate logistic regression model was developed to predict TBI severity using only mean ICP values. Subsequently, 3 multivariate Random Forest (RF) models were constructed using different combinations of mean and complexity metrics of ICP-related data. To avoid overfitting, five-fold cross-validations were performed. Finally, the best-performing multivariate RF model was used to predict patients' discharge Glasgow Outcome Scale-Extended score. The logistic regression model exhibited limited predictive ability with an area under the curve (AUC) of 0.558. Among multivariate models, the RF model, combining the mean and complexity metrics of ICP-related data, achieved the most robust ability with an AUC of 0.815. Finally, in terms of predicting discharge Glasgow Outcome Scale-Extended score, this model had a consistent performance with an AUC of 0.822. Cross-validation analysis confirmed the performance. This study demonstrates the clinical utility of the RF model, which integrates the mean and complexity metrics of ICP data, in accurately predicting the TBI severity and short-term prognosis.

Zhu J ，Shan Y ，Li Y ，Wu X ，Gao G ... -  《-》