新方法让大模型压缩更懂知识,提升事实记忆准确率6-8%。
Beyond Variance: Knowledge-Aware LLM Compression via Fisher-Aligned Subspace Diagnostics
- 用梯度耦合度筛选压缩维度,不再只看方差。
- 在50%降秩下,知识任务准确率比传统方法高6-8%。
- 适合追求知识保持的模型轻量化部署场景。
后训练激活压缩对在资源受限硬件上部署大语言模型至关重要。然而,标准方法如奇异值分解(SVD)对梯度不敏感:它们仅保留高方差维度,而忽略其对事实知识保存的影响。本文提出鱼群对齐子空间压缩(FASC),一种基于知识感知的压缩框架,通过直接建模激活-梯度耦合关系,最小化损失函数的二阶近似。FASC利用费舍尔信息矩阵识别对事实知识至关重要的维度,这些维度通常位于低方差但高梯度敏感性的子空间中。我们提出依赖性违背评分( {ho})作为通用诊断指标,量化激活-梯度耦合程度,揭示了事实知识在Transformer架构中的存储位置。在Mistral-7B和Llama-3-8B上的大量实验表明,相较于基于方差的方法,在50%秩缩减条件下,FASC在知识密集型基准测试(MMLU、LAMA)上保持了6-8%更高的准确率,有效实现7B模型的事实召回能力媲美未压缩的13B模型。分析显示, {ho}是存储知识的根本信号,只有当模型在训练中内化事实关联时,高- {ho}层才会出现。
原文摘要 · Abstract (English)
Post-training activation compression is essential for deploying Large Language Models (LLMs) on resource-constrained hardware. However, standard methods like Singular Value Decomposition (SVD) are gradient-blind: they preserve high-variance dimensions regardless of their impact on factual knowledge preservation. We introduce Fisher-Aligned Subspace Compression (FASC), a knowledge-aware compression framework that selects subspaces by directly modeling activation-gradient coupling, minimizing a second-order surrogate of the loss function. FASC leverages the Fisher Information Matrix to identify dimensions critical for factual knowledge, which often reside in low-variance but high-gradient-sensitivity subspaces. We propose the Dependence Violation Score (\r{ho}) as a general-purpose diagnostic metric that quantifies activation-gradient coupling, revealing where factual knowledge is stored within transformer architectures. Extensive experiments on Mistral-7B and Llama-3-8B demonstrate that FASC preserves 6-8% more accuracy on knowledge-intensive benchmarks (MMLU, LAMA) compared to variance-based methods at 50% rank reduction, effectively enabling a 7B model to match the factual recall of a 13B uncompressed model. Our analysis reveals that \r{ho} serves as a fundamental signal of stored knowledge, with high-\r{ho} layers emerging only when models internalize factual associations during training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。