arXiv:2411.01800cs.CVcs.LG2024-11NeurIPS被引 11

提出SNELL方法,在低内存下实现高效稀疏微调。

Expanding Sparse Tuning for Low Memory Usage

  • 将可调矩阵分解为两个低秩矩阵,避免存储完整权重
  • 采用竞争机制无需存储可调权重索引,减少内存占用
  • 结合非线性核函数提升模型适应能力,适合大模型微调

参数高效微调(PEFT)通过仅调整少量参数,有效适配预训练视觉模型到下游任务。其中,稀疏微调通过仅更新与下游任务最相关的权重,取得更优性能。然而,该方法因需在优化器中存储完整权重矩阵及可调权重索引,导致内存开销上升。本文提出SNELL(Sparse tuning with kerNELized LoRA)方法,通过将可调矩阵分解为两个可学习的低秩矩阵,避免存储原始完整矩阵;进一步设计竞争式稀疏机制,无需存储可调权重索引。为增强低秩矩阵的表达能力,引入非线性核函数进行全矩阵合并,提升合并后矩阵的秩,从而强化模型对下游任务的适应能力。大量实验表明,SNELL在多个下游任务上实现领先性能,同时保持低内存使用,使稀疏微调适用于大规模模型。代码已开源。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) is an effective method for adapting pre-trained vision models to downstream tasks by tuning a small subset of parameters. Among PEFT methods, sparse tuning achieves superior performance by only adjusting the weights most relevant to downstream tasks, rather than densely tuning the whole weight matrix. However, this performance improvement has been accompanied by increases in memory usage, which stems from two factors, i.e., the storage of the whole weight matrix as learnable parameters in the optimizer and the additional storage of tunable weight indexes. In this paper, we propose a method named SNELL (Sparse tuning with kerNELized LoRA) for sparse tuning with low memory usage. To achieve low memory usage, SNELL decomposes the tunable matrix for sparsification into two learnable low-rank matrices, saving from the costly storage of the whole original matrix. A competition-based sparsification mechanism is further proposed to avoid the storage of tunable weight indexes. To maintain the effectiveness of sparse tuning with low-rank matrices, we extend the low-rank decomposition by applying nonlinear kernel functions to the whole-matrix merging. Consequently, we gain an increase in the rank of the merged matrix, enhancing the ability of SNELL in adapting the pre-trained models to downstream tasks. Extensive experiments on multiple downstream tasks show that SNELL achieves state-of-the-art performance with low memory usage, endowing PEFT with sparse tuning to large-scale models. Codes are available at https://github.com/ssfgunner/SNELL.

稀疏微调低内存LoRA大模型

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