arXiv:2512.13733cs.LGcs.AI2025-12被引 2

用可学习的掩码自动选择每层压缩秩,无需微调即可高效压缩大模型。

Low-Rank Compression of Language Models via Differentiable Rank Selection

  • 通过可训练掩码直接学习每层应保留的奇异值数量。
  • 在20%压缩率下,比STRS在多个任务上高出3.5%至12%。
  • 无需微调,性能媲美需微调的方法,适合部署场景。

基于低秩分解的大型语言模型压缩方法已取得进展,特别是引入了激活和损失感知的SVD,提升了压缩率与下游任务性能之间的权衡。然而,如何为每层选择最优秩以联合优化压缩率和准确率仍是挑战:现有方法或依赖启发式规则(搜索空间有限,效果不佳),或为基于梯度的方法但未微调时表现不如启发式方法。为此,我们提出无微调的可学习低秩压缩(LLRC),直接学习用于选择奇异值的掩码权重。利用校准数据集,仅训练掩码权重,在最小化中间激活与原模型差异的同时逐步减少奇异值数量。在常见推理和开放域问答任务上,该方法在多种压缩率下均优于其他无需微调的秩选择方法。例如,在Llama-2-13B模型上,20%压缩率下,于MMLU、BoolQ和OpenbookQA任务上分别优于敏感性截断秩搜索(STRS)12%、3.5%和4.4%。相比其他压缩技术,本方法始终优于SVD-LLM和LLM-Pruner的无微调版本,并在性能上可媲美需微调的LLM-Pruner变体。

原文摘要 · Abstract (English)

Approaches for compressing large-language models using low-rank decomposition have made strides, particularly with the introduction of activation and loss-aware SVD, which improves the trade-off between decomposition rank and downstream task performance. Despite these advancements, a persistent challenge remains--selecting the optimal ranks for each layer to jointly optimise compression rate and downstream task accuracy. Current methods either rely on heuristics that can yield sub-optimal results due to their limited discrete search space or are gradient-based but are not as performant as heuristic approaches without post-compression fine-tuning. To address these issues, we propose Learning to Low-Rank Compress (LLRC), a gradient-based approach which directly learns the weights of masks that select singular values in a fine-tuning-free setting. Using a calibration dataset, we train only the mask weights to select fewer and fewer singular values while minimising the divergence of intermediate activations from the original model. Our approach outperforms competing ranking selection methods that similarly require no post-compression fine-tuning across various compression rates on common-sense reasoning and open-domain question-answering tasks. For instance, with a compression rate of 20% on Llama-2-13B, LLRC outperforms the competitive Sensitivity-based Truncation Rank Searching (STRS) on MMLU, BoolQ, and OpenbookQA by 12%, 3.5%, and 4.4%, respectively. Compared to other compression techniques, our approach consistently outperforms fine-tuning-free variants of SVD-LLM and LLM-Pruner across datasets and compression rates. Our fine-tuning-free approach also performs competitively with the fine-tuning variant of LLM-Pruner.

模型压缩低秩分解无微调

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