arXiv:2605.08734cs.LGcs.AI2026-05

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.

LoRA优化器大模型微调低显存

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。