LoRA-RG: Residual Gradient-Guided Rank Initialization and Expansion

Dayoung Kwon, Jaeseong Lee, Seung-Won Hwang


Abstract
We study how to optimally allocate ranks in Low-Rank Adaptation (LoRA). Despite its importance, ranks are typically chosen manually and fixed across all modules. We argue that this practice is suboptimal, risking either under-parameterization, which hurts performance, or over-parameterization, which wastes memory and computation. To address this, we propose LoRA-RG, a gradient-guided framework that adjusts LoRA ranks through two stages: rank initialization and rank expansion. In the rank initialization stage, LoRA-RG estimates a minimal rank for each layer based on gradient information prior to training. During training, the rank expansion stage further increases ranks only when the current LoRA parameterization is insufficient to capture the remaining update demand. This design dynamically aligns LoRA ranks with the underlying optimization needs. We evaluate LoRA-RG on reasoning, mathematical, and code generation benchmarks, and show that it consistently achieves higher accuracy compared to fixed-rank LoRA and recent rank initialization and adaptation baselines under the same parameter budget.
Anthology ID:
2026.emnlp-main.2000
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
45571–45587
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2000/
DOI:
Bibkey:
Cite (ACL):
Dayoung Kwon, Jaeseong Lee, and Seung-Won Hwang. 2026. LoRA-RG: Residual Gradient-Guided Rank Initialization and Expansion. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 45571–45587, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
LoRA-RG: Residual Gradient-Guided Rank Initialization and Expansion (Kwon et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2000.pdf