arXiv:2411.15596eess.IVcs.CV2024-11中稿 · the 27th Internati…被引 25

轻量级CNN在脑肿瘤分类上表现媲美大模型,且更省资源。

Comparative Analysis of Resource-Efficient CNN Architectures for Brain Tumor Classification

  • 设计轻量级CNN,结构简单但性能接近预训练大模型。
  • 二分类准确率达98.67%~99.62%,多分类达98.09%。
  • 在少样本学习下仍表现稳健,适合医疗场景部署。

准确的脑肿瘤分类对及时诊断和治疗规划至关重要。尽管深度学习模型如ResNet-18、VGG-16表现出高精度,但往往伴随复杂度和计算开销增加。本研究对比分析了自研轻量级CNN与预训练的ResNet18、VGG16模型在脑肿瘤分类任务中的表现,使用Br35H:: Brain Tumor Detection 2020和Brain Tumor MRI Dataset两个公开数据集。自研CNN虽结构简单,但在二分类任务中于Br35H数据集上达到98.67%准确率,在Brain Tumor MRI Dataset上达99.62%;多分类任务中经微调后于Brain Tumor MRI Dataset上达98.09%。相比而言,ResNet18与VGG16保持高性能,但自研模型更具计算效率。此外,该模型在少样本学习(0, 5, 10, 15, 20, 40, 80样本)下表现出良好鲁棒性,准确率随样本数上升而提升。结果表明,精心设计的轻量级CNN是医学影像任务中高效可靠的替代方案。

原文摘要 · Abstract (English)

Accurate brain tumor classification in MRI images is critical for timely diagnosis and treatment planning. While deep learning models like ResNet-18, VGG-16 have shown high accuracy, they often come with increased complexity and computational demands. This study presents a comparative analysis of effective yet simple Convolutional Neural Network (CNN) architecture and pre-trained ResNet18, and VGG16 model for brain tumor classification using two publicly available datasets: Br35H:: Brain Tumor Detection 2020 and Brain Tumor MRI Dataset. The custom CNN architecture, despite its lower complexity, demonstrates competitive performance with the pre-trained ResNet18 and VGG16 models. In binary classification tasks, the custom CNN achieved an accuracy of 98.67% on the Br35H dataset and 99.62% on the Brain Tumor MRI Dataset. For multi-class classification, the custom CNN, with a slight architectural modification, achieved an accuracy of 98.09%, on the Brain Tumor MRI Dataset. Comparatively, ResNet18 and VGG16 maintained high performance levels, but the custom CNNs provided a more computationally efficient alternative. Additionally,the custom CNNs were evaluated using few-shot learning (0, 5, 10, 15, 20, 40, and 80 shots) to assess their robustness, achieving notable accuracy improvements with increased shots. This study highlights the potential of well-designed, less complex CNN architectures as effective and computationally efficient alternatives to deeper, pre-trained models for medical imaging tasks, including brain tumor classification. This study underscores the potential of custom CNNs in medical imaging tasks and encourages further exploration in this direction.

脑肿瘤分类轻量级CNN少样本学习医疗影像

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