对比6种迁移学习模型,发现InceptionV3在胸部X光病灶分类中表现最佳
Performance Evaluation of Transfer Learning Based Medical Image Classification Techniques for Disease Detection
- 用预训练模型迁移学习,避免从零训练大模型
- InceptionV3准确率最高,深度越深的ResNet性能越好
- 轻量级网络结合特征提取器即可高效预测,适合数据少场景
医学图像分类在疾病识别中日益重要,通过分析X光、MRI和CT等图像特征进行分类。近年来,深度学习在该领域备受关注,但直接从头训练大型模型通常不可行。为此,迁移学习(TL)技术被广泛应用,即复用预训练模型完成新任务。本文对六种预训练模型(AlexNet、VGG16、ResNet18、ResNet34、ResNet50、InceptionV3)在自建胸片数据集上的表现进行了全面评估。实验结果表明,InceptionV3在所有标准指标上均优于其他模型;深度更大的ResNet系列性能逐级提升,而VGG16和AlexNet表现尚可但准确率较低。此外,我们还进行了不确定性分析与运行时间比较,验证了模型鲁棒性与计算效率。研究发现,迁移学习在数据有限时普遍有效,但提升幅度受模型结构、数据规模及源-目标任务领域相似性影响。更重要的是,仅需一个训练良好的特征提取器,搭配轻量级前馈网络即可实现高效预测。本研究为医疗图像分类中的迁移学习应用提供了实证依据,并为模型选型提供指导。
原文摘要 · Abstract (English)
Medical image classification plays an increasingly vital role in identifying various diseases by classifying medical images, such as X-rays, MRIs and CT scans, into different categories based on their features. In recent years, deep learning techniques have attracted significant attention in medical image classification. However, it is usually infeasible to train an entire large deep learning model from scratch. To address this issue, one of the solutions is the transfer learning (TL) technique, where a pre-trained model is reused for a new task. In this paper, we present a comprehensive analysis of TL techniques for medical image classification using deep convolutional neural networks. We evaluate six pre-trained models (AlexNet, VGG16, ResNet18, ResNet34, ResNet50, and InceptionV3) on a custom chest X-ray dataset for disease detection. The experimental results demonstrate that InceptionV3 consistently outperforms other models across all the standard metrics. The ResNet family shows progressively better performance with increasing depth, whereas VGG16 and AlexNet perform reasonably well but with lower accuracy. In addition, we also conduct uncertainty analysis and runtime comparison to assess the robustness and computational efficiency of these models. Our findings reveal that TL is beneficial in most cases, especially with limited data, but the extent of improvement depends on several factors such as model architecture, dataset size, and domain similarity between source and target tasks. Moreover, we demonstrate that with a well-trained feature extractor, only a lightweight feedforward model is enough to provide efficient prediction. As such, this study contributes to the understanding of TL in medical image classification, and provides insights for selecting appropriate models based on specific requirements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。