Adaptive Rank Fine-Tuning with LR-LoRA
- The paper introduces LR-LoRA, a method that shifts from fixed global rank to learnable, layer-specific adaptation capacity in transformers.
- It employs a differentiable transfer function with sinc-basis parameterization to adjust low-rank updates, effectively capturing per-layer spectral complexity.
- Empirical results show that LR-LoRA improves performance over standard LoRA with negligible parameter and runtime overhead, demonstrating practical efficiency.
This paper, Parameter-Efficient Fine-Tuning with Learnable Rank, is highly relevant to the idea of Locally Adaptive Ranking (LAR), even though it does not use that exact term. Its central claim is that the rank constraint in standard LoRA should not be treated as a fixed, globally chosen hyperparameter. Instead, the model should be allowed to learn the effective adaptation dimensionality separately across layers and modules during fine-tuning. In that sense, the paper is best understood as proposing a layer-local, module-specific, continuously learned form of rank allocation.
Standard LoRA adapts a frozen pretrained weight matrix by adding a low-rank update: where , , and . This yields the familiar parameter count , but it also imposes a strong inductive bias: The paper argues that this fixed-rank inductive bias is likely suboptimal because transformer layers differ substantially in function, depth, representation geometry, and intrinsic dimensionality. A single global , usually applied uniformly across all adapted layers, implicitly assumes that all layers require the same adaptation capacity. The authors explicitly challenge this assumption and frame their contribution as a shift from “externally prescribing” adaptation dimensionality to learning it during optimization.
That is the key connection to LAR. In this paper, “local” means per adapted layer / per adapter instance, and more specifically per adapted projection matrix, because each adapter gets its own learned transfer function. It is not a single global rank, and it is not only per block unless multiple projections are tied manually in an ablation. The learned notion of rank is therefore module-local: attention and MLP projections in different depths can realize different effective ranks.
The proposed method, Learnable Rank LoRA (LR-LoRA), modifies LoRA by applying an elementwise learnable nonlinearity to the low-rank product: so that the adapted weight becomes
0
The core insight is that while 1 is rank-constrained, 2 need not be. The paper’s claim is broad: any differentiable elementwise nonlinearity applied to 3 removes the strict rank-4 constraint. LR-LoRA instantiates this idea using a sinc-basis parameterization: 5 where 6 are uniformly spaced grid points on 7, and both amplitudes 8 and bandwidths 9 are trainable. The basis is motivated via sampling theory, but the authors repeatedly stress that their scientific point is basis-agnostic: the important thing is not sinc per se, but that a learnable differentiable transfer function lets the optimizer discover the layer’s needed adaptation complexity.
There is no explicit discrete rank-selection vector, no hard mask over singular directions, no thresholding rule, and no sparsity penalty that directly zeros components to produce a final integer rank. This is important for interpreting the method relative to adaptive-rank PEFT. LR-LoRA does not define something like a learnable vector 0 or soft gates over latent rank components, nor does it solve a constrained rank-budget allocation problem as in AdaLoRA-style methods. Instead, rank becomes a realized property of the update matrix 1, measured empirically through its spectrum. The paper uses stable rank rather than exact rank: 2 with 3 the Frobenius norm and 4 the spectral norm. The authors explicitly avoid exact rank because it is numerically threshold-dependent. Thus, “learnable rank” in LR-LoRA really means learnable effective spectral complexity, proxied by stable rank, not a directly optimized integer rank.
Methodologically, the model remains fully differentiable and trainable end-to-end because 5 is differentiable with respect to its parameters and with respect to its input 6. Each adapter has its own 7, so every adapted module learns separate transfer-function parameters. The initialization is chosen carefully: amplitudes are initialized to zero,
8
so that initially
9
and the model starts from the frozen backbone with no effective adapter perturbation. Bandwidths are parameterized via
0
with 1 initialized so that 2 at the start; the default is 3. This ensures 4 and avoids degeneracy. The paper even explains why zero-initialized 5 does not kill learning: since
6
the amplitudes receive nonzero gradients immediately, and once 7 moves away from zero, gradients propagate through 8 into 9 and 0.
The paper does not introduce any additional regularizer on rank, sparsity, or budget. There is no explicit penalty such as nuclear norm, 1 sparsity over rank gates, or KL term on capacity. The training objective is therefore just the ordinary downstream task loss optimized over the adapter factors and transfer-function parameters, with AdamW. The paper does not write the supervised task objective as an explicit formula, but it states that optimization schedules, preprocessing, and evaluation scripts are inherited from the RandLoRA codebase and kept identical across methods. So from the paper alone, the objective can only be summarized as the standard fine-tuning loss over the trainable parameter set 2, with no added rank regularization term reported.
This absence of an explicit regularizer matters for the LAR interpretation. Many adaptive-rank methods allocate rank under a fixed global budget and therefore require explicit scoring, pruning, or budget redistribution. LR-LoRA instead leaves the nominal base factorization rank 3 fixed as an outer cap on the latent factors 4, but the effective rank of the realized update can be smaller or larger than the nominal 5 after the nonlinear mapping. The paper says this directly: “The learned adaptation dimensionality of each layer can therefore be smaller or larger than the nominal 6.” So the model is adaptive without a hard budget-allocation optimization problem.
Empirically, the method strongly supports the local-adaptivity story. The paper repeatedly shows that different layers learn different stable ranks, and that attention and MLP modules follow systematically different patterns. In the main text, the stable-rank profiles plotted across depth for Qwen2-0.5B at 7 show that update complexity is depth-dependent and differs between MLP and attention projections. In the quantitative stable-rank statistics for Qwen2-0.5B at 8, the mean 9 per-layer standard deviation across adapted modules is:
- LoRA: attention 0, MLP 1
- SineLoRA: attention 2, MLP 3
- RandLoRA: attention 4, MLP 5
- LR-LoRA: attention 6, MLP 7
These numbers are central for mapping to LAR. The mean stable rank is higher for LR-LoRA, but more importantly the variance across layers is much larger. The authors argue that SineLoRA and RandLoRA mostly raise expressivity more uniformly, while LR-LoRA induces heterogeneous per-layer rank allocation. In other words, the distinguishing property is not merely “higher rank” but locally differentiated rank.
The paper also provides direct functional evidence that this locality matters. In a parameter-sharing ablation, progressively tying the sinc parameters across layers degrades performance:
- full per-layer learning: 8
- share 9: 0
- share 1: 2
- share 3: 4
- share all: 5
When all transfer-function parameters are tied across layers—so the same nonlinearity is used everywhere—accuracy falls back to the LoRA level. This is one of the clearest pieces of evidence that the benefit is genuinely local, layer-specific adaptivity, not simply “nonlinearity added to LoRA.”
The paper further distinguishes between module types. Restricting LR-LoRA to only attention or only MLP hurts performance relative to adapting both. In the main text, attention-only or MLP-only LR-LoRA loses 6 and 7 points respectively versus full coverage; relative to matched LoRA at the same placement, the gains are 8 for attention and 9 for MLP, versus 0 combined. This superadditivity suggests that attention and MLP layers have distinct capacity requirements that are not well captured by a single uniform rank. The stable-rank table also indicates that MLP adapters tend to realize somewhat higher stable ranks than attention adapters under LR-LoRA (1 vs 2 mean stable rank on Qwen2-0.5B), with both showing much larger variation than baselines.
As implemented, rank learning is independent for each adapted adapter matrix. Adapter placement covers attention 3 and MLP 4 projections. Each instantiated adapter gets its own 5, so the granularity is per projection matrix per layer, not merely per transformer block. The transfer-parameter count is
6
where 7 is the set of adapted modules and each 8 contributes 9 scalars 0. With the default 1, the overhead is tiny: for example, Qwen2-0.5B goes from 2M LoRA parameters to 3M LR-LoRA parameters (+0.95%), Phi-3-8B from 4M to 5M (+0.15%), LLaMA3-8B from 6M to 7M (+0.13%), and RoBERTa-base from 8M to 9M (+0.70%).
In optimization terms, the defaults are stable and simple. For LLM fine-tuning the paper uses AdamW, 3 epochs, learning rate 0, adapter dropout 1, and batch sizes 16 / 8 / 4 for Qwen2-0.5B / Phi-3-8B / LLaMA3-8B. The LR-LoRA-specific defaults are: 2 The paper reports robustness: jointly perturbing 3 by 4 changes accuracy by at most 5 points. Grid resolution saturates quickly: 6 yields 7 in the ablation harness. Learning bandwidths matters: fixing 8 while learning only 9 gives only 00 over LoRA, versus 01 for full LR-LoRA, so learned 02 accounts for about 36% of the total gain. Near-zero amplitude initialization is more stable than random initialization.
There is no final discrete rank extraction step, because the method never converts the learned update into an explicit integer-rank choice. The final deployed adapter is just the merged matrix
03
precomputed once and folded into 04 at inference. Consequently, inference has zero overhead. The paper is explicit that 05 is precomputed and merged into 06, so runtime is identical to the backbone.
Conceptually, the paper positions LR-LoRA as a better inductive bias because it treats adaptation dimensionality as something that should emerge from optimization rather than be imposed a priori. This can be interpreted in several ways relevant to LAR: as adaptive subspace learning, because the low-rank factors still define a latent basis but the nonlinearity reshapes the realized update subspace; as soft rank selection, because the effective spectrum is learned continuously rather than by discrete pruning; and as a form of implicit model selection, because different layers settle on different update complexities under the same training loss. The paper explicitly compares against adaptive-rank PEFT methods such as AdaLoRA, DyLoRA, and ElaLoRA, noting that those methods still rely on a user-specified global capacity budget that is redistributed during training. LR-LoRA differs by not reallocating from a fixed total rank budget; instead, it lets each module’s effective stable rank arise from its own learned transfer function.
The experimental evidence is extensive. The evaluation spans 7 architectures from 125M to 13B parameters, 19 tasks, and four paradigms: commonsense reasoning, GLUE transfer, vision transfer, and instruction tuning. The main language-model backbones are Qwen2-0.5B, Phi-3-8B, and LLaMA3-8B under 15k and 170k training-example regimes. Commonsense tasks are BoolQ, PIQA, SocialIQA, HellaSwag, WinoGrande, ARC-Easy, ARC-Challenge, and OpenBookQA. GLUE uses RoBERTa-base and RoBERTa-large on SST-2, MRPC, CoLA, QNLI, RTE, and STS-B. Vision experiments use CLIP ViT-B/32 and DINOv2 ViT-B/14. Instruction tuning fine-tunes on Stanford Alpaca and evaluates with MT-Bench.
The main commonsense table shows LR-LoRA is best in all six backbone/regime settings at 07:
- Qwen2-0.5B: 55.88 (15k), 61.05 (170k)
- Phi-3-8B: 83.90 (15k), 87.30 (170k)
- LLaMA3-8B: 85.99 (15k), 88.22 (170k)
Relative to LoRA, gains are:
- Qwen2: 08 and 09
- Phi-3: 10 and 11
- LLaMA3: 12 and 13
Relative to RandLoRA, gains range from 14 to 15. LR-LoRA also beats AdaLoRA, DyLoRA, and ElaLoRA in all entries of that table. On GLUE, LR-LoRA-4 reaches average 84.5 on RoBERTa-base and 88.3 on RoBERTa-large, exceeding the strongest baseline by 16 and 17 points respectively. On CoLA the gains are especially large: 18 on base and 19 on large relative to the strongest non-LR-LoRA baseline. On MT-Bench, LR-LoRA scores 4.61 vs LoRA 4.19 on Qwen2-0.5B and 6.55 vs LoRA 6.12 on LLaMA3-8B. On CLIP transfer, the average rises from 81.1 (LoRA) to 82.3; on DINOv2, from 82.0 to 83.2.
A particularly LAR-relevant ablation is the interaction with base rank 20. On Qwen2-0.5B (170k), LR-LoRA’s absolute gain over LoRA decreases as 21 grows: 22 The authors interpret this monotonic pattern as evidence that LR-LoRA is most useful when the fixed-rank bottleneck is binding. That is exactly the regime where adaptive rank allocation should matter most.
So, can LR-LoRA be viewed as an instance of Locally Adaptive Ranking? Yes—conceptually, strongly yes. The mapping is:
- local: each adapted module/layer has its own learned transfer function 23, and empirical effective rank varies across depth and between module types;
- adaptive: these transfer functions are learned jointly during training, not preset by heuristic or manually assigned;
- ranking/rank: the method controls low-rank adaptation capacity by changing the realized spectral complexity of 24, measured by stable rank.
But there is also an important limitation to this mapping. LR-LoRA does not explicitly learn a discrete rank per layer in the way some LAR formulations might imply. There is no integer 25 selected for layer 26, no mask over rank components, and no explicit budgeted redistribution of rank units across layers. The nominal LoRA factor rank 27 is still chosen by the user, and what is learned is the effective stable rank of the final update, which can exceed or fall below 28. Thus, LR-LoRA is best described as a continuous, implicit, per-layer adaptive-rank method, not a discrete rank-allocation method.
Practically, a practitioner should prefer LR-LoRA over fixed-rank LoRA when they suspect that different layers need different adaptation capacities, especially under tight rank budgets. The strongest gains appear at smaller base ranks, on harder reasoning tasks, and across heterogeneous architectures. The trade-offs are favorable: additional parameters are tiny (29 over LoRA), peak memory rises by at most about 30, and throughput drops by at most about 31–32 in reported settings. Inference cost is unchanged after merging. The main complexity cost is implementation: one must support per-adapter transfer functions and the merge-once-per-step training strategy. The paper also notes that LR-LoRA breaks some of LoRA’s backward-pass associativity advantages, because gradients through 33 require materializing dense 34 terms during backprop, though empirically this overhead stays small.
As for caveats, the paper does not provide formal approximation or optimization guarantees for the learned-rank behavior; its evidence is empirical. It also does not provide a discrete rank readout, a direct rank penalty, or a principled stopping criterion for capacity. If someone’s notion of LAR specifically requires explicit per-layer integer-rank assignment under a global budget, LR-LoRA only partially matches that definition. If, however, LAR is understood more broadly as layer-wise adaptive allocation of low-rank adaptation capacity, then LR-LoRA is a clear and strong example of that idea.