arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.09694cs.CLcs.AIcs.CV

LLMs可以压缩LLMs:通过代理的自适应剪枝

LLMs can Compress LLMs: Adaptive Pruning by Agents

Research and Development, Sports Vision, Inc.(研发部,Sports Vision公司);Research and Development, Vizworld Inc.(研发部,Vizworld公司)

Sai Varun Kodathala, Rakesh Vunnam

AI总结:

本文提出通过代理引导的自适应剪枝方法,有效压缩LLMs,提升事实知识保留和模型性能。

摘要:

随着大型语言模型(LLMs)的持续扩展,训练后剪枝已成为一种有前途的方法,用于减少计算成本同时保持性能。现有的方法如SparseGPT和Wanda通过逐层权重重建或激活感知的幅度剪枝实现了高稀疏度,但依赖于均匀或手工设计的启发式方法来确定每层的稀疏度比率。此外,近期研究表明,剪枝后的LLMs会遭受严重的事实知识退化,结构化剪枝方法在事实性问答能力上几乎完全崩溃。我们引入了代理引导的剪枝,其中基础模型充当自适应剪枝代理,能够智能地在每次迭代中选择哪些层进行剪枝,同时保持关键知识路径。我们的方法通过结合受Wanda启发的权重-激活指标和梯度重要性分数构建逐层敏感性图谱,将其标准化为z分数以进行模型无关的比较。这些统计信息由配备自我反思能力的LLM代理处理,使其能够从之前的剪枝结果中学习并迭代优化其策略。一个检查点回滚机制通过在感知度退化超过阈值时回退来维持模型质量。我们在Qwen3模型(4B和8B参数)上评估了我们的方法,稀疏度约为45%,在结构化剪枝基线上显示出显著的改进:在MMLU准确性上提高了56%,在FreebaseQA上的事实知识保留效果提高了19倍,且感知度退化降低了69%。值得注意的是,我们的框架不需要重新训练,具有模型无关性,并且仅在21-40次迭代中需要2-4次回退,就证明了基础模型能够有效引导其他基础模型的压缩。

英文摘要:

As Large Language Models (LLMs) continue to scale, post-training pruning has emerged as a promising approach to reduce computational costs while preserving performance. Existing methods such as SparseGPT and Wanda achieve high sparsity through layer-wise weight reconstruction or activation-aware magnitude pruning, but rely on uniform or hand-crafted heuristics to determine per-layer sparsity ratios. Moreover, recent work has shown that pruned LLMs suffer from severe factual knowledge degradation, with structured pruning methods experiencing near-total collapse in factual question-answering capabilities. We introduce agent-guided pruning, where a foundation model acts as an adaptive pruning agent to intelligently select which layers to prune at each iteration while preserving critical knowledge pathways. Our method constructs layer-wise sensitivity profiles by combining Wanda-inspired weight-activation metrics with gradient importance scores, normalized as z-scores for model-agnostic comparison. These statistics are processed by an LLM agent equipped with self-reflection capabilities, enabling it to learn from previous pruning outcomes and iteratively refine its strategy. A checkpoint rollback mechanism maintains model quality by reverting when perplexity degradation exceeds a threshold. We evaluate our approach on Qwen3 models (4B and 8B parameters) at approximately 45% sparsity, demonstrating substantial improvements over structured pruning baselines: 56% relative improvement in MMLU accuracy, 19x better factual knowledge retention on FreebaseQA, and 69% lower perplexity degradation. Notably, our framework requires no retraining, operates in a model-agnostic manner, and exhibits effective self-correction with only 2-4 rollbacks across 21-40 iterations, demonstrating that foundation models can effectively guide the compression of other foundation models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑