Methods Inf Med 2016; 55(05): 440-449
DOI: 10.3414/ME15-01-0080
Original Articles
Schattauer GmbH

Identifying Important Attributes for Prognostic Prediction in Traumatic Brain Injury Patients

A Hybrid Method of Decision Tree and Neural Network
Saeedeh Pourahmad
1   Colorectal Research Center, Shiraz University of Medical Sciences, Shiraz, Iran
2   Biostatistics Department, School of Medicine, Shiraz University of Medical Sciences, Shiraz, Iran
,
Iman Hafizi-Rastani
2   Biostatistics Department, School of Medicine, Shiraz University of Medical Sciences, Shiraz, Iran
,
Hosseinali Khalili
3   Shiraz Neuro Science Research Center, Department of Neuro Surgery, Shiraz University of Medical Sciences, Shiraz, Iran
,
Shahram Paydar
4   Trauma Research Center, Department of Surgery, Shiraz University of Medical Sciences, Shiraz, Iran
› Institutsangaben

FundingsThis work was supported by the grant number 91–6166 from Shiraz University of Medical Sciences Research Council.
Weitere Informationen

Publikationsverlauf

Received 01. Juni 2015

Accepted in revised form: 23. Mai 2016

Publikationsdatum:
08. Januar 2018 (online)

Preview

Summary

Background: Generally, traumatic brain injury (TBI) patients do not have a stable condition, particularly after the first week of TBI. Hence, indicating the attributes in prognosis through a prediction model is of utmost importance since it helps caregivers with treatment-decision options, or prepares the relatives for the most-likely outcome. Objectives: This study attempted to determine and order the attributes in prognostic prediction in TBI patients, based on early clinical findings. A hybrid method was employed, which combines a decision tree (DT) and an artificial neural network (ANN) in order to improve the modeling process. Methods: The DT approach was applied as the initial analysis of the network architecture to increase accuracy in prediction. Afterwards, the ANN structure was mapped from the initial DT based on a part of the data. Subsequently, the designed network was trained and validated by the remaining data. 5-fold cross-validation method was applied to train the network. The area under the receiver operating characteristic (ROC) curve, sensitivity, specificity, and accuracy rate were utilized as performance measures. The important attributes were then determined from the trained network using two methods: change of mean squared error (MSE), and sensitivity analysis (SA). Results: The hybrid method offered better results compared to the DT method. The accuracy rate of 86.3 % vs. 82.2 %, sensitivity value of 55.1 % vs. 47.6 %, specificity value of 93.6 % vs. 91.1 %, and the area under the ROC curve of 0.705 vs. 0.695 were achieved for the hybrid method and DT, respectively. However, the attributes’ order by DT method was more consistent with the clinical literature. Conclusions: The combination of different modeling methods can enhance their performance. However, it may create some complexities in computations and interpretations. The outcome of the present study could deliver some useful hints in prognostic prediction on the basis of early clinical findings for TBI patients.