arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.04009cs.CVcs.AI
收藏

低秩注意力侧调优用于参数高效微调

Low-rank Attention Side-Tuning for Parameter-Efficient Fine-Tuning

National Key Laboratory for Novel Software Technology(现代软件技术国家重点实验室);School of Artificial Intelligence, Nanjing University(南京大学人工智能学院)

Ningyuan Tang, Minghao Fu, Ke Zhu, Jianxin Wu

AI总结:

提出低秩注意力侧调优(LAST),通过冻结预训练模型参数和输出,训练仅含低秩自注意力的侧网络,在VTAB-1K等任务上以更少内存和时间实现更高准确率。

摘要:

在将大型预训练模型微调至下游任务时,参数高效微调(PEFT)方法能够以少量可训练参数有效微调预训练模型,但存在高GPU内存消耗和训练速度慢的问题。由于这些方法的可学习参数与预训练模型纠缠在一起,微调过程中必须计算并存储与冻结的预训练模型参数相关的梯度。我们提出了低秩注意力侧调优(LAST),该方法通过不仅冻结参数还冻结预训练网络的输出来将可训练模块与预训练模型解耦。LAST训练一个仅由低秩自注意力模块组成的侧网络。通过将预训练模型视为冻结的特征提取器,侧网络接收预训练模型的中间输出,专注于学习任务特定知识。我们还展示了LAST可以在多个优化目标上高度并行,使其在下游任务适应中非常高效,例如在寻找最优超参数时。LAST在VTAB-1K和其他视觉适应任务上超越了先前的最先进方法,相比现有PEFT方法仅使用约30%的GPU内存占用和60%的训练时间,但实现了显著更高的准确率。

英文摘要:

In finetuning a large pretrained model to downstream tasks, parameter-efficient fine-tuning (PEFT) methods can effectively finetune pretrained models with few trainable parameters, but suffer from high GPU memory consumption and slow training speed. Because learnable parameters from these methods are entangled with the pretrained model, gradients related to the frozen pretrained model's parameters have to be computed and stored during finetuning. We propose Low-rank Attention Side-Tuning (LAST), which disentangles the trainable module from the pretrained model by freezing not only parameters but also outputs of the pretrained network. LAST trains a side-network composed of only low-rank self-attention modules. By viewing the pretrained model as a frozen feature extractor, the side-network takes intermediate output from the pretrained model and focus on learning task-specific knowledge. We also show that LAST can be highly parallel across multiple optimization objectives, making it very efficient in downstream task adaptation, for example, in finding optimal hyperparameters. LAST outperforms previous state-of-the-art methods on VTAB-1K and other visual adaptation tasks with roughly only 30\% of GPU memory footprint and 60\% of training time compared to existing PEFT methods, but achieves significantly higher accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑