摘要
基于真实照片的图像修复算法通常通过失真度量(例如,PSNR、SSIM)和感知质量度量(例如,FID、NIQE)进行评估,目标是在不影响感知质量的情况下,尽可能降低失真。为了实现这一目标,当前方法通常尝试从后验分布中采样,或优化失真损失(例如,MSE)和感知质量损失(例如,GAN)的加权和。与之前的工作不同,本文专门关注在完美感知指标约束下最小化 MSE 的最佳估计器,即重建图像的分布等于真实图像的分布。最近的理论结果表明,可以通过将后验均值预测(MMSE 估计)最佳地传输到真实图像的分布来构建这种估计器。受此结果的启发,我们引入了后验均值校正流 (PMRF),这是一种简单但非常有效的算法,它近似于此最佳估计器。具体来说,PMRF 首先预测后验均值,然后使用一个校正流模型将结果传输到高质量图像,该模型近似于所需的最佳传输映射。我们研究了 PMRF 的理论效用,并证明它在各种图像修复任务中始终优于以前的方法。