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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Reasoning | ARC Easy | Accuracy86.96 | 233 | |
| Reasoning | OpenBookQA | Accuracy82.8 | 92 | |
| Commonsense Reasoning | BoolQ | Accuracy87.6 | 41 | |
| Reasoning | ARC Challenge | Accuracy (ACC)69.46 | 29 | |
| Reasoning | WinoGrande Small (WG-S) | Accuracy70.43 | 8 | |
| Reasoning | WinoGrande Medium | Accuracy0.7594 | 8 | |
| Natural Language Understanding | GLUE | CoLA Score88.2 | 3 |