Noise2Score: Tweedie's Approach to Self-Supervised Image Denoising without Clean Images

Recently, there has been extensive research interest in training deep networks to denoise images without clean reference. However, the representative approaches such as Noise2Noise, Noise2Void, Stein’s unbiased risk estimator (SURE), etc. seem to differ from one another and it is difficult to find the coherent mathematical structure. To address this, here we present a novel approach, called Noise2Score, which reveals a missing link in order to unite these seemingly different approaches. Specifically, we show that image denoising problems without clean images can be addressed by finding the mode of the posterior distribution and that the Tweedie’s formula offers an explicit solution through the score function (i.e. the gradient of loglikelihood). Our method then uses the recent finding that the score function can be stably estimated from the noisy images using the amortized residual denoising autoencoder, the method of which is closely related to Noise2Noise or Nose2Void. Our Noise2Score approach is so universal that the same network training can be used to remove noises from images that are corrupted by any exponential family distributions and noise parameters. Using extensive experiments with Gaussian, Poisson, and Gamma noises, we show that Noise2Score significantly outperforms the state-of-the-art self-supervised denoising methods in the benchmark data set such as (C)BSD68, Set12, and Kodak, etc.

[1]  Raymond H. Chan,et al.  Multilevel algorithm for a Poisson noise removal model with total-variation regularization , 2007 .

[2]  Jaakko Lehtinen,et al.  Noise2Noise: Learning Image Restoration without Clean Data , 2018, ICML.

[3]  H. Robbins An Empirical Bayes Approach to Statistics , 1956 .

[4]  Loïc Royer,et al.  Noise2Self: Blind Denoising by Self-Supervision , 2019, ICML.

[5]  Ing Rj Ser Approximation Theorems of Mathematical Statistics , 1980 .

[6]  Pascal Vincent,et al.  A Connection Between Score Matching and Denoising Autoencoders , 2011, Neural Computation.

[7]  Eero P. Simoncelli,et al.  Least Squares Estimation Without Priors or Supervision , 2011, Neural Computation.

[8]  Se Young Chun,et al.  Extending Stein's unbiased risk estimator to train deep denoisers with correlated pairs of noisy images , 2019, NeurIPS.

[9]  Yoshua Bengio,et al.  What regularized auto-encoders learn from the data-generating distribution , 2012, J. Mach. Learn. Res..

[10]  Florian Jug,et al.  Noise2Void - Learning Denoising From Single Noisy Images , 2018, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[11]  Jitendra Malik,et al.  A database of human segmented natural images and its application to evaluating segmentation algorithms and measuring ecological statistics , 2001, Proceedings Eighth IEEE International Conference on Computer Vision. ICCV 2001.

[12]  Luca Antiga,et al.  Automatic differentiation in PyTorch , 2017 .

[13]  Shuiwang Ji,et al.  Noise2Same: Optimizing A Self-Supervised Bound for Image Denoising , 2020, NeurIPS.

[14]  José M. Bioucas-Dias,et al.  Multiplicative Noise Removal Using Variable Splitting and Constrained Optimization , 2009, IEEE Transactions on Image Processing.

[15]  Hua Zhang,et al.  A convex nonlocal total variation regularization algorithm for multiplicative noise removal , 2019, EURASIP J. Image Video Process..

[16]  Se Young Chun,et al.  Unsupervised Training of Denoisers for Low-Dose CT Reconstruction Without Full-Dose Ground Truth , 2020, IEEE Journal of Selected Topics in Signal Processing.

[17]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[18]  Alessandro Foi,et al.  Optimal Inversion of the Anscombe Transformation in Low-Count Poisson Image Denoising , 2011, IEEE Transactions on Image Processing.

[19]  Aaron C. Courville,et al.  AR-DAE: Towards Unbiased Neural Entropy Gradient Estimation , 2020, ICML.

[20]  Nick Moran,et al.  Noisier2Noise: Learning to Denoise From Unpaired Noisy Data , 2019, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[21]  Luc Van Gool,et al.  NTIRE 2018 Challenge on Single Image Super-Resolution: Methods and Results , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[22]  Thierry Blu,et al.  Monte-Carlo Sure: A Black-Box Optimization of Regularization Parameters for General Denoising Algorithms , 2008, IEEE Transactions on Image Processing.

[23]  Jaakko Lehtinen,et al.  High-Quality Self-Supervised Deep Image Denoising , 2019, NeurIPS.

[24]  B. Efron Tweedie’s Formula and Selection Bias , 2011, Journal of the American Statistical Association.

[25]  Aapo Hyvärinen,et al.  Estimation of Non-Normalized Statistical Models by Score Matching , 2005, J. Mach. Learn. Res..

[26]  Elsa D. Angelini,et al.  An Unbiased Risk Estimator for Image Denoising in the Presence of Mixed Poisson–Gaussian Noise , 2014, IEEE Transactions on Image Processing.

[27]  Karen O. Egiazarian,et al.  Image denoising with block-matching and 3D filtering , 2006, Electronic Imaging.

[28]  C. Stein Estimation of the Mean of a Multivariate Normal Distribution , 1981 .

[29]  Se Young Chun,et al.  Training and Refining Deep Learning Based Denoisers without Ground Truth Data. , 2018 .

[30]  Pascal Vincent,et al.  Stacked Denoising Autoencoders: Learning Useful Representations in a Deep Network with a Local Denoising Criterion , 2010, J. Mach. Learn. Res..