Research map / Image generation / Editing and style transfer
Omni-ID: Holistic Identity Representation Designed for Generative Tasks
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Introduces Omni-ID, a fixed-size transformer face-identity representation trained with a few-to-many generative reconstruction objective and two decoders (masked transformer and flow-matching) on the new MFHQ video-face dataset.
Problem
Existing face identity representations (ArcFace, CLIP) are learned discriminatively from single images and lose nuanced, pose- and expression-spanning identity detail needed for generative face synthesis and personalization.
Approach
Learn a holistic identity encoder that consolidates a variable number of unstructured face images into a fixed set of learnable query tokens, trained generatively by reconstructing many unseen target views of the same person from a few inputs, with a multi-decoder objective combining masked reconstruction and conditional flow matching.
Key results
- On the MFHQ test set with 7 inputs, Omni-ID reaches ID similarity 0.742 versus 0.682 for CLIP and 0.535 for ArcFace in controllable face generation.
- Ablation removing MTD pretraining drops ID similarity from 0.708 to 0.468 (1 input), and removing few-to-many pretraining drops it to 0.616.
- Omni-ID's ID similarity improves monotonically with more input images (0.708 at 1 input to 0.742 at 7 inputs on MFHQ).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Use Omni-ID instead of the strongest baseline, CLIP | IP-Adapter + ControlNet · MFHQ Test | ID Similarity: CLIP: 0.648 / 0.670 / 0.680 / 0.682; Omni-ID: 0.708 / 0.728 / 0.737 / 0.742 | Helped |
| Use Omni-ID instead of CLIP | IP-Adapter + ControlNet · MFHQ Test | Pose Error: 2.3/2.3/2.2/2.2 -> 2.1/2.0/2.0/2.0 | Helped |
| Use Omni-ID instead of CLIP | IP-Adapter + ControlNet · Webface Test | ID Similarity: 0.695 / 0.696 / 0.696 / 0.695 / 0.695 -> 0.774 / 0.779 / 0.781 / 0.784 / 0.785 | Helped |
| Use Omni-ID instead of CLIP | IP-Adapter + ControlNet · Webface Test | Pose Error: 2.9/2.8/2.8/2.7/2.8 -> 2.7/2.6/2.6/2.6/2.6 | Helped |
| add MTD pretraining (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.336 / 0.358 -> 0.683 / 0.733 | Helped |
| Use few-to-many MTD pretraining with 3 input images and 8 target images instead of 3 input and 1 target (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.491 / 0.515 -> 0.683 / 0.733 | Helped |
| Use 3 input images and 8 target images instead of 3 input and 5 target images during MTD pretraining (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.582 / 0.615 -> 0.683 / 0.733 | Helped |
| Use 3 input images and 8 target images instead of 8 input and 8 target images during MTD pretraining (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.661 / 0.696 -> 0.683 / 0.733 | Helped |
| Increase MTD masking ratio from 0.95 to 0.99 (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.683 / 0.733 -> 0.670 / 0.700 | Hurt |
| Decrease MTD masking ratio from 0.95 to 0.85 (ablation) | Omni-ID · Flow-Matching face generation | ID Similarity: 0.683 / 0.733 -> 0.609 / 0.650 | Hurt |
| add MTD pretraining (ablation) | Omni-ID · Controllable face generation, MFHQ Test | ID Similarity: 0.468 / 0.473 -> 0.708 / 0.728 | Helped |
| add Flow-Matching Decoder pretraining (ablation) | Omni-ID · Controllable face generation, MFHQ Test | ID Similarity: 0.672 / 0.685 -> 0.708 / 0.728 | Helped |
Surprising findings
- Flow-matching decoder alone is less effective for representation learning; the masked autoencoding objective is needed to learn a useful encoder.
- Using a 99% masking ratio slightly reduces performance, indicating an optimal mask ratio around 95%.
Limitations
- Omni-ID does not represent non-facial attributes such as hair, which can be hallucinated downstream.
- Training data has limited lighting diversity and skin-tone coverage, causing lighting injection and skin tone bias issues.
- Gains from additional input images are marginal beyond a few inputs; models trained with at most 3 inputs.
Experimental setup
Datasets: MFHQ WebFace21M WebFace42M CelebV-HQ VFHQ TalkingHead-1KH CelebV-Text internal licensed image dataset
Benchmarks: MFHQ test set WebFace test set (Webface21M-derived) personalized T2I test (10 identities, 20 prompts)
Metrics: ID similarity pose error
Baselines: ArcFace CLIP ArcFace + CLIP (FaceIDPlus-style) IP-Adapter FaRL ControlNet
Builds on: IP-Adapter CLIP ArcFace FLUX ControlNet
Closest papers by meaning
- FaceA-Net: Facial Attribute-Driven ID Preserving Image Generation Network (AAAI 2025 · similarity 0.88)
- Generative Unlearning for Any Identity (CVPR 2024 · similarity 0.86)
- FaithfulFaces: Pose-Faithful Facial Identity Preservation for Text-to-Video Generation (NeurIPS 2026 · similarity 0.85)
- Vec2Face: Scaling Face Dataset Generation with Loosely Constrained Vectors (ICLR 2025 · similarity 0.85)
- GIF: Generative Inspiration for Face Recognition at Scale (CVPR 2025 · similarity 0.85)
- UIFace: Unleashing Inherent Model Capabilities to Enhance Intra-Class Diversity in Synthetic Face Recognition (ICLR 2025 · similarity 0.85)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
