Contrast Enhancement in Mammograms Using Convolution Neural Networks for Edge Computing Systems

Abstract

A good contrast is significant for analysis of medical images, and if the images have poor contrast, then some methods of contrast enhancement can be of much benefit. In this paper, a convolution neural network-based transfer learning approach is utilized for contrast enhancement of mammographic images. The experiments are conducted on ISP and MIAS datasets, where ISP dataset is used for training and MIAS dataset is used for testing (contrast enhancement). Experimental comparison of the proposed technique is done with the most popular direct and indirect contrast enhancement techniques such as CLAHE, BBHE, RMSHE, and contrast stretching. A qualitative comparison is done using mean square error (MSE), signal to noise ratio (SNR), and peak signal to noise ratio (PSNR). It is observed that the proposed technique outperforms the other techniques HE, RMSHE, CLAHE, BBHE, and contrast stretching.