通过重用无效参数,实现高效扩散模型微调。
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 官方产品;中文卡片由大模型生成,请以原文为准。