arXiv:2602.02848cs.LG2026-02被引 5

提出零和SVD方法,自动分配不同层的压缩秩,提升大模型压缩效率。

Zero Sum SVD: Balancing Loss Sensitivity for Low Rank LLM Compression

  • 基于激活白化与损失估计,全局选择奇异分量,实现零和剪枝
  • 在多个模型上保持性能,压缩率10倍时仍优于传统方法
  • 无需迭代优化,适合部署前快速压缩大型语言模型

大语言模型虽性能优异,但存储与计算开销限制其部署。基于SVD的压缩可通过低秩因子降低存储并加速推理,但性能依赖于整体压缩率下的秩分配策略。现有方法常对相似尺寸矩阵采用均一秩,忽视其损失敏感度差异,或依赖昂贵的预截断优化确定各层秩。本文提出后训练方法零和SVD(ZS-SVD),通过激活白化与白化坐标下的损失梯度估计,实现全局奇异分量选择。该方法遵循‘零和’规则,使累计预测损失变化趋近于零,自动获得非均匀秩分配,无需求解秩分配优化问题。受预训练解附近梯度呈低秩结构的启发,还引入可选轻量级修正:截断后进行一次投影梯度更新,再重新截断。在多种大模型架构上的实验表明,该方法在不同基准与压缩比下均表现一致提升。代码已开源。

原文摘要 · Abstract (English)

Advances in large language models have driven strong performance across many tasks, but their memory and compute costs still hinder deployment. SVD-based compression reduces storage and can speed up inference via low-rank factors, yet performance depends on how rank is allocated under a global compression ratio. Prior methods often use homogeneous ranks for similarly sized matrices, despite large differences in loss sensitivity, or rely on expensive iterative pre-truncation optimization to determine per matrix ranks. We propose \textbf{Zero Sum SVD} (\textbf{ZS-SVD}), a post-training method that performs \emph{global} singular component selection using activation whitening and first-order calibration loss estimates in whitened coordinates. \textbf{ZS-SVD} prunes components across the whole model with a \textbf{zero sum} rule that keeps the cumulative predicted loss change near zero, automatically yielding heterogeneous ranks without solving a rank allocation optimization. Motivated by evidence that gradients near pretrained solutions exhibit low rank structure, we also introduce an optional lightweight correction that applies a \textbf{single} projected gradient update after truncation, followed by re-truncation. Extensive experiments across multiple LLM architectures show consistent gains across diverse benchmarks and compression ratios. Code is available at https://github.com/mint-vu/Zero-Sum-SVD

大模型压缩SVD低秩高效推理

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