用伪彩色增强MRI图像,提升阿尔茨海默病四类分诊准确率。
Colormap-Enhanced Vision Transformers for MRI-Based Multiclass (4-Class) Alzheimer's Disease Classification
- 将伪彩色变换融入Vision Transformer,强化脑部纹理与对比特征。
- 在OASIS-1数据集上达99.79%准确率,AUC为100%。
- 适合医疗影像分析与早期阿尔茨海默病筛查场景。
磁共振成像(MRI)在阿尔茨海默病(AD)的早期诊断与监测中起关键作用。然而,脑部MRI扫描中的细微结构变化常导致传统深度学习模型难以有效提取判别性特征。本文提出PseudoColorViT-Alz,一种基于伪彩色表示的视觉变换器框架,通过结合伪彩色变换与Vision Transformer的全局特征学习能力,放大标准灰度MRI中被掩盖的解剖纹理与对比度线索。我们在OASIS-1数据集上采用四分类设置(非痴呆、中度痴呆、轻度痴呆、极轻度痴呆)评估该方法,模型取得99.79%的准确率与100%的AUC,超越2024–2025年多种基于CNN与孪生网络的方法(准确率96.1%–99.68%)。结果表明,伪彩色增强与Vision Transformer的结合能显著提升MRI-based阿尔茨海默病分类性能。PseudoColorViT-Alz提供了一个鲁棒且可解释的框架,优于现有方法,有望支持临床决策与早期疾病检测。
原文摘要 · Abstract (English)
Magnetic Resonance Imaging (MRI) plays a pivotal role in the early diagnosis and monitoring of Alzheimer's disease (AD). However, the subtle structural variations in brain MRI scans often pose challenges for conventional deep learning models to extract discriminative features effectively. In this work, we propose PseudoColorViT-Alz, a colormap-enhanced Vision Transformer framework designed to leverage pseudo-color representations of MRI images for improved Alzheimer's disease classification. By combining colormap transformations with the global feature learning capabilities of Vision Transformers, our method amplifies anatomical texture and contrast cues that are otherwise subdued in standard grayscale MRI scans. We evaluate PseudoColorViT-Alz on the OASIS-1 dataset using a four-class classification setup (non-demented, moderate dementia, mild dementia, and very mild dementia). Our model achieves a state-of-the-art accuracy of 99.79% with an AUC of 100%, surpassing the performance of recent 2024--2025 methods, including CNN-based and Siamese-network approaches, which reported accuracies ranging from 96.1% to 99.68%. These results demonstrate that pseudo-color augmentation combined with Vision Transformers can significantly enhance MRI-based Alzheimer's disease classification. PseudoColorViT-Alz offers a robust and interpretable framework that outperforms current methods, providing a promising tool to support clinical decision-making and early detection of Alzheimer's disease.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。