让大模型压缩更精准:根据激活重要性重构,瘦身不丢性能
IMPACT: Importance-Aware Activation Space Reconstruction
- 基于激活重要性加权,优化压缩方向
- 比现有方法多压缩55.4%体积,精度相当或更好
- 适合想在小设备上部署大模型的开发者
大型语言模型(LLMs)在多个领域表现优异,但因其规模大,难以在资源受限环境部署。低秩压缩是常见解决方案,通常假设权重具有低秩结构,但该假设在LLMs中并不成立。相反,LLM激活值展现出更显著的低秩特性,促使人们转向最小化激活重建误差的方法。然而,仅此还不够:不同激活维度对模型性能的贡献不均,同等对待会导致精度下降。我们提出IMPACT,一种重要性感知的激活空间重构框架,将压缩与模型性能影响直接关联。IMPACT将压缩建模为优化问题,融合激活结构与基于梯度的重要性,推导出闭式解,其中重建基由重要性加权的激活协方差矩阵生成。这实现了显式以保真度为目标的低秩压缩。在多个模型和任务上的实验表明,IMPACT相比当前最优基线,在保持或超越精度的同时,实现最高达55.4%的模型尺寸缩减。
原文摘要 · Abstract (English)
Large language models (LLMs) achieve strong performance across diverse domains but remain difficult to deploy in resource-constrained environments due to their size. Low-rank compression is a common remedy, typically minimizing weight reconstruction error under the assumption that weights are low-rank. However, this assumption often does not hold in LLMs. In contrast, LLM activations exhibit a more pronounced low-rank structure, motivating approaches that minimize activation reconstruction error. This shift alone, however, is not sufficient: different activation dimensions contribute unequally to model performance, and treating them uniformly can lead to accuracy loss. We introduce IMPACT, an importance-aware activation reconstruction framework that links compression to its effect on model performance. IMPACT formulates compression as an optimization problem that integrates activation structure with gradient-based importance, deriving a closed-form solution where reconstruction bases arise from an importance-weighted activation covariance matrix. This yields low-rank compression explicitly optimized for accuracy preservation. Experiments across multiple models and tasks demonstrate that IMPACT achieves up to 55.4% greater model size reduction while maintaining accuracy comparable to or better than state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。