通过奇异值调节提升剪枝后生成模型的微调效率。
Singular Value Scaling: Efficient Generative Model Compression via Pruned Weights Refinement
- 剪枝后通过调节奇异值差异优化权重初始化。
- 在StyleGAN2/3和DDPM上实现无额外训练成本的压缩提升。
- 适合需高效压缩生成模型的研究者与工程师。
剪枝方法虽能在不增加训练成本的情况下维持模型性能,但通常仅关注关键连接的保留,忽视了剪枝权重对后续微调或蒸馏的影响,导致效率低下。此外,现有生成模型压缩技术多针对GANs(如StyleGAN),专门适配特定架构,对扩散模型(Diffusion models)的压缩研究才刚起步,且多数方法仅适用于单一模型类型。本文提出一种通用的剪枝权重优化方法——奇异值调节(Singular Value Scaling, SVS),适用于GAN与扩散模型。分析发现,剪枝后的权重常具有主导奇异向量,阻碍微调效率并导致性能劣于随机初始化。本方法通过最小化剪枝权重奇异值间的差异,改进初始状态,从而加速微调并引导模型收敛至更优解。在StyleGAN2、StyleGAN3和DDPM上的大量实验表明,SVS可在无额外训练成本下显著提升跨模型类型的压缩表现。代码已开源:https://github.com/LAIT-CVLab/Singular-Value-Scaling。
原文摘要 · Abstract (English)
While pruning methods effectively maintain model performance without extra training costs, they often focus solely on preserving crucial connections, overlooking the impact of pruned weights on subsequent fine-tuning or distillation, leading to inefficiencies. Moreover, most compression techniques for generative models have been developed primarily for GANs, tailored to specific architectures like StyleGAN, and research into compressing Diffusion models has just begun. Even more, these methods are often applicable only to GANs or Diffusion models, highlighting the need for approaches that work across both model types. In this paper, we introduce Singular Value Scaling (SVS), a versatile technique for refining pruned weights, applicable to both model types. Our analysis reveals that pruned weights often exhibit dominant singular vectors, hindering fine-tuning efficiency and leading to suboptimal performance compared to random initialization. Our method enhances weight initialization by minimizing the disparities between singular values of pruned weights, thereby improving the fine-tuning process. This approach not only guides the compressed model toward superior solutions but also significantly speeds up fine-tuning. Extensive experiments on StyleGAN2, StyleGAN3 and DDPM demonstrate that SVS improves compression performance across model types without additional training costs. Our code is available at: https://github.com/LAIT-CVLab/Singular-Value-Scaling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。