集成模型在乳腺癌检测中达到99.94%准确率,优于单一CNN和迁移学习。
A study on Deep Convolutional Neural Networks, Transfer Learning and Ensemble Model for Breast Cancer Detection
- 采用六种CNN架构与集成模型对比,提升诊断性能。
- 集成模型准确率达99.94%,为最高表现。
- 迁移学习未提升原有模型精度,提示需谨慎应用。
深度学习中,迁移学习与集成模型在辅助疾病诊断方面展现出潜力,但实际应用仍有限。现有集成模型设计缺乏系统性,常忽略冗余层问题,且受限于数据不平衡与增强不足。尽管已有多种深度卷积神经网络(D-CNN)用于乳腺癌检测与分类,但对其性能的比较研究较少。本研究比较了六种基于CNN的深度学习架构(SE-ResNet152、MobileNetV2、VGG19、ResNet18、InceptionV3、DenseNet-121)、迁移学习及集成模型在乳腺癌检测中的表现。结果显示,集成模型在乳腺癌检测与分类中取得最高准确率99.94%。然而,迁移学习未能提升原有各模型(包括SE-ResNet152等)的精度。高准确率表明CNN在乳腺癌检测中具有显著潜力。该研究对生物医学工程、计算机辅助诊断及基于机器学习的疾病检测具有重要意义。
原文摘要 · Abstract (English)
In deep learning, transfer learning and ensemble models have shown promise in improving computer-aided disease diagnosis. However, applying the transfer learning and ensemble model is still relatively limited. Moreover, the ensemble model's development is ad-hoc, overlooks redundant layers, and suffers from imbalanced datasets and inadequate augmentation. Lastly, significant Deep Convolutional Neural Networks (D-CNNs) have been introduced to detect and classify breast cancer. Still, very few comparative studies were conducted to investigate the accuracy and efficiency of existing CNN architectures. Realising the gaps, this study compares the performance of D-CNN, which includes the original CNN, transfer learning, and an ensemble model, in detecting breast cancer. The comparison study of this paper consists of comparison using six CNN-based deep learning architectures (SE-ResNet152, MobileNetV2, VGG19, ResNet18, InceptionV3, and DenseNet-121), a transfer learning, and an ensemble model on breast cancer detection. Among the comparison of these models, the ensemble model provides the highest detection and classification accuracy of 99.94% for breast cancer detection and classification. However, this study also provides a negative result in the case of transfer learning, as the transfer learning did not increase the accuracy of the original SE-ResNet152, MobileNetV2, VGG19, ResNet18, InceptionV3, and DenseNet-121 model. The high accuracy in detecting and categorising breast cancer detection using CNN suggests that the CNN model is promising in breast cancer disease detection. This research is significant in biomedical engineering, computer-aided disease diagnosis, and ML-based disease detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。