Breast lesion classification from mammograms using deep neural network and test-time augmentation

Parita Oza,Paawan Sharma,Samir Patel
DOI: https://doi.org/10.1007/s00521-023-09165-w
2023-11-20
Neural Computing and Applications
Abstract:In order to effectively aid in the automatic classification of the breast cancer suspicious region, a new deep-learning (DL) model built on the transfer-learning (TL) technique is proposed in this work. Pretrained convolutional neural networks (CNN) architectures, such as VGG-16, ResNet-50, Inception-V3, and Efficientnet-B7 are used in the proposed model to extract the features from the MIAS and CBIS-DDSM datasets. Three scenarios are used to evaluate the performance of each network under consideration: TL for the original dataset, TL for the pre-processed dataset, and a novel approach of TL with test time augmentation (TTA). Concerning limited training datasets, the proposed approach showed that pre-trained classification networks with test time augmentation are much more effective and efficient, making them more acceptable for medical imaging. With accuracy (99.97%), specificity (99.24%), sensitivity (98.50%), and F1-score (98.74%), the proposed test time augmentation strategy with transfer learning outperforms other state-of-the-art methods on the MIAS dataset. The proposed model could also do well on the CBIS-DDSM dataset with accuracy (99.8%), specificity (95.44%), sensitivity (96.57%), and F1-score (97.22%). Additionally, on the mammography predictions provided by our algorithm, we also sought the advice of a radiology professional. The model’s predictions and the expert’s assessment are almost in sync.
computer science, artificial intelligence
What problem does this paper attempt to address?