提出基于重要性评估的低秩压缩方法,精准筛选关键模型基向量。
Importance-Guided Basis Selection for Low-Rank Decomposition of Large Language Models
- 通过损失函数二阶展开计算基向量移除的影响,结合梯度与曲率判断重要性。
- 在数学推理任务上优于现有方法,深度压缩下性能提升更显著。
- 适用于需要高效压缩大模型的研究者,尤其关注模型轻量化部署。
低秩分解是压缩大语言模型的有效方法,但其效果取决于保留哪些奇异向量基。现有方法如Basel在下游数据上调整奇异值系数并剪枝幅度小的基,这种启发式策略可能与任务表现不一致,因忽略了损失景观的局部几何结构。本文提出基于重要性的基选择(BSI)框架,直接估计移除每个基所导致的期望损失增加。BSI从损失对奇异值的二阶泰勒展开中推导出基于导数的重要度评分,融合一阶敏感性和二阶曲率以量化剪枝影响。为使该准则适用于大模型,我们改进了哈西迪安对角估计器,采用对称参数扰动的哈钦森随机探测法。提供了全面的理论分析,包括剪枝后的损失增长边界、哈西迪安对角估计误差对边界的传播、与哈西迪安谱相关的方差刻画、实现目标估计精度的高概率样本复杂性保证,以及扰动强度的选择指导。在数学推理基准上的大量实验表明,BSI持续优于最先进的低秩分解基线,尤其是在深度压缩条件下表现突出。
原文摘要 · Abstract (English)
Low-rank decomposition is a compelling approach for compressing large language models, but its effectiveness hinges on selecting which singular-vector bases to retain for a target task. Existing methods such as Basel adapt singular-value coefficients on downstream data and prune bases with small re-learned magnitudes, a heuristic that can be misaligned with task performance because it ignores the local geometry of the loss landscape. We present Basis Selection with Importance (BSI), a principled low-rank compression framework that ranks and prunes bases by directly estimating the expected loss increase incurred when each basis is removed. BSI derives a derivative-based importance score from a second-order Taylor expansion of the task loss with respect to singular values, combining first-order sensitivity and second-order curvature to quantify pruning impact. To make this criterion practical for LLMs, we develop an efficient Hessian-diagonal estimator by adapting the Hutchinson randomized-probing method to loss curvature with symmetric parameter perturbations. We provide a comprehensive theoretical analysis, including loss-increase bounds under basis pruning, explicit propagation of Hessian-diagonal estimation error into these bounds, variance characterization tied to the Hessian spectrum, high-probability sample-complexity guarantees for achieving a target estimation accuracy, and guidance on perturbation intensity. Extensive experiments on mathematical reasoning benchmarks demonstrate that BSI consistently outperforms state-of-the-art low-rank decomposition baselines, with especially strong improvements under deep compression.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。