Spectral Bridge Variational Inference: Dynamic LoRA via Bures-Wasserstein Gradient Flows
Yuhang Xi, Yu-Feng Yu, Chuan-Xian Ren, Zhao-Rong Lai
Abstract
Parameter-Efficient Fine-Tuning (PEFT) is essential for adapting Large Language Models, yet existing methods struggle to balance capacity with computational efficiency. Standard approaches enforce rigid low-rank constraints, while dynamic alternatives incur significant memory overheads. To resolve this, we propose Spectral Bridge Variational Inference (SBVI), a geometric framework reformulating LoRA as a continuous Wasserstein gradient flow on the manifold of Gaussian measures. Instead of fixing ranks at initialization, SBVI governs singular value evolution via a stochastic differential equation driven by thermodynamic competition between task gradients and adaptive entropic friction. This induces a spectral bifurcation that automatically prunes noise modes while amplifying signal-rich components, discovering an optimal layer-wise rank distribution. We derive a scalable algorithm with linear complexity using factorized Riemannian retractions and Empirical Bayes friction updates. Experiments on reasoning and coding benchmarks show SBVI achieves state-of-the-art performance, offering superior accuracy and memory efficiency over existing static and dynamic methods. Our code is publicly available at: https://github.com/ xiyuhang2003/SBVI-LoRA
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 71f34fac-5a51-498c-b0f5-82b8319867b1Builds on20
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- Measuring Massive Multitask Language UnderstandingDan Hendrycks, Collin Burns, Steven Basart, Andy Zou et al.ICLR 2021 · 7,905 citations
- PIQA: Reasoning about Physical Commonsense in Natural LanguageYonatan Bisk, Rowan Zellers, Ronan Le Bras, Jianfeng Gao et al.AAAI 2020 · 2,916 citations
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank ProjectionJiawei Zhao, Zhenyu Zhang, Beidi Chen, Zhangyang Wang et al.ICML 2024 · 433 citations
- LoRA+: Efficient Low Rank Adaptation of Large ModelsSoufiane Hayou, Nikhil Ghosh, Bin YuICML 2024 · 388 citations
Related papers
- DoRA: Enhancing Parameter-Efficient Fine-Tuning with Dynamic Rank DistributionYulong Mao, Kaiyu Huang, Changhao Guan, Ganglin Bao et al.ACL 2024 · 15 citations
- FlexLoRA: Entropy-Guided Flexible Low-Rank AdaptationMuqing Liu, Chongjie Si, Yuheng JiaICLR 2026 · 3 citations
- Not All Directions Matter: Towards Structured and Task-Aware Low-Rank Model AdaptationXi Xiao, Chenrui Ma, Yunbei Zhang, Chen Liu et al.ACL 2026 · 6 citations
- COBRA: Contribution-Based Bayesian Rank Allocation for Parameter-Efficient Fine-TuningHongcheng Ding, Xuanze Zhao, LIU XUANHUANG, Jing Jin et al.ICML 2026
- Flat-LoRA: Low-Rank Adaptation over a Flat Loss LandscapeTao Li, Zhengbao He, Yujun Li, Yasheng Wang et al.ICML 2025
