Automated Breast Cancer Detection Models Based on Transfer Learning

Sensors (Basel). 2022 Jan 24;22(3):876. doi: 10.3390/s22030876.

Abstract

Breast cancer is among the leading causes of mortality for females across the planet. It is essential for the well-being of women to develop early detection and diagnosis techniques. In mammography, focus has contributed to the use of deep learning (DL) models, which have been utilized by radiologists to enhance the needed processes to overcome the shortcomings of human observers. The transfer learning method is being used to distinguish malignant and benign breast cancer by fine-tuning multiple pre-trained models. In this study, we introduce a framework focused on the principle of transfer learning. In addition, a mixture of augmentation strategies were used to prevent overfitting and produce stable outcomes by increasing the number of mammographic images; including several rotation combinations, scaling, and shifting. On the Mammographic Image Analysis Society (MIAS) dataset, the proposed system was evaluated and achieved an accuracy of 89.5% using (residual network-50) ResNet50, and achieved an accuracy of 70% using the Nasnet-Mobile network. The proposed system demonstrated that pre-trained classification networks are significantly more effective and efficient, making them more acceptable for medical imaging, particularly for small training datasets.

Keywords: Nasnet-Mobile; ResNet; breast cancer; deep learning; mammogram; medical imaging; transfer learning.

MeSH terms

  • Breast Neoplasms* / diagnostic imaging
  • Disease Progression
  • Female
  • Humans
  • Machine Learning
  • Mammography
  • Neural Networks, Computer