为扩散模型设计分时低秩适配,提升生成多样性与效率。
TimeStep Master: Asymmetrical Mixture of Timestep LoRA Experts for Versatile and Efficient Diffusion Models in Vision
- 按时间步分段使用不同LoRA专家,捕捉多噪声水平特征。
- 细粒度核心+粗粒度上下文协同,实现自适应融合,性能领先。
- 适用于图像、视频生成,适配多种模型结构,通用性强。
扩散模型近年来推动了视觉生成的发展,但下游任务中因微调成本高难以应用。近期低秩适配(LoRA)被用于高效微调扩散模型,但现有方法对不同时间步使用相同LoRA,限制了性能。为此,本文提出一种通用简洁的分时主控(TimeStep Master, TSM)范式,包含两阶段:第一阶段(培育阶段)在不同时间步区间应用不同LoRA,生成可捕获各异噪声水平的时序LoRA专家;第二阶段(组装阶段)设计新颖的非对称混合机制,通过多尺度区间专家的核心-上下文协作,在每个时间步以最小区间专家为核心(无门控),其他大区间专家作为上下文专家(带时变门控)。此设计使模型能精确建模噪声水平,并自适应融合多尺度上下文信息,显著提升生成灵活性。我们在三个典型任务(领域适配、后预训练、模型蒸馏)上验证有效性,覆盖UNet、DiT、MM-DiT等多种架构及图像、视频多模态数据,均取得当前最优性能,展现强大泛化能力。
原文摘要 · Abstract (English)
Diffusion models have driven the advancement of vision generation over the past years. However, it is often difficult to apply these large models in downstream tasks, due to massive fine-tuning cost. Recently, Low-Rank Adaptation (LoRA) has been applied for efficient tuning of diffusion models. Unfortunately, the capabilities of LoRA-tuned diffusion models are limited, since the same LoRA is used for different timesteps of the diffusion process. To tackle this problem, we introduce a general and concise TimeStep Master (TSM) paradigm with two key fine-tuning stages. In the fostering stage (1-stage), we apply different LoRAs to fine-tune the diffusion model at different timestep intervals. This results in different TimeStep LoRA experts that can effectively capture different noise levels. In the assembling stage (2-stage), we design a novel asymmetrical mixture of TimeStep LoRA experts, via core-context collaboration of experts at multi-scale intervals. For each timestep, we leverage TimeStep LoRA expert within the smallest interval as the core expert without gating, and use experts within the bigger intervals as the context experts with time-dependent gating. Consequently, our TSM can effectively model the noise level via the expert in the finest interval, and adaptively integrate contexts from the experts of other scales, boosting the versatility of diffusion models. To show the effectiveness of our TSM paradigm, we conduct extensive experiments on three typical and popular LoRA-related tasks of diffusion models, including domain adaptation, post-pretraining, and model distillation. Our TSM achieves the state-of-the-art results on all these tasks, throughout various model structures (UNet, DiT and MM-DiT) and visual data modalities (Image, Video), showing its remarkable generalization capacity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。