A machine learning model for predicting favorable outcome in severe traumatic brain injury patients after 6 months
2
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
Background: Traumatic brain injury (TBI), which occurs commonly worldwide, is among the more costly of health and socioeconomic problems. Accurate prediction of favorable outcomes in severe TBI patients could assist with optimizing treatment procedures, predicting clinical outcomes, and result in substantial economic savings.Methods: In this study, we examined the capability of a machine learning-based model in predicting “favorable” or “unfavorable” outcomes after 6 months in severe TBI patients using only parameters measured on admission. Three models were developed using logistic regression, random forest, and support vector machines trained on parameters recorded from 2,381 severe TBI patients admitted to the neuro-intensive care unit of Rajaee (Emtiaz) Hospital (Shiraz, Iran) between 2015 and 2017. Model performance was evaluated using three indices: sensitivity, specificity, and accuracy. A ten-fold cross-validation method was used to estimate these indices.Results: Overall, the developed models showed excellent performance with AUC around 0.81, sensitivity and specificity of around 0.78. The top-three factors important in predicting 6-month post-trauma survival status in TBI patients are “Glasgow coma scale motor response,” “pupillary reactivity,” and “age.”Conclusions: Machine learning techniques might be used to predict the 6-month outcome in TBI patients using only the parameters measured on admission when the machine learning is trained using a large data set.
Abstract
Background: Traumatic brain injury (TBI), which occurs commonly worldwide, is among the more costly of health and socioeconomic problems. Accurate prediction of favorable outcomes in severe TBI patients could assist with optimizing treatment procedures, predicting clinical outcomes, and result in substantial economic savings.Methods: In this study, we examined the capability of a machine learning-based model in predicting “favorable” or “unfavorable” outcomes after 6 months in severe TBI patients using only parameters measured on admission. Three models were developed using logistic regression, random forest, and support vector machines trained on parameters recorded from 2,381 severe TBI patients admitted to the neuro-intensive care unit of Rajaee (Emtiaz) Hospital (Shiraz, Iran) between 2015 and 2017. Model performance was evaluated using three indices: sensitivity, specificity, and accuracy. A ten-fold cross-validation method was used to estimate these indices.Results: Overall, the developed models showed excellent performance with AUC around 0.81, sensitivity and specificity of around 0.78. The top-three factors important in predicting 6-month post-trauma survival status in TBI patients are “Glasgow coma scale motor response,” “pupillary reactivity,” and “age.”Conclusions: Machine learning techniques might be used to predict the 6-month outcome in TBI patients using only the parameters measured on admission when the machine learning is trained using a large data set.
참고문헌 (25)
- Nguyen R, "The international incidence of traumatic brain injury : a systematic review and meta-analysis", Can J Neurol Sci, 43, pp.774~785 (2016)
- Prins M, "The pathophysiology of traumatic brain injury at a glance", Dis Model Mech, 6, pp.1307~1315 (2013)
- Werner C, "Pathophysiology of traumatic brain injury", Br J Anaesth, 99, pp.4~9 (2007)
- Weir J, "Does the extended Glasgow Outcome Scale add value to the conventional Glasgow Outcome Scale?", J Neurotrauma, 29, pp.53~58 (2012)
- Hale AT, "Using an artificial neural network to predict traumatic brain injury", J Neurosurg Pediatr, 23, pp.219~226 (2018)
- Hale AT, "Machine-learning analysis outperforms conventional statistical models and CT classification systems in predicting 6-month outcomes in pediatric patients sustaining traumatic brain injury", Neurosurg Focus, 45, pp.E2~ (2018)
- Ledig C, "Robust whole-brain segmentation : application to traumatic brain injury", Med Image Anal, 21, pp.40~58 (2015)
- Gong T, "Automated prognosis analysis for traumatic brain injury CT images", Proceedings of 2015 3rd IAPR Asian Conference on Pattern Recognition(ACPR), pp.386~390 (2015)
- Agoston DV, "Big Data in traumatic brain injury : promise and challenges", Concussion, 2, pp.CNC45~ (2017)
- Folweiler KA, "Unsupervised machine learning reveals novel traumatic brain injury patient phenotypes with distinct acute injury profiles and long-term outcomes", J Neurotrauma, 37, pp.1431~1444 (2020)
- Matsuo K, "Machine learning to predict in-hospital morbidity and mortality after traumatic brain injury", J Neurotrauma, 37, pp.202~210 (2020)
- Shi HY, "In-hospital mortality after traumatic brain injury surgery : a nationwide population-based comparison of mortality predictors used in artificial neural network and logistic regression models", J Neurosurg, 118, pp.746~752 (2013)
- Klemenc-Ketis Z, "Outcome predictors of Glasgow Outcome Scale score in patients with severe traumatic brain injury", Ulus Travma Acil Cerrahi Derg, 17, pp.509~515 (2011)
- Rughani AI, "Use of an artificial neural network to predict head injury outcome", J Neurosurg, 113, pp.585~590 (2010)
- Eftekhar B, "Comparison of artificial neural network and logistic regression models for prediction of mortality in head trauma based on initial clinical data", BMC Med Inform Decis Mak, 5, pp.3~ (2005)
- Pourahmad S, "Comparison of four variable selection methods to determine the important variables in predicting the prognosis of traumatic brain injury patients by support vector machine", J Res Med Sci, 24, pp.97~ (2019)
- Demetriades D, "Outcome and prognostic factors in head injuries with an admission Glasgow Coma Scale score of 3", Arch Surg, 139, pp.1066~1068 (2004)
- Guyon I, "An introduction to variable and feature selection", J Mach Learn Res, 3, pp.1157~1182 (2003)
- Hastie T, "The elements of statistical learning: data mining, inference, and prediction" (2009), Springer-Verlag
- Probst P, "Hyperparameters and tuning strategies for random forest", WIREs Data Min Knowl Discov, 9, pp.e1301~ (2019)
- Liaw A, "Classification and regression by randomforest", R News, 2, pp.18~22 (2002)
- Cortes C, "Support-vector networks", Mach Learn, 20, pp.273~297 (1995)
- Bennis FC, "Improving prediction of favourable outcome after 6 months in patients with severe traumatic brain injury using physiological cerebral parameters in a multivariable logistic regression model", Neurocrit Care, 33, pp.542~551 (2020)
- Rubin ML, "Prognosis of six-month Glasgow Outcome Scale in severe traumatic brain injury using hospital admission characteristics, injury severity characteristics, and physiological monitoring during the first day post-injury", J Neurotrauma, 36, pp.2417~2422 (2019)
- Haveman ME, "Predicting outcome in patients with moderate to severe traumatic brain injury using electroencephalography", Crit Care, 23, pp.401~ (2019)