提出可证明收敛的LoRA改进框架,解决低秩微调不收敛难题。
Randomized Asymmetric Chain of LoRA: The First Meaningful Theoretical Framework for Low-Rank Adaptation
- 设计随机非对称链式结构,修复传统LoRA的收敛缺陷
- 理论证明收敛到全参数微调相同解,且给出收敛速率
- 适用于梯度下降、随机梯度及联邦学习等多种场景
微调已成为将大模型适配特定任务的常用方法。随着模型与数据规模增长,参数高效微调技术愈发重要。其中最广泛使用的方法是低秩适配(LoRA),其更新形式为两个低秩矩阵的乘积。尽管LoRA在微调中表现优异,但通常仍落后于全参数微调(FPFT)。尽管已有大量LoRA变体被经验性研究,其理论优化分析却严重不足。本文首次证明:标准LoRA及其两种扩展(非对称LoRA、链式LoRA)确实存在收敛问题。为此,我们提出随机非对称链式LoRA(RAC-LoRA)——一个严格分析收敛率的通用优化框架。该方法继承了LoRA类启发式的优势,通过若干关键算法修改,实现可证明收敛。本框架实现了全参数微调与低秩适配之间的理论桥梁,提供收敛至相同解的保证及收敛速率。此外,还针对光滑非凸损失函数,涵盖梯度下降、随机梯度下降与联邦学习设置进行了收敛分析。理论结果得到实验验证。
原文摘要 · Abstract (English)
Fine-tuning has become a popular approach to adapting large foundational models to specific tasks. As the size of models and datasets grows, parameter-efficient fine-tuning techniques are increasingly important. One of the most widely used methods is Low-Rank Adaptation (LoRA), with adaptation update expressed as the product of two low-rank matrices. While LoRA was shown to possess strong performance in fine-tuning, it often under-performs when compared to full-parameter fine-tuning (FPFT). Although many variants of LoRA have been extensively studied empirically, their theoretical optimization analysis is heavily under-explored. The starting point of our work is a demonstration that LoRA and its two extensions, Asymmetric LoRA and Chain of LoRA, indeed encounter convergence issues. To address these issues, we propose Randomized Asymmetric Chain of LoRA (RAC-LoRA) -- a general optimization framework that rigorously analyzes the convergence rates of LoRA-based methods. Our approach inherits the empirical benefits of LoRA-style heuristics, but introduces several small but important algorithmic modifications which turn it into a provably convergent method. Our framework serves as a bridge between FPFT and low-rank adaptation. We provide provable guarantees of convergence to the same solution as FPFT, along with the rate of convergence. Additionally, we present a convergence analysis for smooth, non-convex loss functions, covering gradient descent, stochastic gradient descent, and federated learning settings. Our theoretical findings are supported by experimental results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。