Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Neighbor2Neighbor: Self-Supervised Denoising from Single Noisy Images

About

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.

Tao Huang, Songjiang Li, Xu Jia, Huchuan Lu, Jianzhuang Liu• 2021

Related benchmarks

TaskDatasetResultRank
Image DenoisingSIDD (val)
PSNR28
105
Image DenoisingDND
PSNR31.4
99
Image DenoisingBSD300
PSNR (dB)30.79
78
Image DenoisingKodak
PSNR32.1
45
Image DenoisingSet14
PSNR31.09
45
Image DenoisingKodak (test)
PSNR31.88
42
Gaussian DenoisingKodak
PSNR32.1
41
Poisson DenoisingDIV2K
PSNR33.37
40
Poisson DenoisingKodak
PSNR31.44
40
Image DenoisingSIDD raw-RGB (val)
PSNR51.39
24
Showing 10 of 62 rows

Other info

Code

Follow for update