Transfer learning in breast mass detection and classification

Abstract

Covid-19 infection influenced the screening test rate of breast cancer worldwide due to the quarantine measures, routine procedures reduction, and delay of early diagnosis, causing high mortality risk and severity of the disease. X-ray mammography is the gold standard for diagnosing early signs of breast cancer, and Artificial Intelligence enables the detection of suspicious lesions and classifying them in terms of malignancy. This paper aimed to investigate mass detection and classification in a large-scale OPTIMAM dataset with 6000 cases and extracted 3524 images with masses in the mammograms of the Hologic manufacturer. The methodology of the detection step is to train the RetinaNet architecture of ResNet50, ResNet101, and ResNet152 backbones with three types of initializations by ImageNet and COCO weights and from scratch. The dataset was pre-processed to generate two types of input with entire mammograms and patches, which are stated as the first and the second approaches. The results show that in the first approach, RetinaNet of ResNet50 backbone with ImageNet and COCO weights and ResNet152 with the same weights performed 0.91 True Positive Rate at 0.78 False Positive Per Image, respectively. In contrast, in the second approach, ResNet152 with ImageNet weights reached 0.88 TPR at 0.78 FPPI. In the classification step, the Transfer Learning approach was applied with fine-tuning by adding L2-regularization and class weights to balance class distribution in the datasets.