通用预训练模型在脑瘤分类上表现优于医学专用模型。
General vs Domain-Specific CNNs: Understanding Pretraining Effects on Brain MRI Tumor Classification
- 用相同数据对比三种模型,发现通用模型更优
- ConvNeXt-Tiny达93%准确率,医学模型仅68%
- 适合医疗图像少样本任务的模型选择参考
从磁共振成像(MRI)中准确识别脑肿瘤对及时诊断和有效治疗至关重要。尽管深度卷积神经网络(CNN),尤其是基于大规模数据集预训练的模型,在医学图像分析中展现出巨大潜力,但在数据有限的情况下,一个关键问题浮现:在专业医学图像库上预训练的模型是否优于在多样化通用数据集上预训练的模型?本研究对比了三种不同预训练CNN架构在脑瘤分类中的表现:基于医学领域数据预训练的RadImageNet DenseNet121,以及两种现代通用模型EfficientNetV2S和ConvNeXt-Tiny。所有模型均在小型脑MRI数据集上以统一条件训练与微调。实验结果表明,ConvNeXt-Tiny表现最佳,测试准确率达93%,其次为EfficientNetV2S的85%;而RadImageNet DenseNet121仅取得68%准确率且损失更高,显示其泛化能力有限,尽管进行了领域特定预训练。这些发现表明,医学领域预训练并不必然带来更好性能;相反,基于大规模多样化数据预训练、具有更深架构的现代通用CNN,可能在数据稀缺的医疗影像诊断任务中提供更有效的迁移学习能力。
原文摘要 · Abstract (English)
The accurate identification of brain tumors from magnetic resonance imaging (MRI) is essential for timely diagnosis and effective therapeutic intervention. While deep convolutional neural networks (CNNs), particularly those pre-trained on extensive datasets, have shown considerable promise in medical image analysis, a key question arises when working with limited data: do models pre-trained on specialized medical image repositories outperform those pre-trained on diverse, general-domain datasets? This research presents a comparative analysis of three distinct pre-trained CNN architectures for brain tumor classification: RadImageNet DenseNet121, which leverages pre-training on medical-domain data, alongside two modern general-purpose networks, EfficientNetV2S and ConvNeXt-Tiny. All models were trained and fine-tuned under uniform experimental conditions using a modestly sized brain MRI dataset to maintain consistency in evaluation. The experimental outcomes indicate that ConvNeXt-Tiny delivered the best performance, achieving 93% test accuracy, followed by EfficientNetV2S at 85%. In contrast, RadImageNet DenseNet121 attained only 68% accuracy and exhibited higher loss, indicating limited generalization capability despite its domain-specific pre-training. These observations imply that pre-training on medical-domain data does not necessarily guarantee superior performance in data-scarce scenarios. Conversely, contemporary general-purpose CNNs with deeper architectures, pre-trained on large-scale diverse datasets, may offer more effective transfer learning for specialized diagnostic tasks in medical imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。