arXiv:2502.01403cs.CVcs.AI2025-02被引 17

AdaSVD通过自适应压缩比和误差补偿,显著降低大模型内存占用且保持性能。

AdaSVD: Adaptive Singular Value Decomposition for Large Language Models

  • 根据层重要性动态分配压缩比,避免统一压缩导致的性能损失。
  • 通过交替更新奇异矩阵,自适应补偿截断带来的误差。
  • 在多个大模型上验证,内存减少同时性能优于现有SVD方法。

大规模语言模型(LLM)在自然语言处理任务中取得显著成果,但其庞大的内存需求给资源受限设备的部署带来挑战。奇异值分解(SVD)作为一种有前景的模型压缩技术,可显著降低内存开销。然而,现有基于SVD的方法难以有效缓解截断引入的误差,导致与原始模型相比性能明显下降。此外,对所有Transformer层采用统一压缩比,未能考虑各层重要性的差异。为此,我们提出AdaSVD,一种自适应SVD压缩方法。具体而言,AdaSVD引入adaComp,通过交替更新奇异矩阵$ \mathcal{U}$和$ \mathcal{V}^\top$,自适应补偿截断误差;同时引入adaCR,依据各层相对重要性动态分配层级压缩比。在多个LLM/VLM家族及评估指标上的大量实验表明,AdaSVD持续优于当前最先进(SOTA)的SVD方法,在显著降低内存需求的同时实现更优性能。代码与模型将发布于https://github.com/ZHITENGLI/AdaSVD。

原文摘要 · Abstract (English)

Large language models (LLMs) have achieved remarkable success in natural language processing (NLP) tasks, yet their substantial memory requirements present significant challenges for deployment on resource-constrained devices. Singular Value Decomposition (SVD) has emerged as a promising compression technique for LLMs, offering considerable reductions in memory overhead. However, existing SVD-based methods often struggle to effectively mitigate the errors introduced by SVD truncation, leading to a noticeable performance gap when compared to the original models. Furthermore, applying a uniform compression ratio across all transformer layers fails to account for the varying importance of different layers. To address these challenges, we propose AdaSVD, an adaptive SVD-based LLM compression approach. Specifically, AdaSVD introduces adaComp, which adaptively compensates for SVD truncation errors by alternately updating the singular matrices $\mathcal{U}$ and $\mathcal{V}^\top$. Additionally, AdaSVD introduces adaCR, which adaptively assigns layer-specific compression ratios based on the relative importance of each layer. Extensive experiments across multiple LLM/VLM families and evaluation metrics demonstrate that AdaSVD consistently outperforms state-of-the-art (SOTA) SVD-based methods, achieving superior performance with significantly reduced memory requirements. Code and models of AdaSVD will be available at https://github.com/ZHITENGLI/AdaSVD.

模型压缩SVD大模型优化

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