arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.03374cs.CV

DisenBooth:面向主体驱动文本到图像生成的保身份解耦调优

DisenBooth: Identity-Preserving Disentangled Tuning for Subject-Driven Text-to-Image Generation

Tsinghua University(清华大学);Lanzhou University(兰州大学);Beijing National Research Center for Information Science and Technology(北京信息科学与技术国家研究中心)

Hong Chen, Yipeng Zhang, Simin Wu, Xin Wang, Xuguang Duan, Yuwei Zhou, Wenwu Zhu

AI总结:

DisenBooth 提出保身份解耦调优框架,通过解耦身份相关与无关嵌入并设计弱去噪和对比嵌入辅助目标,解决主体驱动文本到图像生成中的身份丢失与文本忽略问题,提升生成灵活性和可控性。

摘要:

主体驱动的文本到图像生成旨在根据文本描述生成给定主体的定制图像,近年来受到越来越多的关注。现有方法主要依赖对预训练生成模型进行微调,其中身份相关信息(例如男孩)和身份无关信息(例如男孩的背景或姿态)在潜在嵌入空间中相互纠缠。然而,高度纠缠的潜在嵌入可能导致主体驱动文本到图像生成失败,具体表现如下:(i)隐藏在纠缠嵌入中的身份无关信息可能主导生成过程,导致生成的图像严重依赖无关信息而忽略给定的文本描述;(ii)纠缠嵌入中携带的身份相关信息无法被适当保留,导致生成图像中主体身份发生变化。为解决这些问题,我们提出了 DisenBooth,一种面向主体驱动文本到图像生成的保身份解耦调优框架。具体而言,DisenBooth 在去噪过程中对预训练扩散模型进行微调。与以往利用纠缠嵌入对每张图像进行去噪的工作不同,DisenBooth 利用解耦嵌入分别保留主体身份并捕获身份无关信息。我们进一步设计了新颖的弱去噪和对比嵌入辅助调优目标来实现解耦。大量实验表明,我们提出的 DisenBooth 框架在主体驱动文本到图像生成中凭借保身份嵌入优于基线模型。此外,通过结合保身份嵌入和身份无关嵌入,DisenBooth 展示了更强的生成灵活性和可控性。

英文摘要:

Subject-driven text-to-image generation aims to generate customized images of the given subject based on the text descriptions, which has drawn increasing attention. Existing methods mainly resort to finetuning a pretrained generative model, where the identity-relevant information (e.g., the boy) and the identity-irrelevant information (e.g., the background or the pose of the boy) are entangled in the latent embedding space. However, the highly entangled latent embedding may lead to the failure of subject-driven text-to-image generation as follows: (i) the identity-irrelevant information hidden in the entangled embedding may dominate the generation process, resulting in the generated images heavily dependent on the irrelevant information while ignoring the given text descriptions; (ii) the identity-relevant information carried in the entangled embedding can not be appropriately preserved, resulting in identity change of the subject in the generated images. To tackle the problems, we propose DisenBooth, an identity-preserving disentangled tuning framework for subject-driven text-to-image generation. Specifically, DisenBooth finetunes the pretrained diffusion model in the denoising process. Different from previous works that utilize an entangled embedding to denoise each image, DisenBooth instead utilizes disentangled embeddings to respectively preserve the subject identity and capture the identity-irrelevant information. We further design the novel weak denoising and contrastive embedding auxiliary tuning objectives to achieve the disentanglement. Extensive experiments show that our proposed DisenBooth framework outperforms baseline models for subject-driven text-to-image generation with the identity-preserved embedding. Additionally, by combining the identity-preserved embedding and identity-irrelevant embedding, DisenBooth demonstrates more generation flexibility and controllability

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

↑