How Finite-Rank Bottlenecks Shape the Low-Rank Adaptation Landscape
Organizations: College of Engineering & Computer Science, VinUniversity · Center for AI Research, VinUniversity, Hanoi, Vietnam.
Abstract
Low-rank adaptation (LoRA) has become the standard method for parameter-efficient fine-tuning of large pretrained models, yet theoretical explanations for why low-rank updates suffice remain incomplete. Most existing analyses rely on the Neural Tangent Kernel (NTK) regime, which linearizes the network around the pretrained weights. In this work, we take a different approach that requires no linearization: our key observation is that, once the backbone is frozen, the loss sees each adapted weight matrix only through a finite-rank linear map induced by the frozen activations. Trace-norm regularization then forces the optimizer to pick a low-rank matrix among all matrices that produce the same observed output. This yields global minimizers whose ranks are controlled by the rank of this map, which we call the bottleneck rank, rather than the sample size. For multi-head attention, we obtain tighter rank upper bounds by exploiting two additional invariances: row-wise softmax is insensitive to additive row constants in the score matrix, and value updates are observable only after attention weighting and frozen output projection. For the factorized LoRA objective with weight decay, we prove a complementary result on the nonconvex side: after an arbitrarily small generic positive semidefinite perturbation, every first-order stationary point is rank-deficient once the LoRA rank exceeds a threshold determined by the bottleneck rank. Finally, we show that global minimizers of the perturbed objective are near-optimal for the original trace-norm problem, and we provide uniform generalization bounds governed by the spectral structure of the frozen features.
Figures & tables
| This work | Jang et al. (2024) | Kim et al. (2025) | |
|---|---|---|---|
| Requires NTK linearization | No | Yes | No |
| Multilayer multi-head attention bound | Yes (Thm. 4.4 ) | No | No |
| Spurious local minima excluded | No (Prop. 4.9 ) | Yes (NTK) | Partial |
| Additional loss assumptions | Differentiability and stagewise regularity | NTK-regime conditions | Restricted strong convexity and smoothness |
| Generalization bounds | Yes | Yes | No |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Type / Shape | Description |
| scalar | Input dimension | |
| scalar | Output dimension | |
| Pretrained (frozen) weight matrix | ||
| Adapted weight matrix ( ) | ||
| Low-rank update matrix | ||
| Left low-rank factor |
| Symbol | Type / Shape | Description |
|---|---|---|
| matrices | Token matrices for query, key, value (sample ) | |
| stacked matrices | for | |
| scalar | Query/key head dimension | |
| scalar | Value head dimension | |
| scalar | Output dims of projections ( ) | |
| matrix | Frozen projection for |
| LoRA fine-tuning | ||||
|---|---|---|---|---|
| Task/Dataset | MNLI | RTE | SST-2 | |
| Batch size | 8 | 8 | 8 | |
| Learning rate | Head | 5e-4 | 1e-4 | 1e-4 |
| Attention | 8e-5 | 1e-3 | 5e-4 | |
| LoRA regularization | 1e-4 | 1e-4 | 1e-4 | |
Low-Rank Adaptation
Low-rank adaptation (LoRA) fine-tunes pretrained models with compact weight updates, reducing trainable parameters and deployment costs.
See all 542 papers