Mechanism

Freezes pretrained weight matrices W0∈Rdout×dinW_0 \in \mathbb{R}^{d_{\text{out}}\times d_{\text{in}}} and injects trainable low-rank matrices B∈Rdout×r,A∈Rr×dinB \in \mathbb{R}^{d_{\text{out}}\times r}, A \in \mathbb{R}^{r \times d_{\text{in}}} (r≪min⁡(din,dout)r \ll \min(d_{\text{in}}, d_{\text{out}})): W=W0+γrBAW = W_0 + \gamma_r B A where standard LoRA sets γr=αr\gamma_r = \frac{\alpha}{r} (which causes gradient norm collapse as rr increases), while rsLoRA sets γr=αr\gamma_r = \frac{\alpha}{\sqrt{r}} so activation and gradient variances remain invariant across high ranks r∈{64,128,256}r \in \{64, 128, 256\}.

References

  1. LoRA: Low-Rank Adaptation of Large Language Models ↗

    Edward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang et al. · 2021

    All 8 authors

    Edward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, Weizhu Chen

  2. A Rank Stabilization Scaling Factor for Fine-Tuning with LoRA ↗

    Damjan Kalajdzievski · 2023