通过结构化剪枝让低秩适配器更高效,训练大模型、部署小模型。
Train Large, Deploy Compact: Structured Compression for Compact Low-Rank Adaptation
- 训练时用过参数初始化,再动态剪枝不重要组件,灵活分配低秩能力。
- 在数学推理、代码生成等任务上超越传统LoRA,在不同稀疏度下表现更优。
- 首次理论证明梯度剪枝比激活剪枝更鲁棒,适合追求高效微调的研究者。
低秩适配(LoRA)是大语言模型参数高效微调的常用方法,但其表达能力常落后于全量微调。在LoRA框架下,关键问题是如何从过参数化空间中获得高表达力的低秩适配器。本文提出PrunedLoRA,一种利用结构化剪枝从过参数化初始化中提取高表达力低秩适配器的新框架。不同于固定低秩预算的方法,PrunedLoRA在微调过程中动态剪枝不重要组件并防止其恢复,实现灵活自适应的秩分配。针对结构化剪枝,通过最小化整体损失的剪枝误差,提出基于梯度的细粒度剪枝与恢复更新策略,具有明确的可解释性。我们首次提供结构化剪枝的鲁棒性理论分析,严格证明在权重扰动下,基于梯度的剪枝相比基于激活的剪枝对整体损失更具鲁棒性。实验表明,PrunedLoRA在数学推理、代码生成和自然语言理解等监督微调任务中持续优于标准LoRA及其变体,并在多种稀疏度下优于现有结构化剪枝方法。
原文摘要 · Abstract (English)
Low-rank adaptation (LoRA) has become a widely used paradigm for parameter-efficient fine-tuning of large language models, yet its representational capacity often lags behind full fine-tuning. Within the context of LoRA, a key open question is how to obtain expressive low-rank adapters from over-parameterized spaces. We propose \textit{PrunedLoRA}, a new framework that leverages structured pruning to obtain highly representative low-rank adapters from an over-parameterized initialization. Unlike prior approaches that impose a fixed low-rank budget, PrunedLoRA dynamically prunes less important components during fine-tuning and prevents their reactivation, enabling flexible and adaptive rank allocation. For structured pruning, by minimizing the pruning error for overall loss, we provide fine-grained pruning and recovery updates in a gradient-based pruning strategy with grounded interpretation. We provide the first theoretical analysis of the robustness of structured pruning and provably show that under the impact of weight perturbation, gradient-based pruning is more robust than activation-based pruning with respect to overall loss. Empirically, PrunedLoRA consistently outperforms LoRA and its variants across supervised fine-tuning tasks in mathematical reasoning, code generation, and natural language understanding, and it also demonstrates advantages over existing structured pruning methods across diverse sparsity levels.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。