Breast Mass Tumor Classification using Deep Learning

Abstract

This study presents pre-trained Convolutional Neural Network (CNN) models to classify pre-segmented mammogram mass tumors as benign or malignant. Transfer learning, particular pre-processing and data augmentation were applied to overcome the limitation of the scarcity of available training dataset. The proposed models are based on modified versions of Inception V3 and ResNet50 to tackle the classification problem mentioned above. The proposed architectures have been tested on the Digital Database for Screening Mammography (DDSM) dataset, and it achieved an accuracy of 0.796, precision of 0.754, and a recall of 0.891 on InceptionV3-like CNN model. On the other hand, an accuracy of 0.857, precision of 0.857, and a recall rate of 0.873 have been achieved with the ResNet50-like CNN network. Overall, the proposed ResNet50-like model achieved a ~5% improvement in accuracy compared to the existing state-of-the-art method for this dataset.

Ähnliche Arbeiten