论文

用于模糊图像复原的逆热扩散动态时间步预测

Dynamic Time Step Prediction in Inverse Heat Dissipation for Blur-Like Image Restoration Tasks

模型推理解码与生成控制

摘要

当使用扩散模型来解决图像恢复问题时,通常采用扩散反演来保留退化图像中的相关图像信息。许多方法不是反转回初始时间步长(即 T),而是反转到预定的中间时间步长,以便更好地保留来自退化源图像的信息。然而,用于反演的预定时间步长对于重建来说并不理想,因为严重退化的图像需要比轻度退化的图像更早的开始时间步长。此外,基于 DDIM 的模型通过添加高斯噪声来破坏原始信号,这可能与模糊、雾霾和低光等模糊退化的性质不匹配。为了解决这些问题,我们提出了两种解决方案:(1)我们采用另一种扩散过程,称为逆散热模型,通过逐渐模糊数据点来扩散输入图像(2)我们建议实现一个时间预测器来估计反演的起始时间步,模型学习适应退化严重程度。对标准基准的大量实验表明,我们的方法在定量和定性评估方面都实现了最先进的性能,并且对许多恢复任务具有出色的泛化能力。

用于模糊图像复原的逆热扩散动态时间步预测的原论文方法或结果图
图 1:我们的方法的实际应用。我们使用现实数据集 RealBlur [42] 中的图像进行去模糊任务,使用互联网图像进行去雾和低光增强任务。我们的方法可以产生高保真度的结果。