arXiv:2410.07170cs.LGcs.AI2024-10NeurIPS被引 21

EVA通过最大化梯度信号,让小参数微调更快更准。

Parameter Efficient Fine-tuning via Explained Variance Adaptation

  • 基于激活方差选择最优方向初始化,提升梯度信号
  • 在多任务中收敛更快,平均性能超越现有方法
  • 支持自适应秩,减少可训练参数量,适合资源受限场景

基础模型(FMs)在大规模数据集上预训练后,需针对下游任务进行微调。当前主流方法为低秩适配(LoRA),其初始化常依赖梯度或权重矩阵的奇异值分解(SVD)。然而,这些方法无法保证最大化期望梯度信号,影响快速适应。为此,本文提出解释方差适配(EVA),通过在小批量激活向量上进行增量SVD,选取收敛后的右奇异向量作为初始化方向,可证明地最大化期望梯度信号,加速微调。EVA根据给定秩预算选择捕捉最大激活方差的方向,支持自适应秩,实现参数量再分配。我们在语言生成与理解、图像分类及强化学习等任务上应用EVA,结果表明其收敛速度优于对比方法,并在各领域任务中取得最高平均得分,同时通过秩重分配减少了可训练参数数量。总体而言,EVA在精度与效率上建立了新的帕累托前沿。

原文摘要 · Abstract (English)

Foundation models (FMs) are pre-trained on large-scale datasets and then fine-tuned for a specific downstream task. The most common fine-tuning method is to update pretrained weights via low-rank adaptation (LoRA). Existing initialization strategies for LoRA often rely on singular value decompositions (SVD) of gradients or weight matrices. However, they do not provably maximize the expected gradient signal, which is critical for fast adaptation. To this end, we introduce Explained Variance Adaptation (EVA), an initialization scheme that uses the directions capturing the most activation variance, provably maximizing the expected gradient signal and accelerating fine-tuning. EVA performs incremental SVD on minibatches of activation vectors and selects the right-singular vectors for initialization once they converged. Further, by selecting the directions that capture the most activation-variance for a given rank budget, EVA accommodates adaptive ranks that reduce the number of trainable parameters. We apply EVA to a variety of fine-tuning tasks as language generation and understanding, image classification, and reinforcement learning. EVA exhibits faster convergence than competitors and achieves the highest average score across a multitude of tasks per domain while reducing the number of trainable parameters through rank redistribution. In summary, EVA establishes a new Pareto frontier compared to existing LoRA initialization schemes in both accuracy and efficiency.

微调LoRA高效训练初始化

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