Attention-Enhanced MobileNetV2 for Accurate and Deployable Brain Stroke Detection
Abstract
Stroke is a serious neurological condition that causes high morbidity and mortality rates worldwide. Rapid and accurate diagnosis is critical to improving patient treatment and quality of life. To address this issue, two lightweight deep learning models based on MobileNetV2 were designed for stroke diagnosis and classification. The first model combines channel and spatial attention mechanisms by integrating CBAM (Convolutional Block Attention Module) into MobileNetV2. The second model applies a multi-attention strategy using ECA (Efficient Channel Attention) and Coordinate Attention blocks. Transfer learning was applied using a pre-trained MobileNetV2 network on ImageNet for both models, preventing overfitting on small medical datasets. The results demonstrate that attention mechanisms are effective in capturing subtle differences between classes. Model 1 (99.4% accuracy) and Model 2 (99.8% accuracy) demonstrated high performance, supported by precision, recall, F1-score, and AUC values. The primary goal in selecting MobileNetV2 was to ensure that the models could be integrated into devices with limited resources, such as smartphones and embedded systems. The results were found to be promising in terms of both high accuracy and low computational cost. The proposed approach has the potential to provide scalable and real-time solutions, making a significant contribution to stroke detection.
Keywords
References
- Aksoy, S., Demircioglu, P., & Bogrekci, I. (2025). An Explainable AI Framework for Stroke Classification Based on CT Brain Images. AI, 6(9), 202. https://doi.org/10.3390/ai6090202
- Alomoush, W., Houssein, E. H., Alrosan, A., Abd-Alrazaq, A., Alweshah, M., & Alshinwan, M. (2024). Joint opposite selection enhanced Mountain Gazelle Optimizer for brain stroke classification. Evolutionary Intelligence, 17(4), 2865-2883. https://doi.org/10.1007/s12065-024-00917-8
- Donkor, E. S. (2018). Stroke in the 21st century: a snapshot of the burden, epidemiology, and quality of life. Stroke Research and Treatment, 2018(1), 3238165. https://doi.org/10.1155/2018/3238165
- Elsayed, M. S., Saleh, G. A., Saleh, A. I., & Khalil, A. T. (2025). An effective brain stroke diagnosis strategy based on feature extraction and hybrid classifier. Scientific Reports, 15(1), 29808. https://doi.org/10.1038/s41598-025-14444-8
- Eshmawi, A. A., Khayyat, M., Algarni, A. D., & Hilali-Jaghdam, I. (2022). An ensemble of deep learning enabled brain stroke classification model in magnetic resonance images. Journal of Healthcare Engineering, 2022(1), 7815434. https://doi.org/10.1155/2022/7815434
- Fachruddin, F., Rasywir, E., & Pratama, Y. (2024). Increasing the accuracy of brain stroke classification using random forest algorithm with mutual information feature selection. Jurnal RESTI (Rekayasa Sistem Dan Teknologi Informasi), 8(4), 555-562. https://doi.org/10.29207/resti.v8i4.5795
- Feigin, V. L., Roth, G. A., Naghavi, M., Parmar, P., Krishnamurthi, R., Chugh, S., Mensah, G. A., Norrving, B., Shiue, I., Ng, M., Estep, K., Cercy, K., Murray, C. J. L., & Forouzanfar, M. H. (2016). Global burden of stroke and risk factors in 188 countries, during 1990–2013: a systematic analysis for the Global Burden of Disease Study 2013. The Lancet Neurology, 15(9), 913-924. https://doi.org/10.1016/S1474-4422(16)30073-4
- Hossen, Md. S., Shuvo, E. A., Arif, S. A., Shaha, P., Saiduzzaman, Md., Nasir, M. K. (2025). An efficient deep learning framework for brain stroke diagnosis using computed tomography (CT) images. https://doi.org/10.48550/arXiv.2507.03558
Details
Primary Language
English
Subjects
Deep Learning, Neural Networks
Journal Section
Research Article
Authors
Early Pub Date
July 21, 2026
Publication Date
-
Submission Date
April 8, 2026
Acceptance Date
July 14, 2026
Published in Issue
Year 2026 Number: Advanced Online Publication