arXiv:2412.09250cs.LGmath.GT2024-12EMNLP被引 8

根据隐藏层内在维度自适应调整LoRA秩,平衡模型表现与效率。

GeLoRA: Geometric Adaptive Ranks For Efficient LoRA Fine-tuning

  • 基于隐藏状态内在维度动态决定每层LoRA秩
  • 在相同参数预算下,性能优于现有基线方法
  • 为低秩微调的效率-表达力权衡提供理论依据

大语言模型微调计算成本高,因需更新全部参数。低秩适配(LoRA)通过仅修改部分权重提升效率,但存在表达力与计算成本的权衡:低秩降低资源消耗但限制表达能力,高秩增强表达力却增加开销。尽管已有自适应LoRA方法,但缺乏优化该权衡的理论基础。本文提出几何低秩适配(GeLoRA),通过计算隐藏状态表示的内在维度,自适应选择LoRA秩。实证表明,内在维度为最优LoRA矩阵秩的下界,可实现效率与表达力的合理平衡。GeLoRA根据各层输入输出表示的内在维度动态调整秩,认识到并非所有模型参数对微调同等重要。在多个任务上的实验验证显示,其在相同参数预算下持续优于近期基线方法。

原文摘要 · Abstract (English)

Fine-tuning large language models (LLMs) is computationally intensive because it requires updating all parameters. Low-Rank Adaptation (LoRA) improves efficiency by modifying only a subset of weights but introduces a trade-off between expressivity and computational cost: lower ranks reduce resources but limit expressiveness, while higher ranks enhance expressivity at increased cost. Despite recent advances in adaptive LoRA techniques, existing methods fail to provide a theoretical basis for optimizing the trade-off between model performance and efficiency. We propose Geometric Low-Rank Adaptation (GeLoRA), a novel framework that computes the intrinsic dimensionality of hidden state representations to adaptively select LoRA ranks. We demonstrate that the intrinsic dimension provides a lower bound for the optimal rank of LoRA matrices, allowing for a principled selection that balances efficiency and expressivity. GeLoRA dynamically adjusts the rank for each layer based on the intrinsic dimensionality of its input and output representations, recognizing that not all model parameters equally impact fine-tuning. Empirical validation on multiple tasks shows that GeLoRA consistently outperforms recent baselines within the same parameter budget.

低秩微调自适应大模型

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