English

DiffLoRA: Generating Personalized Low-Rank Adaptation Weights with Diffusion

Computer Vision and Pattern Recognition 2024-11-18 v3 Artificial Intelligence

Abstract

Personalized text-to-image generation has gained significant attention for its capability to generate high-fidelity portraits of specific identities conditioned on user-defined prompts. Existing methods typically involve test-time fine-tuning or incorporating an additional pre-trained branch. However, these approaches struggle to simultaneously address efficiency, identity fidelity, and the preservation of the model's original generative capabilities. In this paper, we propose DiffLoRA, an efficient method that leverages the diffusion model as a hypernetwork to predict personalized Low-Rank Adaptation (LoRA) weights based on the reference images. By incorporating these LoRA weights into the off-the-shelf text-to-image model, DiffLoRA enables zero-shot personalization during inference, eliminating the need for post-processing optimization. Moreover, we introduce a novel identity-oriented LoRA weights construction pipeline to facilitate the training process of DiffLoRA. The dataset generated through this pipeline enables DiffLoRA to produce consistently high-quality LoRA weights. Notably, the distinctive properties of the diffusion model enhance the generation of superior weights by employing probabilistic modeling to capture intricate structural patterns and thoroughly explore the weight space. Comprehensive experimental results demonstrate that DiffLoRA outperforms existing personalization approaches across multiple benchmarks, achieving both time efficiency and maintaining identity fidelity throughout the personalization process.

Keywords

Cite

@article{arxiv.2408.06740,
  title  = {DiffLoRA: Generating Personalized Low-Rank Adaptation Weights with Diffusion},
  author = {Yujia Wu and Yiming Shi and Jiwei Wei and Chengwei Sun and Yang Yang and Heng Tao Shen},
  journal= {arXiv preprint arXiv:2408.06740},
  year   = {2024}
}

Comments

9 pages,8 figures

R2 v1 2026-06-28T18:11:29.428Z