通过交替正交旋转补偿剪枝损失,提升大模型压缩后的精度。
COEC: Calibrated Orthogonal-Equivalence Compensation for Structured Pruning of Large Language Models

- 用左右交替正交旋转优化剪枝后权重,保持输入不变性。
- 在高稀疏度下显著降低困惑度,零样本准确率普遍提升。
- 无需反向传播或重训练,适配多种剪枝方法与模型架构。
结构化剪枝通过移除权重列来减小大语言模型(LLM)的规模和推理开销,但会引入输出误差导致性能下降。现有无训练补偿方法仅在保留权重输出侧使用加性偏置或单一正交旋转,未改变输入奇异方向,限制了权重自适应能力。本文提出COEC(校准正交等价补偿),一种无训练补偿框架,对保留权重施加交替左、右正交旋转。右旋转在降维的Stiefel流形上优化,奇异值通过广义交叉验证选择每层正则强度。COEC进一步调节校准格拉姆矩阵以抑制高能激活方向主导,并引入对齐惩罚项,保持相邻注意力投影间的几何关系。所有组件均基于小校准集的二阶统计量,无需通过模型反向传播或重训练模型参数。COEC独立于列剪枝准则,可应用于多种结构化剪枝方法。在Llama-3、Llama-3.1和Qwen2.5模型族上,多个结构稀疏度下的实验表明,COEC在所有模型上均提升困惑度,多数设置下提升零样本准确率,稀疏度越高收益越明显。结果表明,剪枝后补偿可部分恢复因列移除损失的性能。
原文摘要 · Abstract (English)
Structured pruning reduces the size and inference cost of large language models (LLMs) by removing weight columns, but the resulting output error can degrade accuracy. Existing training-free compensation methods use an additive bias or a single orthogonal rotation on the output side of the retained weight. These corrections leave its input singular frame unchanged and therefore limit how the retained weight can adapt after column removal. We propose COEC (Calibrated Orthogonal-Equivalence Compensation), a training-free compensation framework that applies alternating left and right orthogonal rotations to the retained weight. The right rotation is optimized on a reduced Stiefel manifold, while singular values are rescaled using generalized cross-validation to select the regularization strength for each layer. COEC further tempers the calibration Gram matrix to reduce the dominance of high-energy activation directions and introduces an alignment penalty that preserves the geometric relation between adjacent attention projections.All components use second-order statistics from a small calibration set and require neither backpropagation through the LLM nor retraining of the model parameters. COEC is independent of the column pruning criterion and can be applied to multiple structured pruning methods. Experiments on the Llama-3, Llama-3.1, and Qwen2.5 model families across multiple structured sparsity levels show that COEC improves perplexity on every model and zero-shot accuracy in most settings over existing compensation methods, with larger gains at higher sparsity. These results show that post-pruning compensation can recover part of the performance lost to column removal.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。