arXiv:2502.14538cs.CL2025-02EMNLP被引 11

LoRA-MGPO让低秩微调更稳定,解决训练震荡问题。

LoRA-MGPO: Mitigating Double Descent in Low-Rank Adaptation via Momentum-Guided Perturbation Optimization

  • 用动量引导扰动优化,避免梯度重复计算。
  • 实验显示收敛更快,损失曲线更平滑,泛化能力更强。
  • 适合追求高效微调稳定性的研究人员和工程应用。

参数高效微调(PEFT)中的低秩适应(LoRA)通过仅训练少量参数来调整大语言模型。然而,随着低秩矩阵秩的增加,LoRA常出现不稳定的“双重下降”现象,表现为训练损失暂时发散,延迟收敛并因趋向尖锐局部极小值而损害泛化性能。为此,我们提出LoRA-MGPO框架,引入动量引导扰动优化(MGPO),通过优化器状态中的动量向量指导权重扰动方向,避免双重梯度计算,从而稳定训练动态。此外,自适应归一化方案基于梯度范数的指数移动平均(EMA)控制扰动幅度,进一步提升稳定性。EMA调节扰动大小,MGPO引导扰动方向,共同实现更平稳的优化轨迹。在自然语言理解与生成多项基准测试中,LoRA-MGPO始终优于LoRA及其他PEFT方法。分析表明,该方法使损失曲线更平滑、收敛更快,并通过抑制对尖锐极小值的吸引,提升泛化能力。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT), particularly Low-Rank Adaptation (LoRA), adapts large language models (LLMs) by training only a small fraction of parameters. However, as the rank of the low-rank matrices used for adaptation increases, LoRA often exhibits an unstable "double descent" phenomenon, characterized by transient divergence in the training loss, which delays convergence and impairs generalization by causing instability due to the attraction to sharp local minima. To address this, we introduce LoRA-MGPO, a framework that incorporates Momentum-Guided Perturbation Optimization (MGPO). MGPO stabilizes training dynamics by mitigating the double descent phenomenon and guiding weight perturbations using momentum vectors from the optimizer's state, thus avoiding dual gradient computations. Additionally, an adaptive normalization scheme scales the magnitude of perturbations based on an exponential moving average (EMA) of gradient norms, further enhancing stability. While EMA controls the magnitude of the perturbations, MGPO guides their direction, ensuring a more stable optimization trajectory. Experiments on a suite of natural language understanding and generation benchmarks show that LoRA-MGPO consistently achieves superior performance over LoRA and other PEFT methods. The analysis indicates that LoRA-MGPO leads to smoother loss curves, faster convergence, and improved generalization by stabilizing the training process and mitigating the attraction to sharp minima.

低秩微调训练稳定PEFT优化算法

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