A Simplified Framework for the Detection of Intracranial Hemorrhage in CT Brain Images Using Deep Learning.

Abstract

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. 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. 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. 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.