mlsanityData to 7 Oct 2026

Enhancing Brain Tumor Classification Using Vision Transformers with Colormap-Based Feature Representation on BRISC2025 Dataset

arXiv 2603.21234 ↗First posted 22 Mar 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Accurate classification of brain tumors from magnetic resonance imaging (MRI) plays a critical role in early diagnosis and effective treatment planning. In this study, we propose a deep learning framework based on Vision Transformers (ViT) enhanced with colormap-based feature representation to improve multi-class brain tumor classification performance. The proposed approach leverages the ability of transformer architectures to capture long-range dependencies while incorporating color mapping techniques to emphasize important structural and intensity variations within MRI scans. Experiments are conducted on the BRISC2025 dataset, which includes four classes: glioma, meningioma, pituitary tumor, and non-tumor cases. The model is trained and evaluated using standard performance metrics such as accuracy, precision, recall, F1-score, and area under the receiver operating characteristic curve (AUC). The proposed method achieves a classification accuracy of 98.90%, outperforming baseline convolutional neural network models including ResNet50, ResNet101, and EfficientNetB2. In addition, the model demonstrates strong generalization capability with an AUC of 99.97%, indicating high discriminative performance across all classes. These results highlight the effectiveness of combining Vision Transformers with colormap-based feature enhancement for accurate and robust brain tumor classification and suggest strong potential for clinical decision support applications.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 37 references that are in the map. First posted on arXiv 22 Mar 2026.

Builds on

18 of 37 references.

References · 18

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
  4. 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,182
  5. 2017Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?cited by 220
  6. 2025Topological Signatures vs. Gradient Histograms: A Comparative Study for Medical Image Classificationcited by 10
  7. 2025HOG-CNN: Integrating Histogram of Oriented Gradients with Convolutional Neural Networks for Retinal Image Classificationcited by 10
  8. 2025HistoViT: Vision Transformer for Accurate and Scalable Histopathological Cancer Diagnosiscited by 9
  9. 2025Transfer Learning with EfficientNet for Accurate Leukemia Cell Classificationcited by 8
  10. 2023Efficiently Training Vision Transformers on Structural MRI Scans for Alzheimer's Disease Detectioncited by 7
  11. 2025RepViT-CXR: A Channel Replication Strategy for Vision Transformers in Chest X-ray Tuberculosis and Pneumonia Classificationcited by 7
  12. 2025Efficient Breast and Ovarian Cancer Classification via ViT-Based Preprocessing and Transfer Learningcited by 6
6 more
  1. 20253D-TDA -- Topological feature extraction from 3D images for Alzheimer's disease classificationcited by 6
  2. 2025Colormap-Enhanced Vision Transformers for MRI-Based Multiclass (4-Class) Alzheimer's Disease Classificationcited by 6
  3. 2025Addressing High Class Imbalance in Multi-Class Diabetic Retinopathy Severity Grading with Augmentation and Transfer Learningcited by 5
  4. 2026Four-Stage Alzheimer's Disease Classification from MRI Using Topological Feature Extraction, Feature Selection, and Ensemble Learningcited by 5
  5. 2026Hybrid Topological and Deep Feature Fusion for Accurate MRI-Based Alzheimer's Disease Severity Classificationcited by 4
  6. 2026Brain Tumor Classification from 3D MRI Using Persistent Homology and Betti Features: A Topological Data Analysis Approach on BraTS2020cited by 2

Built on by

Not cited yet.