Classification of Mammography Images by Transfer Learning

Abstract

Breast cancer is the most common cancer type in women worldwide. Diagnosis and early detection of cancer by mammography images are of great importance in cancer treatment. The use of deep learning in Computer Assisted Diagnostic systems has gained a great momentum especially since 2012. In this study, benign and malignant mass images were reproduced with data augmentation and the data sets obtained were classified with deep learning networks. In this study, a scratch Convolutional Neural Network (CNN) architecture was created and transfer learning was realized with different network models which trained on IMAGENET images. In the transfer learning section, separate training results were obtained by performing feature extraction and fine tuning of network parameters. As a result of the study, the best results were obtained with MobileNet, NASNetLarge and InceptionResNetV2 models which are used in transfer learning models.