TumorXAI:面向可解释性脑部MRI肿瘤分类的自监督深度学习框架 / TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification
1️⃣ 一句话总结
本文提出了一种名为TumorXAI的自监督学习框架,能在标注数据稀缺的情况下,高效地对脑部MRI图像中的17种肿瘤进行分类,并利用可解释性AI技术(如Grad-CAM)让医生直观理解模型的判断依据。
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.
TumorXAI:面向可解释性脑部MRI肿瘤分类的自监督深度学习框架 / TumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification
本文提出了一种名为TumorXAI的自监督学习框架,能在标注数据稀缺的情况下,高效地对脑部MRI图像中的17种肿瘤进行分类,并利用可解释性AI技术(如Grad-CAM)让医生直观理解模型的判断依据。
源自 arXiv: 2605.01999