Stimulating Diffusion Model for Image Denoising via Adaptive Embedding and Ensembling

Tong Li, Hansen Feng, Lizhi Wang*, Lin Zhu, Zhiwei Xiong, Hua Huang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Image denoising is a fundamental problem in computational photography, where achieving high perception with low distortion is highly demanding. Current methods either struggle with perceptual quality or suffer from significant distortion. Recently, the emerging diffusion model has achieved state-of-the-art performance in various tasks and demonstrates great potential for image denoising. However, stimulating diffusion models for image denoising is not straightforward and requires solving several critical problems. For one thing, the input inconsistency hinders the connection between diffusion models and image denoising. For another, the content inconsistency between the generated image and the desired denoised image introduces distortion. To tackle these problems, we present a novel strategy called the Diffusion Model for Image Denoising (DMID) by understanding and rethinking the diffusion model from a denoising perspective. Our DMID strategy includes an adaptive embedding method that embeds the noisy image into a pre-trained unconditional diffusion model and an adaptive ensembling method that reduces distortion in the denoised image. Our DMID strategy achieves state-of-the-art performance on both distortion-based and perception-based metrics, for both Gaussian and real-world image denoising.

Original languageEnglish
Pages (from-to)8240-8257
Number of pages18
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume46
Issue number12
DOIs
Publication statusPublished - 2024

Keywords

  • Computational photography
  • diffusion model
  • distortion-perception
  • image denoising
  • self-supervised

Fingerprint

Dive into the research topics of 'Stimulating Diffusion Model for Image Denoising via Adaptive Embedding and Ensembling'. Together they form a unique fingerprint.

Cite this