Stroke majorly causes death and disability worldwide, and early recognition is one of the key elements of successful treatment of the same. It is common to diagnose strokes using CT scanning, which is fast and readily available, however, manual analysis may take time and may result in mistakes. In this work, a pre-trained Vision Transformer-based transfer learning framework is proposed for the early identification of brain stroke. A few of the encoder blocks of the ViT model are frozen, and the rest are allowed to be fine-tuned in order to learn brain stroke-specific features. The features that have been extracted are given as input to a single-layer Bi-GRU to perform classification. Class imbalance is handled by data augmentation. The model has achieved 94.06% accuracy in classifying brain stroke from the Stroke Dataset.
@article{arxiv.2511.18316,
title = {Stro-VIGRU: Defining the Vision Recurrent-Based Baseline Model for Brain Stroke Classification},
author = {Subhajeet Das and Pritam Paul and Rohit Bahadur and Sohan Das},
journal= {arXiv preprint arXiv:2511.18316},
year = {2025}
}
Comments
Presented at the International Conference on Computational Intelligence and Data Communication, Accepted for publication in the Taylor and Francis Conference Proceedings