Stro-VIGRU: Defining the Vision Recurrent-Based Baseline Model for Brain Stroke Classification
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2025
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866911282896044032 |
|---|---|
| author | Das, Subhajeet Paul, Pritam Bahadur, Rohit Das, Sohan |
| author_facet | Das, Subhajeet Paul, Pritam Bahadur, Rohit Das, Sohan |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2511_18316 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Stro-VIGRU: Defining the Vision Recurrent-Based Baseline Model for Brain Stroke Classification Das, Subhajeet Paul, Pritam Bahadur, Rohit Das, Sohan Computer Vision and Pattern Recognition 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. |
| title | Stro-VIGRU: Defining the Vision Recurrent-Based Baseline Model for Brain Stroke Classification |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2511.18316 |