Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning to Generate Realistic Noisy Images via Pixel-level Noise-aware Adversarial Training

About

Existing deep learning real denoising methods require a large amount of noisy-clean image pairs for supervision. Nonetheless, capturing a real noisy-clean dataset is an unacceptable expensive and cumbersome procedure. To alleviate this problem, this work investigates how to generate realistic noisy images. Firstly, we formulate a simple yet reasonable noise model that treats each real noisy pixel as a random variable. This model splits the noisy image generation problem into two sub-problems: image domain alignment and noise domain alignment. Subsequently, we propose a novel framework, namely Pixel-level Noise-aware Generative Adversarial Network (PNGAN). PNGAN employs a pre-trained real denoiser to map the fake and real noisy images into a nearly noise-free solution space to perform image domain alignment. Simultaneously, PNGAN establishes a pixel-level adversarial training to conduct noise domain alignment. Additionally, for better noise fitting, we present an efficient architecture Simple Multi-scale Network (SMNet) as the generator. Qualitative validation shows that noise generated by PNGAN is highly similar to real noise in terms of intensity and distribution. Quantitative experiments demonstrate that a series of denoisers trained with the generated noisy images achieve state-of-the-art (SOTA) results on four real denoising benchmarks. Part of codes, pre-trained models, and results are available at https://github.com/caiyuanhao1998/PNGAN for comparisons.

Yuanhao Cai, Xiaowan Hu, Haoqian Wang, Yulun Zhang, Hanspeter Pfister, Donglai Wei• 2022

Related benchmarks

TaskDatasetResultRank
Image DenoisingSIDD (val)
PSNR39.98
153
Image DenoisingCC
PSNR36.1
64
Image DenoisingHighISO
PSNR38.24
48
Image DenoisingPoly
PSNR37.41
24
Image DenoisingXiaomi
PSNR35.24
24
Image DenoisingHuawei
PSNR38.02
24
Image DenoisingOPPO
PSNR39.56
24
Image DenoisingSony
PSNR43.15
24
Image DenoisingiPhone
PSNR39.93
24
Image DenoisingSIDD 49 (val)
PSNR40.07
11
Showing 10 of 13 rows

Other info

Code

Follow for update