arXiv:2510.15304cs.CVcs.LG2025-10NeurIPS被引 5

通过层拼接压缩大模型,保留更多原始性能。

Layer as Puzzle Pieces: Compressing Large Language Models through Layer Concatenation

  • 按通道敏感度筛选关键通道,逐层拼接融合减少参数。
  • 剪枝30%参数后,LLaMA-2-7b仍保持83%原始准确率。
  • 适合需要高效压缩且不牺牲性能的模型部署场景。

大型语言模型在自然语言处理中表现优异,但其庞大体积带来高计算与存储开销。现有分层结构化剪枝方法常忽视剪枝部分能力的保留。本文发现三大局限:直接删除层导致性能显著下降、线性权重层聚合能力不足、缺乏有效的微调恢复机制。为此提出CoMe框架,包含渐进式剪枝流程、基于拼接的层融合技术及分层知识蒸馏过程。首先引入通道敏感度指标,结合激活强度与权重范数实现细粒度通道选择;随后采用拼接融合相邻层关键通道,实现渐进式模型压缩;最后通过剪枝阶段建立的层间对应关系,进行高效知识迁移。在七个基准测试上验证,剪枝30%参数后,LLaMA-2-7b平均准确率仍达原模型的83%,性能优于现有方法。代码已开源。

原文摘要 · Abstract (English)

Large Language Models excel at natural language processing tasks, but their massive size leads to high computational and storage demands. Recent works have sought to reduce their model size through layer-wise structured pruning. However, they tend to ignore retaining the capabilities in the pruned part. In this work, we re-examine structured pruning paradigms and uncover several key limitations: 1) notable performance degradation due to direct layer removal, 2) incompetent linear weight layer aggregation, and 3) the lack of effective post-training recovery mechanisms. To address these limitations, we propose CoMe, including a progressive layer pruning framework with a Concatenation-based Merging technology and a hierarchical distillation post-training process. Specifically, we introduce a channel sensitivity metric that utilizes activation intensity and weight norms for fine-grained channel selection. Subsequently, we employ a concatenation-based layer merging method to fuse the most critical channels across adjacent layers, enabling progressive model size reduction. Finally, we propose a hierarchical distillation protocol that leverages the correspondences between the original and pruned model layers established during pruning, thereby enabling efficient knowledge transfer. Experiments on seven benchmarks show that CoMe achieves state-of-the-art performance; when pruning 30% of LLaMA-2-7b's parameters, the pruned model retains 83% of its original average accuracy. Our code is available at https://github.com/MPI-Lab/CoMe.

模型压缩剪枝LLaMA蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。