Transfer Learning based Breast Cancer Classification via Deep Convolutional Neural Network
Markos Wondim Walle, Kula Kekeba Tune, Natnael Tilahun Sinshaw, Sudhir Kumar Mohapatra · International Journal of Engineering and Manufacturing · 2023
Breast cancer is a leading cause of death among women, and the subjectivity of human visual perception and lack of automated detection methods can lead to misclassification of breast cancer images.In this study, a breast cancer classification model using a Convolutional Neural Network (CNN) deep learning algorithm was proposed.The model demonstrated high accuracy in classifying breast images as benign or malignant, with a classification accuracy of 97.1%.The model was also able to run on low computational resources.The study used a dataset of 2009 breast images labeled by two radiologists and included six scenarios based on different hyperparameters, augmentation values, pretrained models, and models built from scratch.While the performance of the proposed model was promising, further improvement may be achieved by using a larger breast image dataset and a machine with more powerful GPU hardware.