arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.14014cs.CVeess.IV

INDIGO+:一种统一的INN引导概率扩散算法,用于盲和非盲图像恢复

INDIGO+: A Unified INN-Guided Probabilistic Diffusion Algorithm for Blind and Non-Blind Image Restoration

Di You, Pier Luigi Dragotti

AI总结:

本文提出INDIGO和BlindINDIGO,一种结合可逆神经网络与预训练扩散模型的统一算法,用于盲和非盲图像恢复,通过INN模拟退化过程并引导扩散采样,实验证明其性能与最新方法相当。

摘要:

生成扩散模型因其生成逼真自然图像的显著能力,正成为图像恢复(IR)任务中最流行的先验之一。尽管取得了令人满意的结果,基于扩散模型的IR方法仍存在若干局限性。首先,大多数非盲方法需要退化模型的解析表达式来指导采样过程。其次,大多数现有的盲方法依赖于预定义退化模型族来训练其深度网络。上述问题限制了这些方法的灵活性及其处理真实世界退化任务的能力。在本文中,我们提出了一种新颖的INN引导概率扩散算法,用于非盲和盲图像恢复,即INDIGO和BlindINDIGO,该算法结合了可逆神经网络(INN)的完美重建特性与预训练扩散模型的强大生成能力。具体地,我们训练INN的前向过程以模拟任意退化过程,并使用其逆过程获得中间图像,通过梯度步骤指导反向扩散采样过程。我们还引入了一种初始化策略,以进一步提高算法的性能和推理速度。实验表明,在合成和真实世界的低质量图像上,我们的算法在定量和视觉上均与近期领先方法相比获得了有竞争力的结果。

英文摘要:

Generative diffusion models are becoming one of the most popular prior in image restoration (IR) tasks due to their remarkable ability to generate realistic natural images. Despite achieving satisfactory results, IR methods based on diffusion models present several limitations. First of all, most non-blind approaches require an analytical expression of the degradation model to guide the sampling process. Secondly, most existing blind approaches rely on families of pre-defined degradation models for training their deep networks. The above issues limit the flexibility of these approaches and so their ability to handle real-world degradation tasks. In this paper, we propose a novel INN-guided probabilistic diffusion algorithm for non-blind and blind image restoration, namely INDIGO and BlindINDIGO, which combines the merits of the perfect reconstruction property of invertible neural networks (INN) with the strong generative capabilities of pre-trained diffusion models. Specifically, we train the forward process of the INN to simulate an arbitrary degradation process and use the inverse to obtain an intermediate image that we use to guide the reverse diffusion sampling process through a gradient step. We also introduce an initialization strategy, to further improve the performance and inference speed of our algorithm. Experiments demonstrate that our algorithm obtains competitive results compared with recently leading methods both quantitatively and visually on synthetic and real-world low-quality images.

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

↑