AdaPreLoRA改进了低秩适配的优化器,让训练更稳定高效。
AdaPreLoRA: Adafactor Preconditioned Low-Rank Adaptation
- 用Adafactor预条件矩阵结合闭式解,解决传统LoRA优化的奇异问题。
- 在多个大模型任务中表现优于或媲美现有方法,内存占用仅相当于普通LoRA。
- 适合追求高效率与低显存的模型微调用户,尤其适用于大模型场景。
低秩适配(LoRA)将权重更新表示为两个低秩因子的乘积,但生成器映射的雅可比矩阵 $J_{G}$ 秩不足,导致任何 $W$-空间预条件器 ${F}_t$ 引出的因子空间预条件器 $J_{G}^* {F}_t J_{G}$ 均为奇异矩阵,标准链式法则无法唯一反推预条件后的 $W$-空间方向回因子空间更新。本文将现有 LoRA 优化器统一为由两个选择构成的框架:(i) 使用何种可逆代理替代 $J_{G}^* {F}_t J_{G}$,(ii) 在 $W$ 空间使用何种 ${F}_t$。现有方法分布在四个家族中:因子空间自适应更新、$J_{G}^* J_{G}$ 的块对角代理、弗罗贝尼乌斯残差伪逆方法以及黎曼流形约束。在此设计空间中,一种基于梯度统计的 ${F}_t$ 配合 $O((m+n)r)$ 内存开销的闭式因子空间求解方案尚未被充分探索。本文提出 extbf{AdaPreLoRA},采用 $W$ 空间的 Adafactor 对角克罗内克预条件器 ${H}_t$,并在由此产生的因子空间解族中选取使两个因子贡献在 ${H}_t$-加权下失衡最小的解;构造上,该因子更新是 ${H}_t$-加权范数下最接近预条件 $W$-空间方向的 LoRA 近似。在 GPT-2(E2E)、Mistral-7B 与 Qwen2-7B(GLUE, ARC, GSM8K)及扩散模型个性化任务中,AdaPreLoRA 性能与现有代表性优化器相当或更优,且峰值显存保持在 LoRA 优化器水平。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) reparameterizes a weight update as a product of two low-rank factors, but the Jacobian $J_{G}$ of the generator mapping the factors to the weight matrix is rank-deficient, so the factor-space preconditioner $J_{G}^* {F}_t J_{G}$ induced by any ${W}$-space preconditioner ${F}_t$ is singular, and consequently the standard chain rule cannot be uniquely inverted to map a preconditioned ${W}$-space direction back to a factor-space update. We cast existing LoRA optimizers in a unified framework parameterized by two choices: (i) which invertible surrogate for $J_{G}^* {F}_t J_{G}$ to use, and (ii) which ${F}_t$ on ${W}$ to use. Existing methods occupy four families along these axes: factor-space adaptive updates, block-diagonal surrogates for $J_{G}^* J_{G}$, Frobenius-residual pseudoinverse methods, and Riemannian manifold constraint. Within this design space, a gradient-statistics-aware ${F}_t$ paired with a closed-form factor-space solve at ${O}((m+n)r)$ memory remains underexplored. We propose \textbf{AdaPreLoRA}, which fills this gap by adopting the Adafactor diagonal Kronecker preconditioner ${H}_t$ on ${W}$ and selecting from the resulting factor-space solution family the element minimizing an ${H}_t$-weighted imbalance between the two factor contributions; by construction, the resulting factor update is the closest LoRA approximation to the preconditioned ${W}$-space direction under the ${H}_t$-weighted norm. Across GPT-2 (E2E), Mistral-7B and Qwen2-7B (GLUE, ARC, GSM8K), and diffusion-model personalization, AdaPreLoRA is competitive with or improves over a representative set of LoRA optimizers while keeping peak GPU memory at the LoRA optimizer level.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。