arXiv:2410.11772cs.CLcs.LG2024-10EMNLP被引 13

按层重要性选关键层微调,省内存还更准。

Layer-wise Importance Matters: Less Memory for Better Performance in Parameter-efficient Fine-tuning of Large Language Models

  • 根据每层重要性动态选关键层微调,不搞一刀切。
  • 在多个大模型和任务上,比传统方法准确率提升1.5%~3.2%。
  • 适配各种层式微调方法,轻量高效,适合资源受限场景。

参数高效微调(PEFT)方法因显著降低内存与计算开销而广泛用于大语言模型的下游任务适配。然而,多数方法对所有层采用统一架构,忽略各层重要性差异,导致微调效果欠佳。为此,我们提出重要性感知稀疏微调(IST),充分利用模型内在稀疏性,通过层级重要性评分筛选最具价值的全层子集。该方法为通用可插拔技术,兼容各类基于层的PEFT方法。借助重要性评分,IST动态更新选定层,大幅降低内存需求。我们提供收敛性理论证明及实证结果,验证其优于均匀更新策略。在多种大模型、PEFT方法和下游任务上的广泛实验表明,IST能有效提升现有层式微调方法性能。代码已开源:https://github.com/Kaiseem/IST。

原文摘要 · Abstract (English)

Parameter-Efficient Fine-Tuning (PEFT) methods have gained significant popularity for adapting pre-trained Large Language Models (LLMs) to downstream tasks, primarily due to their potential to significantly reduce memory and computational overheads. However, a common limitation in most PEFT approaches is their application of a uniform architectural design across all layers. This uniformity involves identical trainable modules and ignores the varying importance of each layer, leading to sub-optimal fine-tuning results. To overcome the above limitation and obtain better performance, we develop a novel approach, Importance-aware Sparse Tuning (IST), to fully utilize the inherent sparsity and select the most important subset of full layers with effective layer-wise importance scoring. The proposed IST is a versatile and plug-and-play technique compatible with various PEFT methods that operate on a per-layer basis. By leveraging the estimated importance scores, IST dynamically updates these selected layers in PEFT modules, leading to reduced memory demands. We further provide theoretical proof of convergence and empirical evidence of superior performance to demonstrate the advantages of IST over uniform updating strategies. Extensive experiments on a range of LLMs, PEFTs, and downstream tasks substantiate the effectiveness of our proposed method, showcasing IST's capacity to enhance existing layer-based PEFT methods. Our code is available at https://github.com/Kaiseem/IST.

参数高效大模型微调稀疏训练层重要性

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