arXiv:2508.00522cs.CL2025-08

提出高效寻找平坦最小值的方法,提升大模型微调泛化能力。

Efficiently Seeking Flat Minima for Better Generalization in Fine-Tuning Large Language Models and Beyond

  • 通过理论证明参数扰动可转移至低秩子空间,避免多矩阵干扰。
  • 在GLUE和视觉语言任务上,性能优于LoRA和全量微调,提升达1.5%。
  • 适合追求高泛化能力的大模型微调研究者使用。

现有研究较少探讨低秩适应(LoRA)的表达能力与泛化能力之间的关联。尖锐感知最小化(SAM)通过引导模型收敛到局部平坦最小值,提升了卷积神经网络和Transformer的泛化能力。然而,由于缺乏有效工具来实证寻找平坦最小值或发展理论方法,该关联在LoRA中尚未充分探索。本文提出平坦最小值LoRA(FMLoRA)及其高效版本EFMLoRA,用于在LoRA中寻求平坦最小值。具体而言,我们从理论上证明了全参数空间中的扰动可被转移到低秩子空间,从而消除低秩子空间中多矩阵间扰动带来的潜在干扰。大规模实验在大语言模型和视觉语言模型上表明,EFMLoRA在效率上与LoRA相当,同时实现相当或更优的性能。例如,在RoBERTa-large模型的GLUE数据集上,EFMLoRA平均比LoRA和全量微调分别高出1.0%和0.5%;在视觉语言模型Qwen-VL-Chat上,于SQA和VizWiz数据集分别提升1.5%和1.0%。这些结果也验证了LoRA的泛化能力与尖锐度密切相关,而此前方法忽略了这一因素。

原文摘要 · Abstract (English)

Little research explores the correlation between the expressive ability and generalization ability of the low-rank adaptation (LoRA). Sharpness-Aware Minimization (SAM) improves model generalization for both Convolutional Neural Networks (CNNs) and Transformers by encouraging convergence to locally flat minima. However, the connection between sharpness and generalization has not been fully explored for LoRA due to the lack of tools to either empirically seek flat minima or develop theoretical methods. In this work, we propose Flat Minima LoRA (FMLoRA) and its efficient version, i.e., EFMLoRA, to seek flat minima for LoRA. Concretely, we theoretically demonstrate that perturbations in the full parameter space can be transferred to the low-rank subspace. This approach eliminates the potential interference introduced by perturbations across multiple matrices in the low-rank subspace. Our extensive experiments on large language models and vision-language models demonstrate that EFMLoRA achieves optimize efficiency comparable to that of LoRA while simultaneously attaining comparable or even better performance. For example, on the GLUE dataset with RoBERTa-large, EFMLoRA outperforms LoRA and full fine-tuning by 1.0% and 0.5% on average, respectively. On vision-language models, e.g., Qwen-VL-Chat, there are performance improvements of 1.5% and 1.0% on the SQA and VizWiz datasets, respectively. These empirical results also verify that the generalization of LoRA is closely related to sharpness, which is omitted by previous methods.

大模型微调泛化能力低秩适配优化方法

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