atlas

Research map / Image generation / Image restoration

sRGB Real Noise Modeling via Noise-Aware Sampling with Normalizing Flows

ICLR 2024 · Image generation › Image restoration · approach: Raw and low-light noise modeling

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes NAFlow, a single conditional normalizing flow with per-camera-configuration Gaussian latents and a Noise-Aware Sampling procedure that generates sRGB noise matching an input image without metadata, improving SIDD denoising with DnCNN.

Problem

Generating realistic sRGB real-world noise typically requires camera metadata (ISO, smartphone model) at sampling time, and existing models like NeCA need multiple separate models per camera; spatial noise correlation is also often not modeled.

Approach

A single conditional normalizing flow (conditioned on the clean image) maps noisy images to a mixture of learnable per-camera-configuration Gaussians in latent space; at inference, the input's latent likelihoods under each Gaussian define a mixture from which noise is sampled (Noise-Aware Sampling), removing the need for metadata. A multi-scale squeeze-based architecture captures spatially correla

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use NAFlow rather than NeCA-W for synthetic noise generationsRGB noise generation · SIDD-Validation averageKLD: 0.0342 -> 0.0305Helped
Use NAFlow rather than NeCA-W to generate training noiseDnCNN · SIDD BenchmarkPSNR: 36.82 -> 37.22Helped
Use NAFlow rather than NeCA-W to generate training noiseDnCNN · SIDD BenchmarkSSIM: 0.932 -> 0.935Helped
Use noise-aware sampling over randomly selected latent distributions (ablation)NAFlow · SIDD-ValidationKLD: 0.4700 -> 0.0291Helped
Use all latent distributions rather than the highest-likelihood distribution alone (ablation)NAFlow · SIDD-ValidationKLD: 0.0294 -> 0.0291Helped
Use multiple distinct latent distributions rather than a single standard normal (ablation)NAFlow · SIDD-ValidationKLD: 0.0544 -> 0.0291Helped
Use multiple distinct latent distributions rather than a single standard normal (ablation)NAFlow · SIDD-ValidationAKLD: 0.1657 -> 0.1293Helped
Learn both the Gaussian mean and covariance (ablation)NAFlow · SIDD-ValidationKLDHelped
Increase the number of scale-level blocks (ablation)NAFlow · SIDD-ValidationKLDHelped

Surprising findings

Limitations

Experimental setup

Datasets: SIDD-Medium SIDD-Validation SIDD-Benchmark SIDD+ BSD500 DND

Benchmarks: SIDD-Validation SIDD+ SIDD Benchmark

Metrics: KLD AKLD PSNR SSIM

Baselines: C2N Flow-sRGB NeCA-S NeCA-W NoiseFlow

Builds on: Glow RealNVP conditional normalizing flow NeCA Flow-sRGB

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy