TensorX
返回文献探索

Paper · arXiv 2305.18292

Mix-of-Show: Decentralized Low-Rank Adaptation for Multi-Concept Customization of Diffusion Models

Yuchao Gu, Xintao Wang, Jay Zhangjie Wu, Yujun Shi, Yunpeng Chen, Zihan Fan, Wuyou Xiao, Rui Zhao, Shuning Chang, Weijia Wu, Yixiao Ge, Ying Shan, Mike Zheng Shou

6 upvotesMay 29, 2023arXiv 预印本
AI 摘要

Mix-of-Show addresses challenges in decentralized multi-concept customization for text-to-image diffusion models using embedding-decomposed LoRA, gradient fusion, and regionally controllable sampling.

text-to-image diffusion modelsStable Diffusionlow-rank adaptationsLoRAsdecentralized multi-concept customizationsingle-client concept tuningcenter-node concept fusionconcept conflictsidentity lossembedding-decomposed LoRAgradient fusionregionally controllable samplingControlNetT2I-Adaptorattribute bindingmissing object problems

Abstract

Public large-scale text-to-image diffusion models, such as Stable Diffusion, have gained significant attention from the community. These models can be easily customized for new concepts using low-rank adaptations (LoRAs). However, the utilization of multiple concept LoRAs to jointly support multiple customized concepts presents a challenge. We refer to this scenario as decentralized multi-concept customization, which involves single-client concept tuning and center-node concept fusion. In this paper, we propose a new framework called Mix-of-Show that addresses the challenges of decentralized multi-concept customization, including concept conflicts resulting from existing single-client LoRA tuning and identity loss during model fusion. Mix-of-Show adopts an embedding-decomposed LoRA (ED-LoRA) for single-client tuning and gradient fusion for the center node to preserve the in-domain essence of single concepts and support theoretically limitless concept fusion. Additionally, we introduce regionally controllable sampling, which extends spatially controllable sampling (e.g., ControlNet and T2I-Adaptor) to address attribute binding and missing object problems in multi-concept sampling. Extensive experiments demonstrate that Mix-of-Show is capable of composing multiple customized concepts with high fidelity, including characters, objects, and scenes.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号