Identity Decoupling for Multi-Subject Personalization of Text-to-Image Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jang, Sangwon, Jo, Jaehyeong, Lee, Kimin, Hwang, Sung Ju
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909367234723840
author Jang, Sangwon
Jo, Jaehyeong
Lee, Kimin
Hwang, Sung Ju
author_facet Jang, Sangwon
Jo, Jaehyeong
Lee, Kimin
Hwang, Sung Ju
contents Text-to-image diffusion models have shown remarkable success in generating personalized subjects based on a few reference images. However, current methods often fail when generating multiple subjects simultaneously, resulting in mixed identities with combined attributes from different subjects. In this work, we present MuDI, a novel framework that enables multi-subject personalization by effectively decoupling identities from multiple subjects. Our main idea is to utilize segmented subjects generated by a foundation model for segmentation (Segment Anything) for both training and inference, as a form of data augmentation for training and initialization for the generation process. Moreover, we further introduce a new metric to better evaluate the performance of our method on multi-subject personalization. Experimental results show that our MuDI can produce high-quality personalized images without identity mixing, even for highly similar subjects as shown in Figure 1. Specifically, in human evaluation, MuDI obtains twice the success rate for personalizing multiple subjects without identity mixing over existing baselines and is preferred over 70% against the strongest baseline.
format Preprint
id arxiv_https___arxiv_org_abs_2404_04243
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Identity Decoupling for Multi-Subject Personalization of Text-to-Image Models
Jang, Sangwon
Jo, Jaehyeong
Lee, Kimin
Hwang, Sung Ju
Computer Vision and Pattern Recognition
Artificial Intelligence
Text-to-image diffusion models have shown remarkable success in generating personalized subjects based on a few reference images. However, current methods often fail when generating multiple subjects simultaneously, resulting in mixed identities with combined attributes from different subjects. In this work, we present MuDI, a novel framework that enables multi-subject personalization by effectively decoupling identities from multiple subjects. Our main idea is to utilize segmented subjects generated by a foundation model for segmentation (Segment Anything) for both training and inference, as a form of data augmentation for training and initialization for the generation process. Moreover, we further introduce a new metric to better evaluate the performance of our method on multi-subject personalization. Experimental results show that our MuDI can produce high-quality personalized images without identity mixing, even for highly similar subjects as shown in Figure 1. Specifically, in human evaluation, MuDI obtains twice the success rate for personalizing multiple subjects without identity mixing over existing baselines and is preferred over 70% against the strongest baseline.
title Identity Decoupling for Multi-Subject Personalization of Text-to-Image Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2404.04243