arXiv:2510.19389cs.LG2025-10被引 2

提出自适应秩分配方法,提升大模型压缩效率与精度。

ARA: Adaptive Rank Allocation for Efficient Large Language Model SVD Compression

  • 设计专用掩码实现秩与参数的高效映射更新
  • 引入额外损失函数,避免陷入局部最优
  • 在80%压缩率下显著降低困惑度并提升零样本准确率

在大语言模型压缩领域,奇异值分解(SVD)是一种广泛应用的低秩分解技术。由于SVD仅作用于线性模块,而这些模块被非线性组件分隔,因此只能独立应用于每个线性模块。在全局压缩比约束下,如何为不同线性模块确定合适的秩成为关键问题。现有方法如启发式算法和基于掩码的训练虽有进展,但仍存在局限:启发式算法探索解空间范围受限,基于掩码的训练难以有效捕捉奇异值谱与可训练参数间的关系。更重要的是,现有方法忽略了增益函数在压缩比为1时非光滑的特性,常导致训练过程陷入次优局部极小值。为此,本文提出自适应秩分配(ARA)方法:(1)设计专用掩码,实现保留秩与可训练参数间的高效映射与更新;(2)引入额外损失函数,引导参数选择趋向全局最优解。实验结果表明,ARA达到当前最优性能。在LLaMA2-7B模型上,80%压缩率下,WikiText2上的困惑度从8.38降至6.42,零样本任务平均准确率提升9.72个百分点,优于均匀压缩方案。

原文摘要 · Abstract (English)

In the field of large language model (LLM) compression, singular value decomposition (SVD) is a widely studied and adopted low-rank decomposition technique. Since SVD operates exclusively on linear modules, and these modules in LLMs are separated by nonlinear components, SVD can only be applied independently to each linear module. Under a global compression ratio constraint, determining the appropriate rank for different linear modules becomes a critical problem. Existing approaches, such as heuristic algorithms and mask-based training, have made progress in addressing this challenge. However, these methods still suffer from several limitations: heuristic algorithms explore the solution space within restricted regions, while mask-based training struggles to efficiently capture the relationship between singular value spectra and trainable parameters. More importantly, current methods overlook the key property that the gain function is non-smooth at a compression ratio of 1, which often leads the training process to suboptimal local minima. To address these issues, we propose an Adaptive Rank Allocation (ARA) method. Specifically, (1) ARA introduces a dedicated mask design that enables efficient mapping and updating between retained ranks and trainable parameters; and (2) it employs an additional loss function to guide parameter selection toward globally optimal solutions. Experimental results demonstrate that ARA achieves state-of-the-art performance. On the LLaMA2-7B model with a 80\% compression ratio, ARA reduces perplexity on WikiText2 from 8.38 to 6.42 and improves average zero-shot task accuracy by 9.72 percentage points compared with uniform compression. These results highlight the effectiveness of our method for rank allocation in SVD-based LLM compression.

模型压缩SVD秩分配大模型

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