CNN Denoisers As Non-Local Filters: The Neural Tangent Denoiser

06/03/2020
by   Julián Tachella, et al.
0

Convolutional Neural Networks (CNNs) are now a well-established tool for solving computational imaging problems. Modern CNN-based algorithms obtain state-of-the-art performance in diverse image restoration problems. Furthermore, it has been recently shown that, despite being highly overparametrized, networks trained with a single corrupted image can still perform as well as fully trained networks, a phenomenon encapsulated in the deep image prior. We introduce a novel interpretation of denoising networks with no clean training data in the context of the neural tangent kernel (NTK), elucidating the strong links with well-known non-local filtering techniques, such as non-local means or BM3D. The filtering function associated with a given network architecture can be obtained in closed form without need to train the network, being fully characterized by the random initialization of the network weights. While the NTK theory accurately predicts the filter associated with networks trained using standard gradient descent, our analysis shows that it falls short to explain the behaviour of networks trained using the popular Adam optimizer. The latter achieves a larger change of weights in hidden layers, adapting the non-local filtering function during training. We evaluate our findings via extensive image denoising experiments.

READ FULL TEXT

page 7

page 17

page 18

11/21/2017

Universal Denoising Networks : A Novel CNN-based Network Architecture for Image Denoising

We design a novel network architecture for learning discriminative image...
11/21/2016

Non-Local Color Image Denoising with Convolutional Neural Networks

We propose a novel deep network architecture for grayscale and color ima...
07/19/2019

Deep Graph-Convolutional Image Denoising

Non-local self-similarity is well-known to be an effective prior for the...
03/06/2018

Nonlocality-Reinforced Convolutional Neural Networks for Image Denoising

We introduce a paradigm for nonlocal sparsity reinforced deep convolutio...
10/31/2019

Denoising and Regularization via Exploiting the Structural Bias of Convolutional Generators

Convolutional Neural Networks (CNNs) have emerged as highly successful t...
07/31/2022

Speckle2Speckle: Unsupervised Learning of Ultrasound Speckle Filtering Without Clean Data

In ultrasound imaging the appearance of homogeneous regions of tissue is...
08/29/2016

About Learning in Recurrent Bistable Gradient Networks

Recurrent Bistable Gradient Networks are attractor based neural networks...