arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.05043cs.CVcs.LGeess.IV

ReF-LDM:一种用于基于参考的人脸图像修复的潜在扩散模型

ReF-LDM: A Latent Diffusion Model for Reference-based Face Image Restoration

MediaTek(联发科);National Yang Ming Chiao Tung University(国立阳明交通大学)

Chi-Wei Hsiao, Yu-Lun Liu, Cheng-Kun Yang, Sheng-Po Kuo, Kevin Jou, Chia-Ping Chen

AI总结:

针对盲人脸修复生成内容与真实外观不符的问题,提出ReF-LDM模型,通过CacheKV机制利用参考图像并结合时间步长缩放身份损失,还构建FFHQ-Ref数据集,提升修复准确性。

摘要:

尽管最近关于盲人脸图像修复的研究已成功从低质量(LQ)输入图像生成细节丰富的高质量(HQ)图像,但生成的内容可能无法准确反映人的真实外观。为解决这一问题,引入拍摄良好的个人图像作为额外参考输入可能是一种有前景的策略。受潜在扩散模型(LDM)近期成功的启发,我们提出ReF-LDM,这是一种LDM的适配模型,旨在基于一张LQ图像和多张HQ参考图像生成HQ人脸图像。我们的模型集成了一种高效机制CacheKV,以在生成过程中利用参考图像。此外,我们设计了时间步长缩放的身份损失,使基于LDM的模型能够专注于学习人脸的判别特征。最后,我们构建了FFHQ-Ref数据集,该数据集包含20,405张高质量人脸图像及其对应的参考图像,可作为基于参考的人脸修复模型的训练和评估数据。

英文摘要:

While recent works on blind face image restoration have successfully produced impressive high-quality (HQ) images with abundant details from low-quality (LQ) input images, the generated content may not accurately reflect the real appearance of a person. To address this problem, incorporating well-shot personal images as additional reference inputs could be a promising strategy. Inspired by the recent success of the Latent Diffusion Model (LDM), we propose ReF-LDM, an adaptation of LDM designed to generate HQ face images conditioned on one LQ image and multiple HQ reference images. Our model integrates an effective and efficient mechanism, CacheKV, to leverage the reference images during the generation process. Additionally, we design a timestep-scaled identity loss, enabling our LDM-based model to focus on learning the discriminating features of human faces. Lastly, we construct FFHQ-Ref, a dataset consisting of 20,405 high-quality (HQ) face images with corresponding reference images, which can serve as both training and evaluation data for reference-based face restoration models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑