arXiv:2501.08727cs.LG2025-01ICCV被引 4

通过张量分解增强低秩微调,提升文生图模型效率与性能

Transformed Low-rank Adaptation via Tensor Decomposition and Its Applications to Text-to-image Models

  • 引入可学习变换+残差结构,缩小预训练权重与目标权重的差距
  • 在Stable Diffusion上实现更低参数量下更优生成效果
  • 兼容DoRA等方法,适合资源受限的文生图模型微调场景

文生图模型的参数高效微调(PEFT)日益流行。其中,低秩适配(LoRA)及其变体因高效性备受关注,可在有限计算资源下完成微调。然而,低秩假设与理想微调权重间的近似误差限制了超低参数效率与高性能的兼顾。为此,本文提出一种新方法,融合变换与残差适配:先对预训练权重施加全秩稠密变换以逼近目标权重,降低残差权重的秩;再用紧凑高效的张量分解结构表示残差部分,显著减少近似误差。为实现极致参数效率,设计灵活有效的张量分解方式处理变换与残差。此外,如DoRA等现有方法均可纳入该统一框架。在主体驱动与可控生成任务中对Stable Diffusion进行微调实验,结果表明,本方法在性能和参数效率上均优于LoRA及多个基线。

原文摘要 · Abstract (English)

Parameter-Efficient Fine-Tuning (PEFT) of text-to-image models has become an increasingly popular technique with many applications. Among the various PEFT methods, Low-Rank Adaptation (LoRA) and its variants have gained significant attention due to their effectiveness, enabling users to fine-tune models with limited computational resources. However, the approximation gap between the low-rank assumption and desired fine-tuning weights prevents the simultaneous acquisition of ultra-parameter-efficiency and better performance. To reduce this gap and further improve the power of LoRA, we propose a new PEFT method that combines two classes of adaptations, namely, transform and residual adaptations. In specific, we first apply a full-rank and dense transform to the pre-trained weight. This learnable transform is expected to align the pre-trained weight as closely as possible to the desired weight, thereby reducing the rank of the residual weight. Then, the residual part can be effectively approximated by more compact and parameter-efficient structures, with a smaller approximation error. To achieve ultra-parameter-efficiency in practice, we design highly flexible and effective tensor decompositions for both the transform and residual adaptations. Additionally, popular PEFT methods such as DoRA can be summarized under this transform plus residual adaptation scheme. Experiments are conducted on fine-tuning Stable Diffusion models in subject-driven and controllable generation. The results manifest that our method can achieve better performances and parameter efficiency compared to LoRA and several baselines.

参数高效微调文生图张量分解LoRA

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。