Application of VGG16 Transfer Learning for Breast Cancer Detection

被引：0

作者：

Fatima, Tanjim ^{[1
]}

Soliman, Hamdy ^{[1
]}

机构：

[1] Department of Computer Science, New Mexico Tech, Socorro, 87801, NM

来源：

Information (Switzerland) | 2025年 / 16卷 / 03期

关键词：

breast cancer detection; deep learning; histopathology; image classification; transfer learning; VGG16;

D O I：

10.3390/info16030227

中图分类号：

学科分类号：

摘要：

Breast cancer is among the primary causes of cancer-related deaths globally, highlighting the critical need for effective and early diagnostic methods. Traditional diagnostic approaches, while valuable, often face limitations in accuracy and accessibility. Recent advancements in deep learning, particularly transfer learning, provide promising solutions for enhancing diagnostic precision in breast cancer detection. Due to the limited capability of the BreakHis dataset, transfer learning was utilized to advance the training of our new model with the VGG16 neural network model, well trained on the rich ImageNet dataset. Moreover, the VGG16 architecture was carefully modified, including the fine-tuning of its layers, yielding our new model: M-VGG16. The new M-VGG16 model is designed to carry out the binary cancer/benign classification of breast samples effectively. The experimental results of our M-VGG16 model showed it achieved high validation accuracy (93.68%), precision (93.22%), recall (97.91%), and a high AUC (0.9838), outperforming other peer models in the same field. This study validates the VGG16 model’s suitability for breast cancer detection via transfer learning, providing an efficient, adaptable framework for improving diagnostic accuracy and potentially enhancing breast cancer detection. Key breast cancer detection challenges and potential M-VGG16 model refinements are also discussed. © 2025 by the authors.

引用

共 50 条

[1] Howlader N., Noone A.M., Krapcho M., Miller D., Brest A., Yu M., Ruhl J., Tatalovich Z., Mariotto A., Lewis D.R., Et al., SEER Cancer Statistics Review, 1975–2017, (2020)
[2] Bashar M.A., Begam N., Breast cancer surpasses lung cancer as the most commonly diagnosed cancer worldwide, Indian J. Cancer, 59, pp. 438-439, (2022)
[3] Arnold M., Morgan E., Rumgay H., Mafra A., Singh D., Laversanne M., Vignat J., Gralow J.R., Cardoso F., Siesling S., Et al., Current and future burden of breast cancer: Global statistics for 2020 and 2040, Breast, 66, pp. 15-23, (2022)
[4] Nover A.B., Jagtap S., Anjum W., Yegingil H., Shih W.Y., Shih W.H., Brooks A.D., Modern breast cancer detection: A technological review, J. Biomed. Imaging, 2009, (2009)
[5] Nelson H.D., Tyne K., Naik A., Bougatsos C., Chan B.K., Humphrey L., Screening for breast cancer: An update for the US Preventive Services Task Force, Ann. Intern. Med, 151, pp. 727-737, (2009)
[6] Yedjou C.G., Tchounwou S.S., Alo R.A., Elhag R., Mochona B., Latinwo L., Application of machine learning algorithms in breast cancer diagnosis and classification, Int. J. Sci. Acad. Res, 2, (2021)
[7] Rodriguez L.G., Caballero J.M., Niguidula J.D., Calibo D.I., Rodriguez C.A., eCommerce Sales Attrition: A Business Intelligence Visualization, Proceedings of the Big Data Technologies and Applications: 8th International Conference, BDTA 2017, pp. 107-112, (2018)
[8] Pan S.J., Yang Q., A survey on transfer learning, IEEE Trans. Knowl. Data Eng, 22, pp. 1345-1359, (2010)
[9] Shin H.C., Roth H.R., Gao M., Lu L., Xu Z., Nogues I., Yao J., Mollura D., Summers R.M., Deep convolutional neural networks for computer-aided detection: CNN architectures, dataset characteristics and transfer learning, IEEE Trans. Med. Imaging, 35, pp. 1285-1298, (2016)
[10] Raghu M., Zhang C., Kleinberg J., Bengio S., Transfusion: Understanding transfer learning for medical imaging, Adv. Neural Inf. Process. Syst, 32, pp. 1-22, (2019)

← 1 2 3 4 5 →