arXiv:2502.15092cs.CL2025-02被引 4

通过可学习的奇异谱优化,实现大模型高效压缩与性能保持。

Optimizing Singular Spectrum for Large Language Model Compression

  • 用可学习对角矩阵重标定奇异值重要性,替代固定排序。
  • 三阶段训练实现从粗粒度压缩到细粒度稀疏化,提升压缩效率。
  • 适配多模型与任务,显著优于现有压缩方法。

大语言模型(LLMs)表现出强大能力,但参数复杂度高导致部署困难。现有基于奇异值分解(SVD)的压缩方法简单地将奇异值作为组件重要性指标,但其排序未必与下游任务性能相关。本文提出SoCo(Singular spectrum optimization for LLM Compression),一种数据驱动的新型压缩框架,通过可学习的对角矩阵为分解后的组件分配重要性分数,并设计三阶段训练流程,逐步优化分数,实现从粗粒度压缩到细粒度稀疏化的演进,有效平衡压缩强度与性能保留。得益于可学习的奇异谱,SoCo能自适应地根据稀疏重要性分数剪枝组件,而非依赖奇异值的固定顺序;更重要的是,被增强的重要组件可补偿被剪枝部分的损失。在多个大语言模型和基准上的实验表明,SoCo在模型压缩方面超越了现有最先进方法。

原文摘要 · Abstract (English)

Large language models (LLMs) have demonstrated remarkable capabilities, yet prohibitive parameter complexity often hinders their deployment. Existing singular value decomposition (SVD) based compression methods simply deem singular values as importance scores of decomposed components. However, this importance ordered by singular values does not necessarily correlate with the performance of a downstream task. In this work, we introduce SoCo (Singular spectrum optimization for large language model Compression), a novel compression framework that learns to rescale the decomposed components of SVD in a data-driven manner. Concretely, we employ a learnable diagonal matrix to assign importance scores for singular spectrum and develop a three-stage training process that progressively refines these scores from initial coarse compression to fine-grained sparsification-thereby striking an effective balance between aggressive model compression and performance preservation. Thanks to the learnable singular spectrum, SoCo adaptively prunes components according to the sparsified importance scores, rather than relying on the fixed order of singular values. More importantly, the remaining components with amplified importance scores can compensate for the loss of the pruned ones. Experimental evaluations across multiple LLMs and benchmarks demonstrate that SoCo surpasses the state-of-the-art methods in model compression.

模型压缩奇异值分解大模型优化自适应剪枝

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