arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2411.12450cs.CVeess.IV

基于扩散模型的盲图像恢复频率感知引导

Frequency-Aware Guidance for Blind Image Restoration via Diffusion Models

The Hong Kong Polytechnic University(香港理工大学);Nanyang Technological University(南洋理工大学);Vector Institute(向量研究所)

Jun Xiao, Zihang Lyu, Hao Xie, Cong Zhang, Yakun Ju, Changjian Shui, Kin-Man Lam

AI总结:

本文提出一种基于二维离散小波变换的频率感知引导损失,以即插即用方式集成到扩散模型中,在空间域和频率域同时约束内容一致性,显著提升盲图像去模糊、湍流成像及多重退化恢复的性能。

摘要:

盲图像恢复在低级视觉任务中仍然是一个重大挑战。近年来,去噪扩散模型在图像合成中展现出卓越的性能。引导扩散模型利用预训练模型强大的生成先验以及可微分的引导损失,在盲图像恢复中取得了有前景的成果。然而,这些模型通常仅在空间域中考虑数据一致性,往往导致图像内容失真。在本文中,我们提出了一种新颖的频率感知引导损失,可以以即插即用的方式集成到各种扩散模型中。我们提出的引导损失基于二维离散小波变换,同时在空间域和频率域中强制内容一致性。实验结果证明了我们的方法在三个盲恢复任务中的有效性:盲图像去模糊、湍流成像以及多重退化的盲恢复。值得注意的是,我们的方法在PSNR分数上取得了显著提升,在图像去模糊中实现了3.72 dB的显著增强。此外,我们的方法在生成具有丰富细节和减少失真的图像方面表现出优越的能力,从而获得最佳的视觉质量。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑