arXiv:2508.19564cs.LGcs.AI2025-08被引 1

让微调大模型更稳定,用双模块设计提升泛化能力

Bi-LoRA: Efficient Sharpness-Aware Minimization for Fine-Tuning Large-Scale Models

  • 双模块LoRA结构分离任务适配与梯度扰动
  • 在不增加内存的前提下实现平缓极小值优化
  • 适合资源有限但需强泛化的模型微调场景

小样本微调大模型面临泛化挑战。虽然尖锐感知最小化(SAM)通过寻找平坦极小值能提升泛化性能,但其高昂的内存和计算开销使其难以应用于大模型。将SAM与低秩适应(LoRA)结合虽有前景,但直接对LoRA参数应用SAM会限制扰动空间,削弱效果。为此,我们提出双向低秩适应(Bi-LoRA),引入辅助LoRA模块以建模SAM的对抗性权重扰动。主模块通过标准梯度下降适配任务,辅模块通过梯度上升捕捉损失曲面的尖锐性。该双模块设计使Bi-LoRA能探索更广的平坦区域,同时保持内存效率。此外,优化与扰动可并行进行,避免了SAM的双重训练成本。跨多种任务与架构的大量实验表明,Bi-LoRA在提升泛化性能方面兼具高效性与有效性。

原文摘要 · Abstract (English)

Fine-tuning large-scale pre-trained models with limited data presents significant challenges for generalization. While Sharpness-Aware Minimization (SAM) has proven effective in improving generalization by seeking flat minima, its substantial extra memory and computation overhead make it impractical for large models. Integrating SAM with parameter-efficient fine-tuning methods like Low-Rank Adaptation (LoRA) is a promising direction. However, we find that directly applying SAM to LoRA parameters limits the sharpness optimization to a restricted subspace, hindering its effectiveness. To address this limitation, we propose Bi-directional Low-Rank Adaptation (Bi-LoRA), which introduces an auxiliary LoRA module to model SAM's adversarial weight perturbations. It decouples SAM's weight perturbations from LoRA optimization: the primary LoRA module adapts to specific tasks via standard gradient descent, while the auxiliary module captures the sharpness of the loss landscape through gradient ascent. Such dual-module design enables Bi-LoRA to capture broader sharpness for achieving flatter minima while remaining memory-efficient. Another important benefit is that the dual design allows for simultaneous optimization and perturbation, eliminating SAM's doubled training costs. Extensive experiments across diverse tasks and architectures demonstrate Bi-LoRA's efficiency and effectiveness in enhancing generalization.

模型微调LoRA泛化提升高效优化

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