Total generalized variational-liked network for image denoising

Zhang Xiaohua,Lian Qiusheng,Zhang Dan
DOI: https://doi.org/10.1007/s10489-022-03717-8
IF: 5.3
2022-08-11
Applied Intelligence
Abstract:Deep convolutional neural networks (DCNN) have been widely used in the field of image denoising because of their fast inference and good performance. However, the design of networks for the DCNN is mostly empirical, and the interpretation and robustness of them remains a major challenge. Inspired by the total generalized variation method, this paper proposes a novel adaptive denoising network. It mainly improves the TGV algorithm in terms of two points. Firstly, the first and second order derivation term are replaced by the learnable operators. Secondly, the regularization terms are learned from the training data by using convolution networks other than the fixed ones. The network design derived from the process for tackling the denoising problem based on the primal-dual hybrid gradient optimization algorithm, is called TGVLNet- Total Generalized Variational-Liked Network, which allows for the image prior and the linear operators to be tuned differently in each iteration, and enhances the flexibility and generalization ability of the network. The experiment results of Gaussian noise removal and signal dependent noise removal manifest that the proposed network has superior performance and generalization. Compared to most of the blind denoising methods with additive white Gaussian noise, the proposed TGVLNet performs better in unseen noise level. It is noting that we train the model only on the synthetic image of the signal dependent noise removal, and use the model to remove the noise of some images on two real datasets, i.e NC12 and Nam, the denoising results also presents much better visual quality and performance, which further verify the generalization and robustness of our method.
computer science, artificial intelligence
What problem does this paper attempt to address?