English

A Closer Look at Parameter-Efficient Tuning in Diffusion Models

Computer Vision and Pattern Recognition 2023-04-13 v2 Machine Learning

Abstract

Large-scale diffusion models like Stable Diffusion are powerful and find various real-world applications while customizing such models by fine-tuning is both memory and time inefficient. Motivated by the recent progress in natural language processing, we investigate parameter-efficient tuning in large diffusion models by inserting small learnable modules (termed adapters). In particular, we decompose the design space of adapters into orthogonal factors -- the input position, the output position as well as the function form, and perform Analysis of Variance (ANOVA), a classical statistical approach for analyzing the correlation between discrete (design options) and continuous variables (evaluation metrics). Our analysis suggests that the input position of adapters is the critical factor influencing the performance of downstream tasks. Then, we carefully study the choice of the input position, and we find that putting the input position after the cross-attention block can lead to the best performance, validated by additional visualization analyses. Finally, we provide a recipe for parameter-efficient tuning in diffusion models, which is comparable if not superior to the fully fine-tuned baseline (e.g., DreamBooth) with only 0.75 \% extra parameters, across various customized tasks.

Keywords

Cite

@article{arxiv.2303.18181,
  title  = {A Closer Look at Parameter-Efficient Tuning in Diffusion Models},
  author = {Chendong Xiang and Fan Bao and Chongxuan Li and Hang Su and Jun Zhu},
  journal= {arXiv preprint arXiv:2303.18181},
  year   = {2023}
}

Comments

8pages, now our code is available at: https://github.com/Xiang-cd/unet-finetune

R2 v1 2026-06-28T09:43:31.225Z