PoLAR: Polar-Decomposed Low-Rank Adapter Representation

Bingcong Li (ETH Zurich) · Liang Zhang (Renmin University of China) · Niao He (ETH Zurich) · Kai Lion (Department of Computer Science, ETHZ - ETH Zurich)
benchmark evaluationcommonsense reasoningconvergence ratedirection matricesfine-tuning performancegeneral language understandinglow-rank adaptationlow-rank updatemathematical problem solvingparameterizationpolar decompositionriemannian optimizationscale matrixstable rankstiefel manifolds

We show that low-rank adaptation of large-scale models suffers from a low stable rank that is well below the linear algebraic rank of the subspace, degrading fine-tuning performance. To mitigate the underutilization of the allocated subspace, we propose PoLAR, a parameterization inspired by the polar decomposition that factorizes the low-rank update into two direction matrices constrained to Stiefel manifolds and an unconstrained scale matrix. Our theory shows that PoLAR yields an exponentially faster convergence rate on a canonical low-rank adaptation problem. Pairing the parameterization with Riemannian optimization leads to consistent gains on three different benchmarks testing general language understanding, commonsense reasoning, and mathematical problem solving with base model sizes ranging from 350M to 27B.