arXiv:2603.21234cs.CV2026-03被引 2

用彩色特征增强ViT,提升脑肿瘤分类准确率至98.9%

Enhancing Brain Tumor Classification Using Vision Transformers with Colormap-Based Feature Representation on BRISC2025 Dataset

  • ViT结合颜色映射强化MRI图像结构与强度特征
  • 在BRISC2025数据集上达到98.90%准确率和99.97% AUC
  • 适合医学影像分析、临床辅助诊断场景

从磁共振成像(MRI)中准确分类脑肿瘤对早期诊断和治疗规划至关重要。本文提出一种基于视觉变换器(ViT)的深度学习框架,通过彩色映射特征表示增强多类脑肿瘤分类性能。该方法利用变压器架构捕捉长距离依赖关系,并结合颜色映射技术突出MRI扫描中的重要结构与强度变化。实验在包含胶质瘤、脑膜瘤、垂体瘤及非肿瘤四类的BRISC2025数据集上进行,采用准确率、精确率、召回率、F1分数及受试者工作特征曲线下面积(AUC)等标准指标评估。所提方法实现98.90%的分类准确率,优于ResNet50、ResNet101和EfficientNetB2等基线卷积神经网络模型。此外,模型表现出强泛化能力,AUC达99.97%,表明在各类别间具有高区分性能。结果验证了将视觉变换器与彩色特征增强结合的有效性,为临床决策支持提供了良好潜力。

原文摘要 · Abstract (English)

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.

脑肿瘤分类ViT医学影像彩色映射

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。