arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.03556cs.CVcs.LG

基于并行视觉注意力的扩散模型个性化人脸修复

Personalized Face Inpainting with Diffusion Models by Parallel Visual Attention

Carnegie Mellon University(卡内基梅隆大学);INSAIT, Sofia University(索非亚大学INSAIT学院)

Jianjin Xu, Saman Motamed, Praneetha Vaddamanu, Chen Henry Wu, Christian Haene, Jean-Charles Bazin, Fernando de la Torre

AI总结:

本文提出 PVA 并行视觉注意力机制并结合扩散模型,利用参考身份编码器特征实现个性化人脸修复,在身份保持、语言可控性和微调效率上显著优于现有方法。

摘要:

人脸修复在照片复原、图像编辑和虚拟现实等多种应用中具有重要作用。尽管人脸生成模型已取得显著进展,但在修复过程中确保人物独特的面部身份得以保持仍是一个难以实现的目标。以 MyStyle 为代表的当前最先进技术需要耗费大量资源进行微调,并且每个新身份都需要大量图像。此外,现有方法往往难以适配用户指定的语义属性,例如胡须或表情。为改善修复结果并降低推理期间的计算复杂度,本文提出将 Parallel Visual Attention(PVA,并行视觉注意力)与扩散模型结合使用。具体而言,我们在去噪网络的每个交叉注意力模块中插入并行注意力矩阵,这些矩阵关注由身份编码器从参考图像中提取的特征。我们在 CelebAHQ-IDI 上训练新增的注意力模块和身份编码器,该数据集是为身份保持的人脸修复而提出的。实验表明,与包括 MyStyle、Paint by Example 和 Custom Diffusion 在内的多种基准相比,PVA 在人脸修复以及语言引导的人脸修复任务中均取得了无与伦比的身份相似度。研究结果表明,PVA 可确保良好的身份保持,同时提供有效的语言可控性。此外,与 Custom Diffusion 相比,PVA 对每个新身份仅需 40 个微调步骤,这意味着速度显著提升超过 20 倍。

英文摘要:

Face inpainting is important in various applications, such as photo restoration, image editing, and virtual reality. Despite the significant advances in face generative models, ensuring that a person's unique facial identity is maintained during the inpainting process is still an elusive goal. Current state-of-the-art techniques, exemplified by MyStyle, necessitate resource-intensive fine-tuning and a substantial number of images for each new identity. Furthermore, existing methods often fall short in accommodating user-specified semantic attributes, such as beard or expression. To improve inpainting results, and reduce the computational complexity during inference, this paper proposes the use of Parallel Visual Attention (PVA) in conjunction with diffusion models. Specifically, we insert parallel attention matrices to each cross-attention module in the denoising network, which attends to features extracted from reference images by an identity encoder. We train the added attention modules and identity encoder on CelebAHQ-IDI, a dataset proposed for identity-preserving face inpainting. Experiments demonstrate that PVA attains unparalleled identity resemblance in both face inpainting and face inpainting with language guidance tasks, in comparison to various benchmarks, including MyStyle, Paint by Example, and Custom Diffusion. Our findings reveal that PVA ensures good identity preservation while offering effective language-controllability. Additionally, in contrast to Custom Diffusion, PVA requires just 40 fine-tuning steps for each new identity, which translates to a significant speed increase of over 20 times.

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

↑