Stroke detection in brain CT images using convolutional neural networks: model development, optimization and interpretability
Journal article
Authors | Abdi, H., Sattar, U., Dattana, V., Hasan, R., Dattana, V. and Mahmood, S. |
---|---|
Abstract | Stroke detection using medical imaging plays a crucial role in early diagnosis and treatment planning. In this study, we propose a Convolutional Neural Network (CNN)-based model for detecting strokes from brain Computed Tomography (CT) images. The model is trained on a dataset consisting of 2501 images, including both normal and stroke cases, and employs a series of preprocessing steps, including resizing, normalization, data augmentation, and splitting into training, validation, and test sets. The CNN architecture comprises three convolutional blocks followed by dense layers optimized through hyperparameter tuning to maximize performance. Our model achieved a validation accuracy of 97.2%, with precision and recall values of 96%, demonstrating high efficacy in stroke classification. Additionally, interpretability techniques such as Local Interpretable Model-agnostic Explanations (LIME), occlusion sensitivity, and saliency maps were used to visualize the model’s decision-making process, enhancing transparency and trust for clinical use. The results suggest that deep learning models, particularly CNNs, can provide valuable support for medical professionals in detecting strokes, offering both high performance and interpretability. The model demonstrates moderate generalizability, achieving 89.73% accuracy on an external, patient-independent dataset of 9900 CT images, underscoring the need for further optimization in diverse clinical settings. |
Keywords | convolutional neural network; stroke detection; brain computed tomography images; medical imaging; model interpretability; hyperparameter optimization; data augmentation; area under the receiver operating characteristic curve |
Year | 2025 |
Journal | Information |
Journal citation | 16 (5), pp. 1-29 |
Publisher | MDPI Open Access Journals |
ISSN | 2078-2489 |
Digital Object Identifier (DOI) | https://doi.org/10.3390/info16050345 |
Web address (URL) | https://www.mdpi.com/2078-2489/16/5/345 |
Accepted author manuscript | File Access Level Restricted |
Publisher's version | License File Access Level Open |
Output status | Published |
Publication dates | 24 Apr 2025 |
Publication process dates | |
Accepted | 23 Apr 2025 |
Deposited | 28 Apr 2025 |
https://repository.derby.ac.uk/item/qxw9y/stroke-detection-in-brain-ct-images-using-convolutional-neural-networks-model-development-optimization-and-interpretability
Download files
25
total views2
total downloads6
views this month0
downloads this month