Breast mass regions classification from mammograms using convolutional neural networks and transfer learning.

Abstract

This study introduces a novel approach aimed at enhancing the quality of digital mammography images through pre-processing techniques, to improve breast cancer detection accuracy. The primary objective is to enhance image resolution, thus leading to more precise breast tissue segmentation and subsequent classification utilizing convolutional neural networks (CNNs). Three recognized public mammography databases: CBIS-DDSM, Mini-MIAS, and Inbreast were used as pre-processing data. Our statistical findings revealed that the EDSR method (PSNR = 39.05 dB/ SSIM = 0.90) consistently outperformed the visual quality of images when compared to SR-RDN (PSNR = 32.68 dB/SSIM = 0.82). Similarly, UNet demonstrated superior performance over SegNet, boasting an average Intersection over Union (IoU) of 0.862, an average Dice coefficient of 0.991, and an accuracy rate of 0.947 in Region of Interest (RoI) segmentation results. In conclusion, the ResNet model contributed to enhanced accuracy compared to conventional machine learning algorithms. However, it did not surpass state-of-the-art deep CNN-based classifiers, achieving an accuracy rate of 75%. Abbreviations: AUC: Area under curve; CAD: Computer aided system; CC: Cranio caudal; CNN: Convolutional neural network; DNN: Deep neural network; DDSM: Digital Database for Screening Mammography; DM: Digital mammography; DL: Deep learning; EDSR: Enhanced Deep Residual Network; E2E: End to End; ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks; ESPCN: Efficient sub-pixel convolutional neural network; GAN: Generative adversarial network; HR: High resolution; IoU: Intersection over Union; LR: Low resolution; MDSR: Multi-scale deep super-resolution; MLO: Mediolateral Oblique; PSNR: Peak signal to Noise Ratio; RoI: Region of interest; RDN: Residua Dense Network; RDB: Residual Dense Block; RNN: Recurrent Neural Network; ReLU: Rectified Linear Unit; SR-GAN: Super-Resolution Using a Generative Adversarial Network; SSIM: Structural Similarity Index Metric; SISR: Single image super resolution; SegNet: Segmentation Network; TP: True positive; TN: True negative; FP: False positive; FN: False negative; VGG: Visual geometric group; VDSR: Very Deep Network for SR