Brain Stroke Classification Using CT Scans with Transformer-Based Models and Explainable AI
<b>Background & Objective:</b> Stroke remains a leading cause of mortality and long-term disability worldwide, demanding rapid and accurate diagnosis to improve patient outcomes. Computed tomography (CT) scans are widely used in emergency settings due to their speed, availability, and cost-effectiveness. This study proposes an artificial intelligence (AI)-based framework for multiclass stroke classification (ischemic, hemorrhagic, and no stroke) using CT scan images from the Ministry of Health of the Republic of Turkey. <b>Methods:</b> We adopted MaxViT, a state-of-the-art Vision Transformer (ViT)-based architecture, as the primary deep learning model for stroke classification. Additional transformer variants, including Vision Transformer (ViT), Transformer-in-Transformer (TNT), and ConvNeXt, were evaluated for comparison. To improve model generalization and handle class imbalance, classical data augmentation techniques were applied. Furthermore, explainable AI (XAI) was integrated using Grad-CAM++ to provide visual insights into model decisions. <b>Results:</b> The MaxViT model with augmentation achieved the highest performance, reaching an accuracy and F1-score of 98.00%, outperforming the baseline Vision Transformer and other evaluated models. Grad-CAM++ visualizations confirmed that the proposed framework effectively identified stroke-related regions, enhancing transparency and clinical trust. <b>Conclusions:</b> This research contributes to the development of a trustworthy AI-assisted diagnostic tool for stroke, facilitating its integration into clinical practice and improving access to timely and optimal stroke diagnosis in emergency departments.
Authors
- Maha A. Thafar (ORCID: https://orcid.org/0000-0003-0539-7361)
- Shomukh Qari
Institutions
- Taif University (SA)
Publication Details
- Journal
- Diagnostics
- Published
- 2025-09-29
- DOI
- https://doi.org/10.3390/diagnostics15192486
- Citations
- 2
- Primary Topic
- Acute Ischemic Stroke Management
- Type
- article
- Field-Weighted Citation Impact
- 2.71
Funders
- Taif University