Deep-CNN for Disease Classification using Enhanced Mammographic Images

Abstract

Breast cancer has become one of the most common diseases that women face today as a result of poor nutrition and other environmental factors. A mammogram image of the breast will help detect breast cancer, but still, sometimes doctors and radiologists were unable to detect it due to poor image quality or abnormal region that appears to be normal. In this paper, a deep CNN-based classification model is proposed that classifies the mammogram image as normal, masses and micro-calcification. Firstly, the PSNR values of the mammogram images is improved using a median filter with the local contrast modification (LCM) method. It is further enhanced by Adaptive-CLAHE in con- junction with the Wiener filter. After image enhancement, the region of interest is segmented through morphological feature extraction and Otsu thresholding method. In order to increase the number of samples in the mammogram image dataset, image data augmentation is applied on segmented images. Finally, a pre- trained ResNet model is used for classification of mammogram images. The proposed model has shown improved PSNR for mammogram images and achieved higher classification accuracy of 98.91%, thus outperformed other existing methods. Addition- ally, the explainability and causality of the proposed model is also discussed to show the learning process of the model.