arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.04151cs.LGcs.CL
收藏

Chain of LoRA:通过残差学习实现语言模型的高效微调

Chain of LoRA: Efficient Fine-tuning of Language Models via Residual Learning

Princeton University(普林斯顿大学);Nanyang Technological University(南洋理工大学)

Wenhan Xia, Chengwei Qin, Elad Hazan

AI总结:

本文提出Chain of LoRA(COLA)迭代优化框架,通过残差学习合并并重置LoRA模块,在不增加计算和内存开销的情况下弥合LoRA与全参数微调的差距,在多种模型和七个基准任务上持续优于LoRA。

摘要:

微调是将预训练大型语言模型适配到特定任务的主要方法。随着模型规模和任务多样性的扩大,参数高效微调方法变得至关重要。其中最广泛使用的一类方法是低秩适应(LoRA)及其变体。LoRA将权重更新编码为两个低秩矩阵的乘积。尽管具有优势,但在某些任务的泛化误差方面,LoRA仍不及全参数微调。我们引入了Chain of LoRA(COLA),这是一个受Frank-Wolfe算法启发的迭代优化框架,旨在弥合LoRA与全参数微调之间的差距,且不会产生额外的计算成本或内存开销。COLA采用残差学习过程,将学习到的LoRA模块合并到预训练语言模型参数中,并为新生的LoRA模块重新初始化优化。我们提供了理论收敛保证以及实验结果来验证算法的有效性。在多种模型(OPT和llama-2)及七个基准任务上,我们证明COLA能够在没有额外计算或内存成本的情况下持续优于LoRA。

英文摘要:

Fine-tuning is the primary methodology for tailoring pre-trained large language models to specific tasks. As the model's scale and the diversity of tasks expand, parameter-efficient fine-tuning methods are of paramount importance. One of the most widely used family of methods is low-rank adaptation (LoRA) and its variants. LoRA encodes weight update as the product of two low-rank matrices. Despite its advantages, LoRA falls short of full-parameter fine-tuning in terms of generalization error for certain tasks. We introduce Chain of LoRA (COLA), an iterative optimization framework inspired by the Frank-Wolfe algorithm, to bridge the gap between LoRA and full parameter fine-tuning, without incurring additional computational costs or memory overheads. COLA employs a residual learning procedure where it merges learned LoRA modules into the pre-trained language model parameters and re-initilize optimization for new born LoRA modules. We provide theoretical convergence guarantees as well as empirical results to validate the effectiveness of our algorithm. Across various models (OPT and llama-2) and seven benchmarking tasks, we demonstrate that COLA can consistently outperform LoRA without additional computational or memory costs.

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

↑