Image denoising based on improved stacked sparse denoising autoencoder

Hongqiang MA,Shiping MA,Yuelei XU,Chao LV,Peng XIN,Mingming ZHU
DOI: https://doi.org/10.3778/j.issn.1002-8331.1709-0155
2018-01-01
Abstract:In order to improve the image denoising performance of Stacked Sparse Denoising Auto-encoder(SSDA)and solve the problems of high computational complexity, difficult parameters adjustment and slow training convergence speed, the image denoising algorithm based on Stacked Marginalized Sparse Denoising Auto-encoder(SMSDA)is pro-posed.Marginalized Sparse Denoising Auto-encoder(MSDA)is formed by marginalizing the loss function of SDA net-work because of the fast convergence speed of the Marginalized Denoising Auto-encoder(MDA),which has the character-istics of SAE and MDA.Then,multiple MSDA is stacked to form deep neural network SMSDA.The unsupervised greedy layer-wise training algorithm is used to train each layer of network for avoiding the local optimization of the model parame-ters.The BP(Back Propagation)algorithm is used to fine tune the whole network and can obtain the optimal weight.Last, SMSDA is used to denoise a given image.Simulation results show that the proposed algorithm has higher Peak Signal to Noise Ratio(PSNR)while reducing the computational complexity and improving the convergence rate,retains more details of the original image and has better denoising performance than SSDA.
What problem does this paper attempt to address?