cs.LGJul 24, 2026

On the Convergence of Stochastic Low-Rank Adaptation

Authors: Ru WangChengchang LiuJohn C. S. Lui

Organizations: The Chinese University of Hong Kong · Westlake University

Abstract

Low-rank adaptation (LoRA) optimizes J(B,A)=L(Wbase+sBA)J(B,A)=\mathcal L(W_\mathrm{base}+sBA) over two adapters BRm×rB \in \mathbb{R}^{m \times r} and ARr×nA \in \mathbb{R}^{r \times n} that form a low-rank update to a frozen pretrained weight matrix WbaseRm×nW_\mathrm{base} \in \mathbb{R}^{m \times n}. The prior analysis shows LoRA-GD takes exp{O(ε2)}\exp\{\mathcal{O}(ε^{-2})\} oracle calls to find an εε-stationary point such that J(B,A)ε\|\nabla J(B,A)\|\leq ε in the deterministic setting. We sharpen the analysis and show that O(ε4)\mathcal{O}(ε^{-4}) full-gradient evaluations suffice for the same first-order criterion. We further study stochastic LoRA under unbiased gradient estimates and finite variance. We propose LoRA-NSGDM, which finds an εε-stationary point with O(ε8)\mathcal{O}(ε^{-8}) stochastic oracle complexity. Under the additional mean-square smoothness condition, we use variance reduction strategy and propose LoRA-STORM, which improves the stochastic oracle complexity to O(ε6)\mathcal{O}(ε^{-6}).

Explore similar work

CardsList