A Generalized Deep Learning Model for Denoising Image Datasets

Advance in technology world has lots of contributions from artificial intelligence which is a highly growing area. The failure of traditional algorithms has led to the employment of deep learning algorithms in various fields like pattern recognition, recommendation systems and classification systems. Removal of noise from images can be done using traditional noise removal filters. These filters can either remove more noise that wanted or leave unwanted noise than what is needed in the data. Utilization of Convolutional neural networks designed based on the dataset requirements along with the noise removal filter can yield better results. In this work, evaluation of the performance of convolutional neural network (CNN) against existing image denoising algorithms has been successfully executed . The proposed model is a generalized CNN model which can recognize and classify any type of noisy image given. Two types of model were compared where one model 1 uses the Adam optimizer and model 2 uses the Stochastic Gradient Descent (SGD) optimizer. The image dataset used here is MNIST handwritten dataset, which is trained, tested and validated with both the models by adding three different types of noise viz, Poisson, Salt and Pepper as well as Gaussian Noise. More accuracy and better results were given by the model 2 which uses the SGD optimizer.