Intracranial Hemorrhage Detection in CT Scans using Deep Learning

Publication Date

8-1-2020

Document Type

Conference Proceeding

Publication Title

2020 IEEE Sixth International Conference on Big Data Computing Service and Applications (BigDataService)

DOI

10.1109/BigDataService49289.2020.00033

First Page

169

Last Page

172

Abstract

In intracranial hemorrhage treatment patient mortality depends on prompt diagnosis based on a radiologist's assessment of CT scans. In this paper, we investigate the intracranial hemorrhage detection problem and built a deep learning model to accelerate the time used to identify the hemorrhages. To assist with this process, a deep learning model can be used to accelerate the time it takes to identify them. In particular, we built a convolutional neural network based on ResNet for the classification. Using 752,803 DICOM files collected from four international universities by the Radiological Society of North America (RSNA) [1], we trained and tested a ResNet- 50 based model for predicting the hemorrhage type. Our model has an accuracy of 93.3% in making the correct multiclass prediction and an average per-class recall score of 76%. We show it is possible to achieve an average recall of 86% while maintaining 70% precision via tuning the prediction thresholds. Lastly, we show real-world applicability by deploying a simple web application. The source code for training, metrics evaluation and web application is available at [2].

Keywords

deep learning, feature recognition, head computed tomography, Intracranial hemorrhage

Department

Computer Engineering

Share

COinS