arXiv:2510.07782cs.CL2025-10中稿 · ICLR

通过旋转约束补偿结构化剪枝误差,提升大模型压缩后性能。

RCPU: Rotation-Constrained Error Compensation for Structured Pruning of Large Language Models

  • 剪枝后参数在旋转约束下更新,保持输出几何结构
  • 在Llama-7B和13B上,困惑度与任务准确率均优于基线
  • 结合方差感知重要性评分,优先保留关键输入维度

本文提出一种旋转约束误差补偿方法,用于缓解大语言模型(LLM)结构化剪枝带来的误差。由于预训练使用海量数据,而剪枝仅依赖少量校准数据,导致输出不匹配不可避免。直接最小二乘拟合易过拟合校准集,破坏预训练权重。为此,本文在旋转约束下更新剪枝参数,保持输出表示的范数与内积关系,同时重新对齐剪枝子空间与原始输出。此外,强贡献于输出主方向的组件若被移除将难以恢复,而输入维度方差大的成分更影响主方向,因此设计了方差感知重要性评分,优先保留此类维度。结合该评分与旋转约束更新,方法在几何保真前提下有效补偿误差。实验在Llama-7B和Llama-2-13B上进行,评估基于WikiText2及多语言理解基准,结果表明其在困惑度与任务准确率上持续优于现有基线。

原文摘要 · Abstract (English)

In this paper, we propose a rotation-constrained compensation method to address the errors introduced by structured pruning of large language models (LLMs). LLMs are trained on massive datasets and accumulate rich semantic knowledge in their representation space. In contrast, pruning is typically carried out with only a small amount of calibration data, which makes output mismatches unavoidable. Although direct least-squares fitting can reduce such errors, it tends to overfit to the limited calibration set, destructively modifying pretrained weights. To overcome this difficulty, we update the pruned parameters under a rotation constraint. This constrained update preserves the geometry of output representations (i.e., norms and inner products) and simultaneously re-aligns the pruned subspace with the original outputs. Furthermore, in rotation-constrained compensation, removing components that strongly contribute to the principal directions of the output makes error recovery difficult. Since input dimensions with large variance strongly affect these principal directions, we design a variance-aware importance score that ensures such dimensions are preferentially kept in the pruned model. By combining this scoring rule with rotation-constrained updates, the proposed method effectively compensates errors while retaining the components likely to be more important in a geometry-preserving manner. In the experiments, we apply the proposed method to Llama-7B and Llama-2-13B, and evaluate it on WikiText2 and multiple language understanding benchmarks. The results demonstrate consistently better perplexity and task accuracy compared with existing baselines.

模型剪枝大模型压缩误差补偿几何保持

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