Automated intracranial hemorrhage detection in traumatic brain injury using 3D CNN.

Abstract

Intracranial hemorrhage (ICH) is a prevalent and potentially fatal consequence of traumatic brain injury (TBI). Timely identification of ICH is crucial to ensure timely intervention and to optimize better patient outcomes. However, the current methods for diagnosing ICH from head computed tomography (CT) scans require skilled personnel (Radiologists and/or Neurosurgeons) who may be unavailable in all centers, especially in rural areas. The aim of this study is to develop a neurotrauma screening tool for identifying ICH from head CT scans of TBI patients. We prospectively collected head CT scans from the Department of Neurosurgery, All India Institute of Medical Sciences, New Delhi. Approximately 738 consecutive head CT scans from patients enrolled in the department were collected for this study spanning a duration of 9 months, that is, January 2020 to September 2020. The metadata collected along with the head CT scans consisted of demographic and clinical details and the radiologist's report which was used as the gold standard. A deep learning-based 3D convolutional neural network (CNN) model was trained on the dataset. The pre-processing, hyperparameters, and augmentation were common for training the 3D CNN model whereas the training modules were set differently. The model was trained along with the save best model option and was monitored by validation metrics. The Institute Ethics Committee permission was taken before starting the study. We developed a 3D CNN model for automatically detecting the ICH from head CT scans. The screening tool was tested in 20 cases and trained on 200 head CT scans, with 99 normal head CT and 101 CT scans with some type of ICH. The final model performed with 90% sensitivity, 70% specificity, and 80% accuracy. Our study reveals that the automated screening tool exhibits a commendable level of accuracy and sensitivity in detecting ICH from the head CT scans. The results indicate that the 3D CNN approach has a potential for further exploring the TBI-related pathologies.