UniPortrait: A Unified Framework for Identity-Preserving Single- and Multi-Human Image Personalization

Junjie He,Yifeng Geng,Liefeng Bo
2024-09-06
Abstract:This paper presents UniPortrait, an innovative human image personalization framework that unifies single- and multi-ID customization with high face fidelity, extensive facial editability, free-form input description, and diverse layout generation. UniPortrait consists of only two plug-and-play modules: an ID embedding module and an ID routing module. The ID embedding module extracts versatile editable facial features with a decoupling strategy for each ID and embeds them into the context space of diffusion models. The ID routing module then combines and distributes these embeddings adaptively to their respective regions within the synthesized image, achieving the customization of single and multiple IDs. With a carefully designed two-stage training scheme, UniPortrait achieves superior performance in both single- and multi-ID customization. Quantitative and qualitative experiments demonstrate the advantages of our method over existing approaches as well as its good scalability, e.g., the universal compatibility with existing generative control tools. The project page is at <a class="link-external link-https" href="https://aigcdesigngroup.github.io/UniPortrait-Page/" rel="external noopener nofollow">this https URL</a> .
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve This paper proposes a unified framework named **UniPortrait**, aiming to address key issues in the personalization of single and multiple person images. Specifically: 1. **High-fidelity identity preservation**: Maintaining highly realistic facial features while allowing for image editing during image generation. 2. **Free-form text description**: Supporting free-form text input without the need for predefined layouts or specific formats. 3. **Personalization of multi-identity images**: Avoiding the problem of identity blending when handling images with multiple people, and maintaining the uniqueness of each identity. ### Main Contributions 1. Proposes the **UniPortrait** framework, which can uniformly handle the personalization of single and multiple person images with high fidelity and controllability. 2. Introduces the **ID embedding module**, which adopts a decoupling strategy to embed detailed facial identity information while maintaining good editability. 3. Designs the **ID routing mechanism**, which addresses the identity blending problem in the personalization of multi-identity images, without compromising the integrity of each identity, the diversity of generated images, and the flexibility of prompt design. Through the above methods, the paper aims to improve the effectiveness of image personalization tasks, especially in scenarios involving multiple identities.