通过补偿丢失的注意力组件,高效恢复剪枝大模型性能。
Restoring Pruned Large Language Models via Lost Component Compensation
- 基于注意力激活差异,定位并回补剪枝损失的关键信息
- 在不破坏稀疏性前提下,显著提升剪枝模型的通用与任务性能
- 适配各类剪枝方式,可直接部署于现有推理系统
剪枝是降低大语言模型规模和推理成本的常用技术,但常导致性能下降。现有恢复方法多采用参数高效微调(PEFT),如LoRA,但多数PEFT方法针对密集模型设计,忽视剪枝模型特性,恢复效果不佳。本文提出面向剪枝模型的针对性恢复策略——RestoreLCC(通过丢失组件补偿恢复剪枝大模型),其核心思想是:剪枝引发的信息损失反映在注意力激活中,通过对比分析关键注意力头的激活差异,提取丢失成分并注入对应剪枝头中实现补偿。该方法兼容结构化、半结构化和非结构化剪枝,在多项实验中均优于当前最优基线,且保持剪枝模型的稀疏性与推理效率。
原文摘要 · Abstract (English)
Pruning is a widely used technique to reduce the size and inference cost of large language models (LLMs), but it often causes performance degradation. To mitigate this, existing restoration methods typically employ parameter-efficient fine-tuning (PEFT), such as LoRA, to recover the pruned model's performance. However, most PEFT methods are designed for dense models and overlook the distinct properties of pruned models, often resulting in suboptimal recovery. In this work, we propose a targeted restoration strategy for pruned models that restores performance while preserving their low cost and high efficiency. We observe that pruning-induced information loss is reflected in attention activations, and selectively reintroducing components of this information can significantly recover model performance. Based on this insight, we introduce RestoreLCC (Restoring Pruned LLMs via Lost Component Compensation), a plug-and-play method that contrastively probes critical attention heads via activation editing, extracts lost components from activation differences, and finally injects them back into the corresponding pruned heads for compensation and recovery. RestoreLCC is compatible with structured, semi-structured, and unstructured pruning schemes. Extensive experiments demonstrate that RestoreLCC consistently outperforms state-of-the-art baselines in both general and task-specific performance recovery, without compromising the sparsity or inference efficiency of pruned models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。