LLM2D
后验均值钳位流:朝着最小均方误差的 PHOTO-REALISTIC 图像恢复
Posterior-Mean Rectified Flow: Towards Minimum MSE Photo-Realistic Image Restoration
作者: Guy Ohayon, Tomer Michaeli, Michael Elad
发布日期: 2/5/2025
arXiv ID: 2410.00418

摘要

arXiv:2410.00418v3 宣告类型: replace-cross 摘要:现实照片级图像恢复算法通常通过失真度量(例如,PSNR、SSIM)和感知质量度量(例如,FID、NIQE)进行评估,目标是在不牺牲感知质量的情况下达到最低可能的失真。为了实现这一目标,当前的方法通常尝试从后验分布中采样,或者优化失真损失(例如,MSE)和感知质量损失(例如,GAN)的加权和。与以往的工作不同,本文特别关注在完美感知索引约束下的最小MSE的最优估计器,即重建图像的分布与地面真实图像的分布相同。最近的一个理论成果表明,可以通过将后验均值预测(MMSE估计)最优传输到地面真实图像的分布来构建这样的估计器。受这一成果的启发,我们引入了后验均值校正流(PMRF),这是一种简单而高效的算法,用于近似此最优估计器。具体而言,PMRF首先预测后验均值,然后使用一个近似所需最优传输映射的校正流模型将结果传输到高质量图像。我们探讨了PMRF的理论用途,并展示了在多种图像恢复任务中,它始终优于以往的方法。