Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning Adapter Rank via Symmetry Breaking

About

Low-rank adaptation is effective partly because downstream updates lie in a low-dimensional subspace, but the latent rank coordinates of LoRA are not identifiable: any invertible reparameterization of the adapter factors leaves the weight update unchanged. We show that variational inference with a diagonal rank-wise posterior turns this non-identifiability into a useful inductive bias. By breaking LoRA's rotational gauge symmetry, the variational objective selects a preferred basis in rank space, enabling automatic relevance determination over rank directions. This yields Low-Rank Variational Dropout (LRVD), a Bayesian framework that performs inference directly in the low-rank adaptation space rather than the ambient weight space. As an instantiation, BayesLoRA jointly learns effective adapter rank and predictive uncertainty with only $\mathcal{O}(r)$ additional parameters. Empirically, BayesLoRA induces stable rank structure aligned with the dominant singular directions of learned updates, yields compact predictive calibration and matches or exceeds strong low-rank sparsification baselines at comparable training cost.

Cooper Doyle, Andy Hu, Rebecca Chan, Anna Leontjeva• 2025

Related benchmarks

TaskDatasetResultRank
ReasoningARC Easy
Accuracy86.96
233
ReasoningOpenBookQA
Accuracy82.8
92
Commonsense ReasoningBoolQ
Accuracy87.6
41
ReasoningARC Challenge
Accuracy (ACC)69.46
29
ReasoningWinoGrande Small (WG-S)
Accuracy70.43
8
ReasoningWinoGrande Medium
Accuracy0.7594
8
Natural Language UnderstandingGLUE
CoLA Score88.2
3
Showing 7 of 7 rows

Other info

Follow for update