Neighbor2Neighbor: Self-Supervised Denoising from Single Noisy Images

Tao Huang, Songjiang Li, Xu Jia, Huchuan Lu, Jianzhuang Liu
1/8/2021
eess.IVcs.CV

Official code: TaoHuang2018/Neighbor2Neighbor (302 stars, BSD-3-Clause, last commit 2022-08-10).

Abstract

In the last few years, image denoising has benefited a lot from the fast development of neural networks. However, the requirement of large amounts of noisy-clean image pairs for supervision limits the wide use of these models. Although there have been a few attempts in training an image denoising model with only single noisy images, existing self-supervised denoising approaches suffer from inefficient network training, loss of useful information, or dependence on noise modeling. In this paper, we present a very simple yet effective method named Neighbor2Neighbor to train an effective image denoising model with only noisy images. Firstly, a random neighbor sub-sampler is proposed for the generation of training image pairs. In detail, input and target used to train a network are images sub-sampled from the same noisy image, satisfying the requirement that paired pixels of paired images are neighbors and have very similar appearance with each other. Secondly, a denoising network is trained on sub-sampled training pairs generated in the first stage, with a proposed regularizer as additional loss for better performance. The proposed Neighbor2Neighbor framework is able to enjoy the progress of state-of-the-art supervised denoising networks in network architecture design. Moreover, it avoids heavy dependence on the assumption of the noise distribution. We explain our approach from a theoretical perspective and further validate it through extensive experiments, including synthetic experiments with different noise distributions in sRGB space and real-world experiments on a denoising benchmark dataset in raw-RGB space.

View on arXivView PDF

Code Implementations(1)

BSD-3-Clause

Neighbor2Neighbor: Self-Supervised Denoising from Single Noisy Images

30245PythonMar 9, 20214 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{huang2021neighborneighbor,
  title  = {Neighbor2Neighbor: Self-Supervised Denoising from Single Noisy Images},
  author = {Tao Huang and Songjiang Li and Xu Jia and Huchuan Lu and Jianzhuang Liu},
  year   = {2021},
  eprint = {2101.02824},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2101.02824}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2101.02824.svg)](https://codeofpaper.com/paper/2101.02824)

Discussion