Title:A Simplified Framework for the Detection of Intracranial Hemorrhage in CT Brain Images Using Deep Learning
Volume: 17
Author(s): Praveen Kumaravel, Sasikala Mohan*, Janani Arivudaiyanambi, Nijisha Shajil and Hari Nishanthi Venkatakrishnan
Affiliation:
- Department of Electronics and Communication Engineering, College of Engineering, Guindy, Anna University, Sardar Patel Road, Chennai-600 025,India
Keywords:
Deep CNN, transfer learning, classification, intracranial hemorrhage, ICH, principal component analysis.
Abstract:
Background: The need for accurate and timely detection of Intracranial hemorrhage
(ICH) is of utmost importance to avoid untoward incidents that may even lead to death. Hence, this
presented work leverages the ability of a pretrained deep convolutional neural network (CNN) for
the detection of ICH in computed tomography (CT) brain images.
Methods: Different frameworks have been analyzed for their effectiveness for the classification of
CT brain images into hemorrhage or non-hemorrhage conditions. All these frameworks were investigated
on the CQ500 dataset. Furthermore, an exclusive preprocessing pipeline was designed
for both normal and ICH CT images. Firstly, a framework involving the pretrained deep CNN,
AlexNet, has been exploited for both feature extraction and classification using the transfer learning
method. Secondly, a modified AlexNet-Support vector machine (SVM) classifier is explored,
and finally, a feature selection method, Principal Component Analysis (PCA), has been introduced
in the AlexNet-SVM classifier model, and its efficacy is also explored. These models were trained
and tested on two different sets of CT images, one containing the original images without preprocessing
and another set consisting of preprocessed images.
Results: The modified AlexNet-SVM classifier has shown an improved performance in comparison
to the other investigated frameworks and has achieved a classification accuracy of 99.86% and
sensitivity and specificity of 0.9986 for the detection of ICH in the brain CT images.
Conclusion: This research has given an overview of a simple and efficient framework for the classification
of hemorrhage and non-hemorrhage images. The proposed simplified deep learning
framework also manifests its ability as a screening tool to assist the radiological trainees in the accurate
detection of ICH.
