Detection and classification of breast cancer types using VGG16 and ResNet50 deep learning techniques

Ashwini P., Suguna N., Vadivelan N.

Abstract


Breast cancer has become a major worldwide health issue, accounting for a large portion of the mortality rate among women. As a result, the need for early detection techniques to enhance prognosis is increasing. Many techniques are being used to detect breast cancer early, and treatment outcomes are frequently favorable when the disease is detected early. Mammography is a commonly used and very successful method for identifying breast cancer among these modalities. Through additional image processing operations like resizing and normalizing, this technology allows the detection of malignant spots from mammography pictures of the affected area. The goal of our research is to improve breast cancer detection and diagnosis speed and accuracy. In this study, we investigate the use of deep learning methods, particularly the visual geometry group (VGG16) and ResNet50 models, for mammography-based breast cancer detection. We assess the performance of the VGG16 and ResNet50 models by training and testing on a mammogram dataset that consists of 322 images from the mammographic image analysis society (MIAS) dataset. The suggested models aim to classify these images into normal, benign, and malignant groupings. Our results show better performance when compared to existing approaches. The proposed methods VGG16 and ResNet50 show promising results, achieving a classification accuracy of 91.23% and 99.01% respectively.

Keywords


Adaptive moment estimation optimizer; Breast cancer classification mammogram; Deep learning; ResNet50; VGG16

Full Text:

PDF


DOI: http://doi.org/10.11591/ijece.v14i5.pp5481-5488

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

International Journal of Electrical and Computer Engineering (IJECE)
p-ISSN 2088-8708, e-ISSN 2722-2578

This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).