arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.15313cs.CV
收藏

用于乳腺癌分类的深度迁移学习

Deep Transfer Learning for Breast Cancer Classification

Michigan State University(密歇根州立大学);AIMS Ghana(加纳非洲数学科学研究所)

Prudence Djagba, J. K. Buwa Mbouobda

AI总结:

本研究比较了VGG、ViT和Resnet在浸润性导管癌图像分类中的表现,发现Resnet-34准确率达90.40%,而VGG-16因参数少具有更高F1分数,表明深度迁移学习能有效提升乳腺癌诊断的准确性和可及性。

摘要:

乳腺癌是一个影响全球数百万女性的重大全球性健康问题。尽早且准确地分类乳腺癌对于有效治疗和改善患者预后至关重要。深度迁移学习已成为一种很有前景的技术,通过利用预训练模型并在相关任务之间迁移知识来改善乳腺癌分类。在本研究中,我们考察了使用 VGG、Vision Transformers (ViT) 和 Resnet 对浸润性导管癌 (IDC) 图像进行分类,并对这些算法进行了比较分析。结果显示,Resnet-34 在分类癌症图像方面具有 $90.40\%$ 的准确率的巨大优势。然而,预训练的 VGG-16 表现出更高的 F1 分数,因为需要更新的参数较少。我们相信,乳腺癌诊断领域将极大地受益于深度迁移学习的使用。迁移学习通过允许用少量数据训练深度学习模型,可能有助于提高乳腺癌筛查的准确性和可及性。

英文摘要:

Breast cancer is a major global health issue that affects millions of women worldwide. Classification of breast cancer as early and accurately as possible is crucial for effective treatment and enhanced patient outcomes. Deep transfer learning has emerged as a promising technique for improving breast cancer classification by utilizing pre-trained models and transferring knowledge across related tasks. In this study, we examine the use of a VGG, Vision Transformers (ViT) and Resnet to classify images for Invasive Ductal Carcinoma (IDC) cancer and make a comparative analysis of the algorithms. The result shows a great advantage of Resnet-34 with an accuracy of $90.40\%$ in classifying cancer images. However, the pretrained VGG-16 demonstrates a higher F1-score because there is less parameters to update. We believe that the field of breast cancer diagnosis stands to benefit greatly from the use of deep transfer learning. Transfer learning may assist to increase the accuracy and accessibility of breast cancer screening by allowing deep learning models to be trained with little data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑