arXiv:2510.18030cs.CLcs.AI2025-10ACL被引 3

通过全局迭代剪枝提升大模型效率,显著改善下游任务表现。

From Local to Global: Revisiting Structured Pruning Paradigms for Large Language Models

  • 基于目标损失的全局重要性度量,迭代剪枝保持精度
  • 40%-50%稀疏度下,语言建模困惑度降低,下游准确率提升
  • 支持一次剪枝多次部署,适配任务定制化剪枝

结构化剪枝是高效部署大语言模型的实用方法,能生成紧凑且硬件友好的架构。然而主流的局部剪枝范式缺乏任务针对性:通过优化层内重建而非任务目标,往往仅保留困惑度或通用零样本行为,难以利用微弱的任务校准信号,导致下游性能提升有限。本文重新审视全局结构化剪枝,提出 GISP(Global Iterative Structured Pruning),一种后训练方法,通过一阶、基于损失的重要度分数,在结构层面聚合并进行块级归一化,以移除注意力头和MLP通道。基于此全局重要性度量,GISP采用迭代剪枝策略,而非一次性剪枝,有效稳定高稀疏度下的准确率,避免困惑度崩溃,且无需中间微调。更重要的是,迭代剪枝形成嵌套子网络,支持“剪枝一次,部署多次”的工作流。此外,GISP直接以目标损失定义结构重要性,便于适配任务特定目标,本研究中分别使用困惑度用于语言建模,以及基于边界的目标用于决策类任务。大量实验表明,在 Llama2-7B/13B、Llama3-8B、Mistral-0.3-7B 上,GISP持续降低 WikiText-2 困惑度,并提升下游准确率,尤其在 40%-50% 稀疏度下表现突出;在 DeepSeek-R1-Distill-Llama-3-8B 与 Qwen3-8B 上,针对 GSM8K 的任务对齐校准显著提升精确匹配率。代码已开源:https://github.com/uncc-efficient-ai/GISP。

原文摘要 · Abstract (English)

Structured pruning is a practical approach to deploying large language models (LLMs) efficiently, as it yields compact, hardware-friendly architectures. However, the dominant local paradigm is task-agnostic: by optimizing layer-wise reconstruction rather than task objectives, it tends to preserve perplexity or generic zero-shot behavior but fails to capitalize on modest task-specific calibration signals, often yielding limited downstream gains. We revisit global structured pruning and present GISP, Global Iterative Structured Pruning, a post-training method that removes attention heads and MLP channels using first-order, loss-based important scores aggregated at the structure level with block-wise normalization. Built on this global importance metric, GISP adopts an iterative schedule, rather than one-shot pruning, stabilizes accuracy at higher sparsity, and mitigates perplexity collapse without requiring intermediate fine-tuning. Importantly, the iterative pruning forms nested subnetworks that support a ''prune-once, deploy-many'' workflow. Furthermore, GISP defines structural importance directly with respect to a target loss, making it easy to adapt pruning to task-specific objectives. In this work, we use perplexity for language modeling and a margin-based objective for decision-style tasks. Extensive experiments show that across Llama2-7B/13B, Llama3-8B, and Mistral-0.3-7B, GISP consistently lowers WikiText-2 perplexity and improves on downstream accuracy, with especially strong gains at 40-50% sparsity; on DeepSeek-R1-Distill-Llama-3-8B and Qwen3-8B with GSM8K, task-aligned calibration substantially boosts exact-match accuracy. The implementation is available at https://github.com/uncc-efficient-ai/GISP.

模型剪枝大模型优化高效推理

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