arXiv:2409.06633cs.CV2024-09中稿 · ICLR被引 3

通过重用无效参数,实现高效扩散模型微调。

SaRA: High-Efficient Diffusion Model Fine-tuning with Progressive Sparse Low-Rank Adaptation

  • 发现前10%-20%最小参数对生成无贡献,提出可重用策略。
  • 结合低秩稀疏训练,使微调后模型泛化能力优于LoRA。
  • 仅需一行代码修改,兼容现有方法,适合快速部署。

近年来,扩散模型在图像与视频生成任务中取得显著进展,预训练模型如Stable Diffusion系列发挥关键作用。受模型剪枝启发,我们提出一种新型微调方法,旨在重新利用预训练扩散模型中无效参数,赋予其新任务能力。实验发现,按绝对值排序的最小10%至20%参数对生成过程无贡献。基于此,我们提出SaRA方法,将这些临时无效参数重新组织为稀疏权重矩阵,以学习特定任务知识。为缓解过拟合,设计基于核范数的低秩稀疏训练方案。同时引入渐进式参数调整策略,充分释放重训练参数潜力。最后提出非结构化反向传播机制,显著降低微调内存开销。在SD模型上的微调实验验证了该方法的有效性,生成性能优于传统方法,且保持更强泛化能力。此外,仅需单行代码修改即可实现,与现有方法无缝兼容。

原文摘要 · Abstract (English)

In recent years, the development of diffusion models has led to significant progress in image and video generation tasks, with pre-trained models like the Stable Diffusion series playing a crucial role. Inspired by model pruning which lightens large pre-trained models by removing unimportant parameters, we propose a novel model fine-tuning method to make full use of these ineffective parameters and enable the pre-trained model with new task-specified capabilities. In this work, we first investigate the importance of parameters in pre-trained diffusion models, and discover that the smallest 10% to 20% of parameters by absolute values do not contribute to the generation process. Based on this observation, we propose a method termed SaRA that re-utilizes these temporarily ineffective parameters, equating to optimizing a sparse weight matrix to learn the task-specific knowledge. To mitigate overfitting, we propose a nuclear-norm-based low-rank sparse training scheme for efficient fine-tuning. Furthermore, we design a new progressive parameter adjustment strategy to make full use of the re-trained/finetuned parameters. Finally, we propose a novel unstructural backpropagation strategy, which significantly reduces memory costs during fine-tuning. Our method enhances the generative capabilities of pre-trained models in downstream applications and outperforms traditional fine-tuning methods like LoRA in maintaining model's generalization ability. We validate our approach through fine-tuning experiments on SD models, demonstrating significant improvements. SaRA also offers a practical advantage that requires only a single line of code modification for efficient implementation and is seamlessly compatible with existing methods.

扩散模型微调优化低秩适应高效训练

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