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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zahin, Abrar Hossain, Saha, Amit Kumar, Mridha, Tanvir, Rahman, Saifur, Prome, Jannatul Ferdous, Husna, Raima, Jahan, Israt, Reza, Ahmed Wasif
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914541986643968
author Zahin, Abrar Hossain
Saha, Amit Kumar
Mridha, Tanvir
Rahman, Saifur
Prome, Jannatul Ferdous
Husna, Raima
Jahan, Israt
Reza, Ahmed Wasif
author_facet Zahin, Abrar Hossain
Saha, Amit Kumar
Mridha, Tanvir
Rahman, Saifur
Prome, Jannatul Ferdous
Husna, Raima
Jahan, Israt
Reza, Ahmed Wasif
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2605_01999
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification
Zahin, Abrar Hossain
Saha, Amit Kumar
Mridha, Tanvir
Rahman, Saifur
Prome, Jannatul Ferdous
Husna, Raima
Jahan, Israt
Reza, Ahmed Wasif
Artificial Intelligence
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.
title TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification
topic Artificial Intelligence
url https://arxiv.org/abs/2605.01999