Adapting the pre-trained convolutional neural networks to improve the anomaly detection and classification in mammographic images

Abstract

Mortality from breast cancer (BC) is among the top causes of cancer death in women. BC can be effectively treated when diagnosed early, improving the likelihood that a patient will survive. BC masses and calcification clusters must be identified by mammography in order to prevent disease effects and commence therapy at an early stage. A mammography misinterpretation may result in an unnecessary biopsy of the false-positive results, lowering the patient’s odds of survival. This study intends to improve breast mass detection and identification in order to provide better therapy and reduce mortality risk. A new deep-learning (DL) model based on a combination of transfer-learning (TL) and long short-term memory (LSTM) is proposed in this study to adequately facilitate the automatic detection and diagnosis of the BC suspicious region using the 80–20 method. Since DL designs are modelled to be problem-specific, TL applies the knowledge gained during the solution of one problem to another relevant problem. In the presented model, the learning features from the pre-trained networks such as the squeezeNet and DenseNet are extracted and transferred with the features that have been extracted from the INbreast dataset. To measure the proposed model performance, we selected accuracy, sensitivity, specificity, precision, and area under the ROC curve (AUC) as our metrics of choice. The classification of mammographic data using the suggested model yielded overall accuracy, sensitivity, specificity, precision, and AUC values of 99.236%, 98.8%, 99.1%, 96%, and 0.998, respectively, demonstrating the model’s efficacy in detecting breast tumors.