English

TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification

Artificial Intelligence 2026-05-11 v1

Abstract

Classifying brain tumors using magnetic resonance imaging (MRI) is crucial for early diagnosis and treatment; however, tumor heterogeneity and a dearth of annotated datasets restrict the use of supervised deep learning approaches. In this work, we use self-supervised learning (SSL) to study multi-class brain tumor classification. Using a ResNet-50 backbone, we evaluate four SSL frameworks including SimCLR, BYOL, DINO, and Moco v3 on a publicly available dataset of 4,448 MRIs with 17 distinct tumor types. On the dataset, SimCLR achieved 99.64% accuracy, 99.64% precision, 99.64% recall, and 99.64% F1-score. The workflow includes preprocessing, fine-tuning, linear evaluation, and SSL pretraining with data augmentations. Results show that, when labels are limited, SSL-pretrained models outperform supervised baselines in terms of F1-score, recall, accuracy, and precision. Additionally, by providing visual insights into model decisions, Explainable AI techniques (Grad-CAM, Grad-CAM++, EigenCAM) enhance interpretability. These results demonstrate SSL's scalability and dependability in diagnosing brain tumors from unlabeled medical data.

Keywords

Cite

@article{arxiv.2605.01999,
  title  = {TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification},
  author = {Abrar Hossain Zahin and Amit Kumar Saha and Tanvir Mridha and Saifur Rahman and Jannatul Ferdous Prome and Raima Husna and Israt Jahan and Ahmed Wasif Reza},
  journal= {arXiv preprint arXiv:2605.01999},
  year   = {2026}
}

Comments

16 pages, 9 figures, 6 Tables

R2 v1 2026-07-01T12:47:38.790Z