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

基于身份保持的潜在扩散模型的面部修复

Face inpainting with Identity Preserving Latent Diffusion Models

Institute for Systems and Robotics(系统与机器人研究所)

João Santos, Carlos Santiago, Manuel Marques

AI总结:

本文提出ID-ControlNet,利用潜在扩散模型实现面部修复,通过身份嵌入保持身份一致性,实验表明其在CelebA-HQ等数据集上优于传统方法,接近最先进的身份感知方法。

摘要:

面部修复技术能够以视觉逼真方式恢复缺失或遮挡的面部区域,但保持最终输出的身份仍是一个基本挑战。身份一致性对于下游应用如人脸识别、数字取证和人机交互至关重要,其中细微的身份扭曲可能显著降低性能或信任度。尽管扩散基生成模型在图像修复中取得了显著进展,但它们通常难以忠实保留个体特定的面部特征。另一方面,现有身份感知方法通常依赖于昂贵的微调、辅助监督或对多样遮挡、姿态和面部变化的鲁棒性有限。为了解决这些限制,我们提出ID-ControlNet,一种基于潜在扩散模型的身份保持面部修复框架。基于ControlNet架构,我们的方法将扩散过程条件化为从预训练的人脸识别网络中提取的面部身份嵌入。这种设计使能够重建遮挡的面部区域,同时保持全局面部一致性和身份保真度。此外,我们引入了身份一致性和三元组损失训练策略,以显式地强制生成的面部与目标身份表示之间的对齐。在CelebA-HQ、FFHQ和新的E-Mask数据集上的大量实验表明,ID-ControlNet在身份保持方面显著优于标准扩散基修复方法,实现了与最先进身份感知方法相当的性能。

英文摘要:

Face inpainting techniques recover missing or occluded facial regions in a visually realistic manner, but preserving the identity in the final output remains a fundamental challenge. Identity consistency is crucial for downstream applications such as face recognition, digital forensics, and human-computer interaction, where even subtle identity distortions can significantly degrade performance or trust. Although diffusion-based generative models have recently achieved remarkable progress in image inpainting, they often struggle to faithfully retain individual-specific facial characteristics. On the other hand, existing identity-aware methods typically rely on costly fine-tuning, auxiliary supervision, or exhibit limited robustness to diverse occlusions, poses, and facial variations. To address these limitations, we propose ID-ControlNet, an identity-preserving face inpainting framework built upon latent diffusion models. Based on ControlNet architecture, our approach conditions the diffusion process on facial identity embeddings extracted from a pretrained face recognition network. This design enables reconstruction of occluded facial regions while maintaining global facial coherence and identity fidelity. Furthermore, we introduce an identity consistency and triplet loss training strategy that explicitly enforces alignment between the generated face and the target identity representation. Extensive experiments on CelebA-HQ, FFHQ, and on a new E-Mask dataset demonstrate that ID-ControlNet significantly improves identity preservation over standard diffusion-based inpainting methods, achieving performance comparable to SOTA identity-aware approaches.

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

↑