Modular Customization of Diffusion Models via Blockwise-Parameterized Low-Rank Adaptation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhu, Mingkang, Chen, Xi, Wang, Zhongdao, Yu, Bei, Zhao, Hengshuang, Jia, Jiaya
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910870218473472
author Zhu, Mingkang
Chen, Xi
Wang, Zhongdao
Yu, Bei
Zhao, Hengshuang
Jia, Jiaya
author_facet Zhu, Mingkang
Chen, Xi
Wang, Zhongdao
Yu, Bei
Zhao, Hengshuang
Jia, Jiaya
contents Recent diffusion model customization has shown impressive results in incorporating subject or style concepts with a handful of images. However, the modular composition of multiple concepts into a customized model, aimed to efficiently merge decentralized-trained concepts without influencing their identities, remains unresolved. Modular customization is essential for applications like concept stylization and multi-concept customization using concepts trained by different users. Existing post-training methods are only confined to a fixed set of concepts, and any different combinations require a new round of retraining. In contrast, instant merging methods often cause identity loss and interference of individual merged concepts and are usually limited to a small number of concepts. To address these issues, we propose BlockLoRA, an instant merging method designed to efficiently combine multiple concepts while accurately preserving individual concepts' identity. With a careful analysis of the underlying reason for interference, we develop the Randomized Output Erasure technique to minimize the interference of different customized models. Additionally, Blockwise LoRA Parameterization is proposed to reduce the identity loss during instant model merging. Extensive experiments validate the effectiveness of BlockLoRA, which can instantly merge 15 concepts of people, subjects, scenes, and styles with high fidelity.
format Preprint
id arxiv_https___arxiv_org_abs_2503_08575
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Modular Customization of Diffusion Models via Blockwise-Parameterized Low-Rank Adaptation
Zhu, Mingkang
Chen, Xi
Wang, Zhongdao
Yu, Bei
Zhao, Hengshuang
Jia, Jiaya
Computer Vision and Pattern Recognition
Recent diffusion model customization has shown impressive results in incorporating subject or style concepts with a handful of images. However, the modular composition of multiple concepts into a customized model, aimed to efficiently merge decentralized-trained concepts without influencing their identities, remains unresolved. Modular customization is essential for applications like concept stylization and multi-concept customization using concepts trained by different users. Existing post-training methods are only confined to a fixed set of concepts, and any different combinations require a new round of retraining. In contrast, instant merging methods often cause identity loss and interference of individual merged concepts and are usually limited to a small number of concepts. To address these issues, we propose BlockLoRA, an instant merging method designed to efficiently combine multiple concepts while accurately preserving individual concepts' identity. With a careful analysis of the underlying reason for interference, we develop the Randomized Output Erasure technique to minimize the interference of different customized models. Additionally, Blockwise LoRA Parameterization is proposed to reduce the identity loss during instant model merging. Extensive experiments validate the effectiveness of BlockLoRA, which can instantly merge 15 concepts of people, subjects, scenes, and styles with high fidelity.
title Modular Customization of Diffusion Models via Blockwise-Parameterized Low-Rank Adaptation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.08575