Frequency-Aware Guidance for Blind Image Restoration via Diffusion Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xiao, Jun, Lyu, Zihang, Xie, Hao, Zhang, Cong, Ju, Yakun, Shui, Changjian, Lam, Kin-Man
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917841315299328
author Xiao, Jun
Lyu, Zihang
Xie, Hao
Zhang, Cong
Ju, Yakun
Shui, Changjian
Lam, Kin-Man
author_facet Xiao, Jun
Lyu, Zihang
Xie, Hao
Zhang, Cong
Ju, Yakun
Shui, Changjian
Lam, Kin-Man
contents Blind image restoration remains a significant challenge in low-level vision tasks. Recently, denoising diffusion models have shown remarkable performance in image synthesis. Guided diffusion models, leveraging the potent generative priors of pre-trained models along with a differential guidance loss, have achieved promising results in blind image restoration. However, these models typically consider data consistency solely in the spatial domain, often resulting in distorted image content. In this paper, we propose a novel frequency-aware guidance loss that can be integrated into various diffusion models in a plug-and-play manner. Our proposed guidance loss, based on 2D discrete wavelet transform, simultaneously enforces content consistency in both the spatial and frequency domains. Experimental results demonstrate the effectiveness of our method in three blind restoration tasks: blind image deblurring, imaging through turbulence, and blind restoration for multiple degradations. Notably, our method achieves a significant improvement in PSNR score, with a remarkable enhancement of 3.72\,dB in image deblurring. Moreover, our method exhibits superior capability in generating images with rich details and reduced distortion, leading to the best visual quality.
format Preprint
id arxiv_https___arxiv_org_abs_2411_12450
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Frequency-Aware Guidance for Blind Image Restoration via Diffusion Models
Xiao, Jun
Lyu, Zihang
Xie, Hao
Zhang, Cong
Ju, Yakun
Shui, Changjian
Lam, Kin-Man
Computer Vision and Pattern Recognition
Image and Video Processing
Blind image restoration remains a significant challenge in low-level vision tasks. Recently, denoising diffusion models have shown remarkable performance in image synthesis. Guided diffusion models, leveraging the potent generative priors of pre-trained models along with a differential guidance loss, have achieved promising results in blind image restoration. However, these models typically consider data consistency solely in the spatial domain, often resulting in distorted image content. In this paper, we propose a novel frequency-aware guidance loss that can be integrated into various diffusion models in a plug-and-play manner. Our proposed guidance loss, based on 2D discrete wavelet transform, simultaneously enforces content consistency in both the spatial and frequency domains. Experimental results demonstrate the effectiveness of our method in three blind restoration tasks: blind image deblurring, imaging through turbulence, and blind restoration for multiple degradations. Notably, our method achieves a significant improvement in PSNR score, with a remarkable enhancement of 3.72\,dB in image deblurring. Moreover, our method exhibits superior capability in generating images with rich details and reduced distortion, leading to the best visual quality.
title Frequency-Aware Guidance for Blind Image Restoration via Diffusion Models
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2411.12450