Comparison of Artificial Neural Network and Logistic Regression Models for Prediction of Psychological Symptom Six Months after Mild Traumatic Brain Injury

Abstract

Background: Nowadays, outcome prediction models using logistic regression (LR) and artificial neural network (ANN) analysis have been developed in many areas of healthcare research. Objectives: In this study, we have compared the performance of multivariable LR and ANN models, in prediction of psychological symptoms six months after mild traumatic brain injury. Methods: In a prospective cohort study, information of 100 mild traumatic brain injury patients, during a six months period between 2014 and 2016 were included. Data were divided into two training (n = 50) and testing (n = 50) groups, randomly. 300 ANNs and LRs were studied in the first group and then the predicted values were compared in the second group using the two final models. The receiver operating characteristic (ROC) curve and accuracy rate were used to compare these models. Results: The results showed that accuracy rate for the neural network model was 90.65%, while it was 75.96% for the LR model. Conclusions: The ANN models appeared to be more powerful in predicting psychological symptoms versus the LR models.