用微分方程优化LoRA,提升训练稳定性和收敛性
ODELoRA: Training Low-Rank Adaptation by Solving Ordinary Differential Equations
- 将LoRA参数优化建模为微分方程,保持参数平衡结构
- 在矩阵感知任务中实现线性收敛,训练更稳定
- 适合需要高稳定性与高效微调的深度网络场景
低秩适应(LoRA)因其可减少训练参数和内存开销,已成为深度迁移学习中广泛采用的参数高效微调方法。然而,传统LoRA训练方法独立优化低秩因子矩阵,使用标准梯度算法,这种解耦优化在理论上和实证上均非最优,未能充分利用LoRA参数化的内在结构。本文提出一种新的连续时间优化动态,以常微分方程(ODE)形式模拟全量微调在平衡流形上的梯度流,称为ODELoRA。为准确追踪其轨迹,采用欧拉法与龙格-库塔法等成熟的时间离散化方案。该框架提供统一的ODE视角来理解和设计LoRA训练算法。我们在强凸目标下建立了某些离散化方案的线性收敛性,并进一步扩展至矩阵感知设定。此外,我们证明了ODELoRA具备稳定的特征学习能力,这对不同规模问题维度下的深度神经网络训练至关重要。矩阵感知任务的实验验证了理论线性收敛行为,物理信息神经网络的实验也表明,相比现有基线,ODELoRA在训练稳定性方面表现更优。
原文摘要 · Abstract (English)
Low-rank adaptation (LoRA) has emerged as a widely adopted parameter-efficient fine-tuning method in deep transfer learning, due to its reduced number of trainable parameters and lower memory requirements enabled by Burer-Monteiro factorization on adaptation matrices. However, classical LoRA training methods treat the low-rank factor matrices individually and optimize them using standard gradient-based algorithms. Such decoupled optimization schemes are theoretically and empirically suboptimal, as they fail to fully exploit the intrinsic structure of the LoRA parameterization. In this work, we propose a novel continuous-time optimization dynamic for LoRA factor matrices in the form of an ordinary differential equation (ODE) that emulates the gradient flow of full fine-tuning on the balanced manifold. We term this approach ODELoRA. To faithfully track the trajectories of ODELoRA, we adopt well-established and theoretically grounded time-discretization schemes, including Euler and Runge--Kutta methods. Our framework provides a unified ODE-based perspective for understanding and designing LoRA training algorithms. We establish linear convergence of the proposed method under strongly convex objectives for certain discretization schemes under mild conditions, and further extend our analysis to the matrix sensing setting. Moreover, we show that ODELoRA achieves stable feature learning, a property that is crucial for training deep neural networks at different scales of problem dimensionality. Empirical results on matrix sensing tasks confirm the derived linear convergence behavior, and experiments on training physics-informed neural networks further demonstrate the superiority of ODELoRA over existing baselines, especially in the training stability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。