Title of article :
Comparison of Artificial Neural Network and Logistic Regression Models for Prediction of Psychological Symptom Six Months after Mild Traumatic Brain Injury
Author/Authors :
Shafiei, Elham Kashan University of Medical Sciences, Kashan , Fakharian, Esmaeil Kashan University of Medical Sciences, Kashan , Omidi, Abdollah Department of Clinical Psychology - Kashan University of Medical Sciences, Kashan , Akbari, Hossein Department of Epidemiology and Biostatistics - School of Public Health - Kashan University of Medical Sciences, Kashan , Delpisheh, Ali Ilam University of Medical Sciences, Ilam , Nademi, Arash Department of Statistics - Ilam Branch - Islamic Azad University, Ilam
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.
Keywords :
Artificial Neural Network , Logistic Regression , Mental Disorder , Mild Traumatic Brain Injury , Prediction , Principle Component Analysis , Psychological Symptom
Journal title :
Astroparticle Physics