Denoising as Adaptation: Noise-Space Domain Adaptation for Image Restoration (doi:10.21979/N9/DMB2QK)

View:

Part 1: Document Description
Part 2: Study Description
Entire Codebook

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

doi:10.21979/N9/DMB2QK

DR-NTU (Data)

2025-02-04

1

Liao, Kang; Yue, Zongsheng; Wang, Zhouxia; Loy, Chen Change, 2025, "Denoising as Adaptation: Noise-Space Domain Adaptation for Image Restoration", https://doi.org/10.21979/N9/DMB2QK, DR-NTU (Data), V1

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

doi:10.21979/N9/DMB2QK

Liao, Kang (Nanyang Technological University)

Yue, Zongsheng (Nanyang Technological University)

Wang, Zhouxia (Nanyang Technological University)

Loy, Chen Change (Nanyang Technological University)

LaTex

PyTorch

DR-NTU (Data)

Liao, Kang

Liao, Kang

2025-01-23

https://doi.org/10.21979/N9/DMB2QK

Computer and Information Science, Computer and Information Science, Diffusion Models, Domain Adaptation, Image Restoration

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.

Code and Data

S-Lab License 1.0 <br/> Copyright 2025 S-Lab <br/><br/> Redistribution and use for non-commercial purpose in source and binary forms, with or without modification, are permitted provided that the following conditions are met: <br/> 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. <br/> 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. <br/> 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission. <br/><br/> THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. <br/><br/> In the event that redistribution and/or use for commercial purpose in source or binary forms, with or without modification is required, please contact the contributor(s) of the work.

<a href="https://github.com/KangLiao929/Noise-DA">Github</a>

10.48550/arXiv.2406.18516

Liao, K., Yue, Z., Wang, Z., & Loy, C. C. (2024). Denoising as adaptation: Noise-space domain adaptation for image restoration. arXiv preprint arXiv:2406.18516.