Machine learning-based models to predict the need for neurosurgical intervention after moderate traumatic brain injury.

Background and Aims: Traumatic brain injury (TBI) is a widespread global health issue with significant economic consequences. However, no existing model exists to predict the need for neurosurgical intervention in moderate TBI patients with positive initial computed tomography scans. This study determines the efficacy of machine learning (ML)-based models in predicting the need for neurosurgical intervention. Methods: This is a retrospective study of patients admitted to the neuro-intensive care unit of Emtiaz Hospital, Shiraz, Iran, between January 2018 and December 2020. The most clinically important variables from patients that met our inclusion and exclusion criteria were collected and used as predictors. We developed models using multilayer perceptron, random forest, support vector machines (SVM), and logistic regression. To evaluate the models, their F1-score, sensitivity, specificity, and accuracy were assessed using a fourfold cross-validation method. Results: Based on predictive models, SVM showed the highest performance in predicting the need for neurosurgical intervention, with an F1-score of 0.83, an area under curve of 0.93, sensitivity of 0.82, specificity of 0.84, a positive predictive value of 0.83, and a negative predictive value of 0.83. Conclusion: The use of ML-based models as decision-making tools can be effective in predicting with high accuracy whether neurosurgery will be necessary after moderate TBIs. These models may ultimately be used as decision-support tools to evaluate early intervention in TBI patients.