Extra Proximal-Gradient Inspired Non-local Network
arXiv:1911.07144
Abstract
Variational method and deep learning method are two mainstream powerful approaches to solve inverse problems in computer vision. To take advantages of advanced optimization algorithms and powerful representation ability of deep neural networks, we propose a novel deep network for image reconstruction. The architecture of this network is inspired by our proposed accelerated extra proximal gradient algorithm. It is able to incorporate non-local operation to exploit the non-local self-similarity of the images and to learn the nonlinear transform, under which the solution is sparse. All the parameters in our network are learned from minimizing a loss function. Our experimental results show that our network outperforms several state-of-the-art deep networks with almost the same number of learnable parameter.
12 pages, 7 figures
References in corpus (5)
- Adam: A Method for Stochastic Optimization
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Understanding the Effective Receptive Field in Deep Convolutional Neural Networks
- Learning Deep CNN Denoiser Prior for Image Restoration
- One Network to Solve Them All --- Solving Linear Inverse Problems using Deep Projection Models