论文标题

差异:盲人恢复,散布错误收缩

DifFace: Blind Face Restoration with Diffused Error Contraction

论文作者

Yue, Zongsheng, Loy, Chen Change

论文摘要

尽管基于深度学习的盲人恢复方法取得了前所未有的成功,但它们仍然受到两个主要局限性。首先,在训练数据中面对复杂的降解时,大多数人都会恶化。其次,这些方法需要多种约束,例如忠诚,感知和对抗性损失,这些损失需要费力的超参数调整以稳定和平衡其影响。在这项工作中,我们提出了一种名为Difface的新颖方法,该方法能够在没有复杂损失设计的情况下更优雅地应对看不见和复杂的降解。我们方法的关键是建立从观察到的低质量(LQ)图像到其高质量(HQ)对应物的后验分布。特别是,我们设计了从LQ图像到预训练扩散模型的中间状态的过渡分布,然后通过递归应用预训练的扩散模型,逐渐从该中间状态逐渐从该中间状态传播到HQ目标。过渡分布仅依赖于在某些合成数据中经过$ L_2 $损失训练的恢复主链,这有利地避免了现有方法中繁琐的训练过程。此外,过渡分布可以收缩恢复主链的误差,从而使我们的方法对未知降解更强大。全面的实验表明,Difface优于当前的最新方法,尤其是在严重降解的情况下。代码和模型可在https://github.com/zsyoaoa/difface上找到。

While deep learning-based methods for blind face restoration have achieved unprecedented success, they still suffer from two major limitations. First, most of them deteriorate when facing complex degradations out of their training data. Second, these methods require multiple constraints, e.g., fidelity, perceptual, and adversarial losses, which require laborious hyper-parameter tuning to stabilize and balance their influences. In this work, we propose a novel method named DifFace that is capable of coping with unseen and complex degradations more gracefully without complicated loss designs. The key of our method is to establish a posterior distribution from the observed low-quality (LQ) image to its high-quality (HQ) counterpart. In particular, we design a transition distribution from the LQ image to the intermediate state of a pre-trained diffusion model and then gradually transmit from this intermediate state to the HQ target by recursively applying a pre-trained diffusion model. The transition distribution only relies on a restoration backbone that is trained with $L_2$ loss on some synthetic data, which favorably avoids the cumbersome training process in existing methods. Moreover, the transition distribution can contract the error of the restoration backbone and thus makes our method more robust to unknown degradations. Comprehensive experiments show that DifFace is superior to current state-of-the-art methods, especially in cases with severe degradations. Code and model are available at https://github.com/zsyOAOA/DifFace.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源