GoRA动态调整低秩适配的秩和初始化,提升大模型微调效果。
GoRA: Gradient-driven Adaptive Low Rank Adaptation
- 基于梯度信息动态分配最优秩并自适应初始化权重
- 在数学推理任务中比标准LoRA高5.13分,逼近全量微调效果
- 首次统一秩选择与初始化,兼顾性能与效率,适合高效微调场景
低秩适配(LoRA)是高效微调大语言模型的关键方法,其效果受秩选择和权重初始化两大因素影响。尽管已有多种LoRA变体通过改进其中一个方面来提升性能,但往往牺牲可用性或计算效率。本文分析并识别了现有方法的核心局限,提出一种新框架GoRA(梯度驱动的自适应低秩适配),在统一框架内同时自适应调整秩和初始化策略。GoRA利用训练过程中的梯度信息,动态分配最优秩并自适应初始化低秩适配器权重。据我们所知,GoRA是首个不仅解决以往方法仅关注秩或初始化的局限,还能将两者统一于单一框架的方法,实现更有效且高效的模型适配。跨多种架构与模态的大量实验表明,GoRA持续优于现有基于LoRA的方法,同时保持原始LoRA的高效性。例如,在对Llama3.1-8B-Base进行数学推理微调时,GoRA相较标准LoRA提升5.13分,且在高秩设置下甚至超越全量微调2.05分。代码已公开:https://github.com/hhnqqq/MyTransformers。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) is a crucial method for efficiently fine-tuning large language models (LLMs), with its effectiveness influenced by two key factors: rank selection and weight initialization. While numerous LoRA variants have been proposed to improve performance by addressing one of these aspects, they often compromise usability or computational efficiency. In this paper, we analyze and identify the core limitations of existing approaches and propose a novel framework--GoRA (Gradient-driven Adaptive Low Rank Adaptation)--that simultaneously adapts both the rank and initialization strategy within a unified framework. GoRA leverages gradient information during training to dynamically assign optimal ranks and initialize low-rank adapter weights in an adaptive manner. To our knowledge, GoRA is the first method that not only addresses the limitations of prior approaches--which often focus on either rank selection or initialization in isolation--but also unifies both aspects within a single framework, enabling more effective and efficient adaptation. Extensive experiments across various architectures and modalities show that GoRA consistently outperforms existing LoRA-based methods while preserving the efficiency of vanilla LoRA. For example, when fine-tuning Llama3.1-8B-Base for mathematical reasoning, GoRA achieves a 5.13-point improvement over standard LoRA and even outperforms full fine-tuning by 2.05 points under high-rank settings. Code is available at: https://github.com/hhnqqq/MyTransformers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。