arXiv:2410.00418v3 通知类型: 交叉替换 摘要：照片 realism 图像恢复算法通常通过失真度量（例如 PSNR、SSIM）和感知质量度量（例如 FID、NIQE）进行评估，目标是在不失真感知质量的情况下获得最低可能的失真。为了实现这一目标，当前方法通常尝试从后验分布中采样，或者优化失真损失（例如 MSE）和感知质量损失（例如 GAN）的加权和。与以往工作不同，本文特别关注在完美感知指标约束下最小化 MSE 的最佳估计器，即重建图像的分布等于真实图像的分布。最近的理论结果表明，可以通过最优运输后验均值预测（MMSE 估计）到真实图像的分布来构造这样的估计器。受这一结果的启发，我们引入了后验均值校正流（Posterior-Mean Rectified Flow，PMRF），这是一种简单但非常有效的算法，旨在近似这种最佳估计器。具体而言，PMRF 首先预测后验均值，然后使用近似所需最优运输映射的校正流模型将结果传输到高质量图像。我们研究了 PMRF 的理论用途，并证明了它在各种图像恢复任务中始终优于 previous 方法。