Convolutional autoencoder-based deep learning for intracerebral hemorrhage classification using brain CT images.
Authors
Affiliations (6)
Affiliations (6)
- Sensing and Computing Lab, School of Electronics Engineering, Kalinga Institute of Industrial Technology, Bhubaneswar, India.
- School of Mathematics, Physics and Computing, University of Southern Queensland, Springfield Central, QLD Australia.
- National Heart Centre Singapore, Singapore, Singapore.
- Duke-NUS Medical School, Singapore, Singapore.
- Department of Computer Science and Engineering, Heritage Institute of Technology, Kolkata, India.
- Department of Radio-Diagnosis, Kalinga Institute of Medical Science, Bhubaneswar, India.
Abstract
Intracerebral haemorrhage (ICH) is a common form of stroke that affects millions of people worldwide. The incidence is associated with a high rate of mortality and morbidity. Accurate diagnosis using brain non-contrast computed tomography (NCCT) is crucial for decision-making on potentially life-saving surgery. Limited access to expert readers and inter-observer variability imposes barriers to timeous and accurate ICH diagnosis. We proposed a hybrid deep learning model for automated ICH diagnosis using NCCT images, which comprises a convolutional autoencoder (CAE) to extract features with reduced data dimensionality and a dense neural network (DNN) for classification. In order to ensure that the model generalizes to new data, we trained it using tenfold cross-validation and holdout methods. Principal component analysis (PCA) based dimensionality reduction and classification is systematically implemented for comparison. The study dataset comprises 1645 ("ICH" class) and 1648 ("Normal" class belongs to patients with non-hemorrhagic stroke) labelled images obtained from 108 patients, who had undergone CT examination on a 64-slice computed tomography scanner at Kalinga Institute of Medical Sciences between 2020 and 2023. Our developed CAE-DNN hybrid model attained 99.84% accuracy, 99.69% sensitivity, 100% specificity, 100% precision, and 99.84% F1-score, which outperformed the comparator PCA-DNN model as well as the published results in the literature. In addition, using saliency maps, our CAE-DNN model can highlight areas on the images that are closely correlated with regions of ICH, which have been manually contoured by expert readers. The CAE-DNN model demonstrates the proof-of-concept for accurate ICH detection and localization, which can potentially be implemented to prioritize the treatment using NCCT images in clinical settings.