一次训练即可任意压缩大模型,无需重新计算
Choose Your Model Size: Any Compression of Large Language Models Without Re-Computation
- 通过奇异值分解与迭代剪枝,实现参数高效压缩
- 在多个大模型上验证,压缩后性能优于现有方法
- 支持任意目标大小压缩,可与量化技术无缝结合
由于大模型体积庞大且推理成本高,其在资源受限环境中的应用仍具挑战。本文提出一种名为任意压缩的迭代剪枝(ACIP)的新方法,仅需一次随机梯度下降训练,即可确定压缩与性能的权衡。通过线性层的SVD重参数化,并施加稀疏诱导惩罚,迭代剪枝奇异值。关键在于,参数剪枝顺序生成全局评分图,使模型可无重计算地压缩至任意目标尺寸。我们在多个开源大语言模型及下游任务上评估了ACIP,结果优于现有基于因子分解的压缩方法。此外,证明其可无缝集成常见量化压缩技术。
原文摘要 · Abstract (English)
The adoption of Foundation Models in resource-constrained environments remains challenging due to their large size and inference costs. A promising way to overcome these limitations is post-training compression, which aims to balance reduced model size against performance degradation. This work presents Any Compression via Iterative Pruning (ACIP), a novel algorithmic approach to determine a compression-performance trade-off from a single stochastic gradient descent run. To achieve parameter efficiency, we use an SVD-reparametrization of linear layers and iteratively prune their singular values with a sparsity-inducing penalty. Importantly, the pruning order of the parameters is used to derive a global score map that allows compressing a model to any target size without re-computation. We evaluate ACIP on a large selection of open-weight LLMs and downstream tasks, demonstrating state-of-the-art results compared to existing factorization-based compression methods. We also show that ACIP seamlessly complements common quantization-based compression techniques.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。