SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2406.18516 · 2024

Denoising as Adaptation: Noise-Space Domain Adaptation for Image Restoration

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 20 functions out of this paper's own repositories and ran 17 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
kangliao929/noise-da canonical 17 of 20
FunctionStatusWhere it lives
charbonnier_loss Ran kangliao929/noise-da/models/loss.py
pointer only (licence: NOASSERTION) · get_code("8f4257e8a5570fbd")
default Ran kangliao929/noise-da/models/network.py
pointer only (licence: NONE) · get_code("fbf9ec7be545688e")
dict2str Ran kangliao929/noise-da/core/praser.py
pointer only (licence: NOASSERTION) · get_code("60fc13986769e4c7")
dict_to_nonedict Ran kangliao929/noise-da/core/praser.py
pointer only (licence: NOASSERTION) · get_code("38e77ca35f1f8bcb")
exists Ran kangliao929/noise-da/models/network.py
pointer only (licence: NONE) · get_code("608e364a9d2376a3")
get_first_number Ran kangliao929/noise-da/core/dataset_utils.py
pointer only (licence: NOASSERTION) · get_code("a1abc686d2e52151")
is_image_file Ran kangliao929/noise-da/core/base_dataset.py
pointer only (licence: NONE) · get_code("ab4109634b75ef8b")
mae Ran kangliao929/noise-da/models/metric.py
pointer only (licence: NOASSERTION) · get_code("0570fbc1e2582ae7")
make_dataset Ran kangliao929/noise-da/core/base_dataset.py
pointer only (licence: NOASSERTION) · get_code("fd901efe5f883205")
make_dataset Ran kangliao929/noise-da/core/dataset_utils.py
pointer only (licence: NOASSERTION) · get_code("f3e6bba202d28b8b")
mse_loss Ran kangliao929/noise-da/models/loss.py
pointer only (licence: NONE) · get_code("632959a67ae7fdac")
mse_loss_alpha Ran kangliao929/noise-da/models/loss.py
pointer only (licence: NOASSERTION) · get_code("bbdabf9fe4ed5152")
pil_loader Ran kangliao929/noise-da/core/base_dataset.py
pointer only (licence: NOASSERTION) · get_code("534e80dab769ec2f")
postprocess Ran kangliao929/noise-da/core/util.py
pointer only (licence: NOASSERTION) · get_code("4fd0e47deee01d3d")
psnr Ran kangliao929/noise-da/models/metric.py
pointer only (licence: NOASSERTION) · get_code("095495f59dcb8431")
set_gpu Ran kangliao929/noise-da/core/util.py
pointer only (licence: NOASSERTION) · get_code("24ef4b46c563b220")
tensor2img Ran kangliao929/noise-da/core/util.py
pointer only (licence: NOASSERTION) · get_code("f6bbbcfb2db7dfbc")
extract Not yet run kangliao929/noise-da/models/network.py
pointer only (licence: NONE) · get_code("e894fcce48231d2f")
init_obj Not yet run kangliao929/noise-da/core/praser.py
pointer only (licence: NOASSERTION) · get_code("89ecc1cd0adb6951")
ssim Not yet run kangliao929/noise-da/models/metric.py
pointer only (licence: NOASSERTION) · get_code("bacc659261a196d8")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Although learning-based image restoration methods have made significant progress, they still struggle with limited generalization to real-world scenarios due to the substantial domain gap caused by training on synthetic data. Existing methods address this issue by improving data synthesis pipelines, estimating degradation kernels, employing deep internal learning, and performing domain adaptation and regularization. Previous domain adaptation methods have sought to bridge the domain gap by learning domain-invariant knowledge in either feature or pixel space. However, these techniques often struggle to extend to low-level vision tasks within a stable and compact framework. In this paper, we show that it is possible to perform domain adaptation via the noise space using diffusion models. In particular, by leveraging the unique property of how auxiliary conditional inputs influence the multi-step denoising process, we derive a meaningful diffusion loss that guides the restoration model in progressively aligning both restored synthetic and real-world outputs with a target clean distribution. We refer to this method as denoising as adaptation. To prevent shortcuts during joint training, we present crucial strategies such as channel-shuffling layer and residual-swapping contrastive learning in the diffusion model. They implicitly blur the boundaries between conditioned synthetic and real data and prevent the reliance of the model on easily distinguishable features. Experimental results on three classical image restoration tasks, namely denoising, deblurring, and deraining, demonstrate the effectiveness of the proposed method.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2406.18516")
get_code_for_paper("2406.18516")
have("2406.18516")

Connect an agent — have() is free.