Mechanism

  • PiSSA: Instead of initializing A∼N(0,σ2),B=0A \sim \mathcal{N}(0,\sigma^2), B=0 like LoRA, performs SVD on W0=USV⊤W_0 = U S V^\top, initializes the trainable adapter (B,A)(B, A) with the top-rr principal singular values and vectors (which carry the steepest task gradients), and freezes the residual Wres=W0−BinitAinitW_{\text{res}} = W_0 - B_{\text{init}}A_{\text{init}}.
    • GaLore: Performs full-rank weight updates while projecting the gradient matrix Gt∈Rm×nG_t \in \mathbb{R}^{m\times n} into a slowly changing low-rank subspace Pt⊤GtQtP_t^\top G_t Q_t to cut AdamW optimizer state memory by up to 65%.

References

  1. PiSSA: Principal Singular Values and Singular Vectors Adaptation of Large Language Models ↗

    Fanxu Meng, Zhaohui Wang, Muhan Zhang · 2024

  2. GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection ↗

    Jiawei Zhao, Zhenyu Zhang, Beidi Chen, Zhangyang Wang, Anima Anandkumar, Yuandong Tian · 2024