BiDoRA通过分层优化提升低秩适配效率,减少过拟合。
BiDoRA: Bi-level Optimization-Based Weight-Decomposed Low-Rank Adaptation
- 分两阶段异步优化权重的大小与方向,解耦更新过程。
- 在GLUE上显著优于DoRA,p值2.4×10⁻⁴,相关性达-8.042。
- 适合资源受限场景下高效微调大模型,尤其小数据任务。
参数高效微调(PEFT)是适应大语言模型(LLM)到下游任务的灵活高效方法。其中,权重分解低秩适配(DoRA)通过将权重矩阵分解为幅值和方向分量,更贴近全量微调(FT)。然而,DoRA同时优化这两个分量,导致模型过表达、易过拟合,并形成耦合更新模式,限制学习能力。为此,我们提出基于双层优化的权重分解低秩适配(BiDoRA),采用双层优化框架,在不同训练与验证数据子集上分步异步优化幅值与方向。该解耦机制有效缓解过拟合,支持更灵活的更新,更接近全量微调。例如,权重分解分析显示,BiDoRA的幅值-方向更新相关性为-8.042,显著优于DoRA的-1.784。在涵盖自然语言理解、生成、词元分类及极小生物医学数据集的多种任务上,BiDoRA持续超越DoRA及众多主流PEFT方法。该优势在GLUE基准测试中经威尔科克森符号秩检验验证,其显著性水平为2.4×10⁻⁴。代码已开源于https://github.com/t2ance/BiDoRA。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning (PEFT) is a flexible and efficient method for adapting large language models (LLMs) to downstream tasks. Among these methods, weight-decomposed low-rank adaptation (DoRA) is a promising approach that decomposes weight matrices into magnitude and direction components to mimic full fine-tuning (FT) better. However, DoRA's simultaneous optimization of these components makes it over-expressive, increases the risk of overfitting, and creates a coupled updating pattern that limits its learning capacity. To address these issues, we propose Bi-level Optimization-Based Weight-Decomposed Low-Rank Adaptation (BiDoRA), a novel PEFT method based on a bi-level optimization framework. BiDoRA fundamentally differs from DoRA by optimizing the magnitude and direction in two separate, asynchronous loops using distinct training and validation data splits. This decoupled optimization process effectively mitigates overfitting and allows for more flexible updates that align even more closely with FT. For instance, weight decomposition analysis shows BiDoRA achieves a magnitude-direction update correlation of $-8.042$, significantly closer to the FT ideal compared to $-1.784$ for DoRA. Evaluation of BiDoRA on diverse tasks spanning natural language understanding, generation, token classification, and extremely small biomedical datasets reveals that it consistently outperforms DoRA and a wide range of leading PEFT methods. This improvement is statistically significant, as demonstrated on the GLUE benchmark where BiDoRA surpasses DoRA with a p-value of $2.4\times10^{-4}$ in terms of the Wilcoxon signed-rank test. The code for BiDoRA is available at https://github.com/t2ance/BiDoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。