Detection of Breast Cancer using Convolutional Neural Networks with Learning Transfer Mechanisms

Abstract

Breast cancer is the leading cause of mortality in women worldwide. One of the biggest challenges for physicians and technological support systems is early detection, because it is easier to treat and establish curative treatments. Currently, assistive technology systems use images to detect patterns of behavior with respect to patients who have been found to have some type of cancer. This work aims to identify and classify breast cancer using deep learning models and convolutional neural networks (CNN) with transfer learning. For the breast cancer detection process, 7803 real images with benign and malignant labels were used, which were provided by BreaKHis on the Kaggle platform. The convolutional basis (parameters) of pre-trained models VGG16, VGG19, Resnet-50 and Inception-V3 were used. The TensorFlow framework, keras and Python libraries were also used to retrain the parameters of the models proposed for this study. Metrics such as accuracy, error ratio, precision, recall and f1-score were used to evaluate the models. The results show that the models based on VGG16, VGG19 ResNet-50 and Inception-V3 obtain an accuracy of 88%, 86%, 97% and 96% respectively, recall of 84%, 82%, 96% and 96% respectively, in addition to f1-score of 86%, 83%, 96% and 95% respectively. It is concluded that the model that shows the best results is Resnet-50, obtaining high results in all the metrics considered, although it should be noted that the Inception-V3 model achieves very similar results in relation to Resnet-50, in all the metrics. In addition, these two models exceed the 95% threshold of correct results.