Enhancing Brain Tumor Classification Using Vision Transformers with Colormap-Based Feature Representation on BRISC2025 Dataset
Organizations: Department of Data Science and Mathematics, Embry-Riddle Aeronautical University, 3700 Willow Creek Rd, Prescott, Arizona 86301, USA
Abstract
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.
Figures & tables
| Hyperparameter | Value |
|---|---|
| Model Backbone | ViT-Base-Patch16-224 |
| Input Image Size | |
| Patch Size | |
| Number of Classes | 4 |
| Batch Size | 32 |
| Number of Epochs | 50 |
| OASIS / OASIS-derived MRI Dataset: 4-Class Classification Results | |||||||
|---|---|---|---|---|---|---|---|
| Method | # Classes | Dataset | Prec | Rec | Accuracy | F1-score | AUC |
| ResNet50 [ 27 ] | 4 | BRISC2025 | 98.36 | 98.36 | 98.20 | 98.34 | - |
| ResNet101 [ 27 ] | 4 | BRISC2025 | 98.20 | 98.26 | 98.09 | 98.21 | - |
| EfficientNetB2 [ 27 ] | 4 | BRISC2025 | 9.851 | 98.54 | 98.37 | 98.52 | - |
| Ours | 4 | BRISC2025 | 98.99 | 98.98 | 98.90 | 98.98 | 99.97 |