WaveDM: Wavelet-Based Diffusion Models for Image Restoration
CoRR(2023)
摘要
Latest diffusion-based methods for many image restoration tasks outperform
traditional models, but they encounter the long-time inference problem. To
tackle it, this paper proposes a Wavelet-Based Diffusion Model (WaveDM). WaveDM
learns the distribution of clean images in the wavelet domain conditioned on
the wavelet spectrum of degraded images after wavelet transform, which is more
time-saving in each step of sampling than modeling in the spatial domain. To
ensure restoration performance, a unique training strategy is proposed where
the low-frequency and high-frequency spectrums are learned using distinct
modules. In addition, an Efficient Conditional Sampling (ECS) strategy is
developed from experiments, which reduces the number of total sampling steps to
around 5. Evaluations on twelve benchmark datasets including image raindrop
removal, rain steaks removal, dehazing, defocus deblurring, demoiréing, and
denoising demonstrate that WaveDM achieves state-of-the-art performance with
the efficiency that is comparable to traditional one-pass methods and over
100× faster than existing image restoration methods using vanilla
diffusion models.
更多查看译文
关键词
Diffusion models,image restoration,wavelet transform
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要