arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.21311cs.CV

一种可解释的视觉变换器框架用于自动脑肿瘤分类

an interpretable vision transformer framework for automated brain tumor classification

Department of Computer Science, Nazarbayev University(纳扎尔巴耶夫大学计算机科学系);Department of Computer Science, Nnamdi Azikiwe University(恩内迪·阿齐基韦大学计算机科学系)

Chinedu Emmanuel Mbonu, Tochukwu Sunday Belonwu, Okwuchukwu Ejike Chukwuogo, Kenechukwu Sylvanus Anigbogu

AI总结:

本文提出一种基于视觉变换器的框架,用于自动分类四种脑肿瘤类型,通过预训练和增强技术提升分类准确率与可解释性。

摘要:

脑肿瘤是最重要的神经疾病之一,早期准确诊断直接影响患者生存率。本文提出一种深度学习框架,利用预训练的视觉变换器(ViT-B/16)和临床驱动的预处理流程,对7023例MRI扫描进行四类脑肿瘤分类。通过对比受限自适应直方图均衡化增强局部对比度,采用两阶段微调策略和数据增强技术,结合指数移动平均和测试时间增强提升性能,最终实现99.29%的测试准确率和99.25%的宏F1分数。

英文摘要:

Brain tumors represent one of the most critical neurological conditions, where early and accurate diagnosis is directly correlated with patient survival rates. Manual interpretation of Magnetic Resonance Imaging (MRI) scans is time-intensive, subject to inter-observer variability, and demands significant specialist expertise. This paper proposes a deep learning framework for automated four-class brain tumor classification distinguishing glioma, meningioma, pituitary tumor, and healthy brain tissue from a dataset of 7,023 MRI scans. The proposed system employs a Vision Transformer (ViT-B/16) pretrained on ImageNet-21k as the backbone, augmented with a clinically motivated preprocessing and training pipeline. Contrast Limited Adaptive Histogram Equalization (CLAHE) is applied to enhance local contrast and accentuate tumor boundaries invisible to standard normalization. A two-stage fine-tuning strategy is adopted: the classification head is warmed up with the backbone frozen, followed by full fine-tuning with discriminative learning rates. MixUp and CutMix augmentation is applied per batch to improve generalization. Exponential Moving Average (EMA) of weights and Test-Time Augmentation (TTA) further stabilize and boost performance. Attention Rollout visualization provides clinically interpretable heatmaps of the brain regions driving each prediction. The proposed model achieves a test accuracy of 99.29%, macro F1-score of 99.25%, and perfect recall on both healthy and meningioma classes, outperforming all CNN-based baselines

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑