LLM2D
后验均值归一化流:迈向最小均方误差的photo-realistic图像恢复
Posterior-Mean Rectified Flow: Towards Minimum MSE Photo-Realistic Image Restoration
作者: Guy Ohayon, Tomer Michaeli, Michael Elad
发布日期: 2/4/2025
arXiv ID: oai:arXiv.org:2410.00418v2

摘要

arXiv:2410.00418v2 公告类型: 替换-交叉 摘要:照片逼真图像恢复算法通常通过失真度量(例如,PSNR,SSIM)和感知质量度量(例如,FID,NIQE)进行评估,目标是在不牺牲感知质量的情况下达到尽可能低的失真度量。为了实现这一目标,当前方法通常尝试从后验分布中采样,或者优化失真损失(例如,MSE)和感知质量损失(例如,GAN)的加权和。与以往工作不同,本文专门关注在完美感知指数约束下的最小均方误差的最优估计器,即重建图像的分布等于真实图像的分布。最近一个理论结果表明,可以通过优化地将后验均值预测(MMSE估计)运送到真实图像的分布来构造这样的估计器。受这一结果的启发,我们引入了后验均值校正流(PMRF),这是一种简单而高效的算法,近似这种最优估计器。特别是,PMRF 首先预测后验均值,然后使用一个近似所需最优输运映射的校正流模型,将结果转移到一个高质量的图像上。我们研究了PMRF的理论有效性,并证明它在多种图像恢复任务中始终优于以往的方法。