arXiv:2410.21872cs.CVcs.AI2024-10被引 17

用Vision Mamba模型实现脑肿瘤多分类100%准确率,提升医疗影像诊断效率

Advancing Efficient Brain Tumor Multi-Class Classification -- New Insights from the Vision Mamba Model in Transfer Learning

  • 首次将Vision Mamba模型用于脑肿瘤多分类,结合迁移学习
  • 在独立测试集上达到100%分类准确率,显著优于现有模型
  • 轻量高效,适合临床部署,可推广至其他医学影像任务

早期精准诊断脑肿瘤对提高患者生存率至关重要,但其类型多样、形态复杂,诊断挑战大。本研究聚焦预训练模型在脑肿瘤分类中的应用,重点部署Mamba模型。我们微调了多个主流迁移学习模型,并将其应用于脑肿瘤多分类任务。相比从头训练,迁移学习在标注数据有限的医学影像领域展现出显著优势。尤为关键的是,我们首次引入Vision Mamba(Vim)这一新型网络架构,并在脑肿瘤分类中取得突破性成果:在独立测试集上实现100%分类准确率,凸显其在肿瘤分类中的潜力。实验表明,相较于现有最优模型,Vim模型兼具轻量化、高效与高精度,为临床应用提供新视角。此外,基于迁移学习与Vim模型的脑肿瘤分类框架,可广泛适用于其他医学影像分类问题。

原文摘要 · Abstract (English)

Early and accurate diagnosis of brain tumors is crucial for improving patient survival rates. However, the detection and classification of brain tumors are challenging due to their diverse types and complex morphological characteristics. This study investigates the application of pre-trained models for brain tumor classification, with a particular focus on deploying the Mamba model. We fine-tuned several mainstream transfer learning models and applied them to the multi-class classification of brain tumors. By comparing these models to those trained from scratch, we demonstrated the significant advantages of transfer learning, especially in the medical imaging field, where annotated data is often limited. Notably, we introduced the Vision Mamba (Vim), a novel network architecture, and applied it for the first time in brain tumor classification, achieving exceptional classification accuracy. Experimental results indicate that the Vim model achieved 100% classification accuracy on an independent test set, emphasizing its potential for tumor classification tasks. These findings underscore the effectiveness of transfer learning in brain tumor classification and reveal that, compared to existing state-of-the-art models, the Vim model is lightweight, efficient, and highly accurate, offering a new perspective for clinical applications. Furthermore, the framework proposed in this study for brain tumor classification, based on transfer learning and the Vision Mamba model, is broadly applicable to other medical imaging classification problems.

脑肿瘤分类Vision Mamba迁移学习医学影像

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