arXiv:2507.08044cs.CVcs.AI2025-07ICCV被引 2

用数据驱动方法初始化LoRA权重,提升微调速度与效果

ConsNoTrainLoRA: Data-driven Weight Initialization of Low-rank Adapters using Constraints

  • 基于预训练与微调激活的约束,推导出无需训练的权重初值
  • 在图像生成、分类等任务上显著优于随机和现有数据驱动初始化
  • 支持不同位置设置可变秩,适合追求高效微调的研究者

基础模型在大规模数据上预训练后,通常使用参数高效微调(PEFT)技术如低秩适配器(LoRA)在小规模数据上微调。以往工作在所有连接点上均以固定秩随机初始化LoRA权重矩阵。本文提出数据驱动的权重初始化方法ConsNoTrainLoRA(CNTLoRA),将LoRA初始化建模为域偏移问题,利用预训练与微调激活间的多重约束,通过重构这些约束得到闭式解,该解依赖于预训练权重和微调激活向量,初始化时无需训练。该权重估计被分解用于初始化上下投影矩阵,并支持可变秩。在图像生成、图像分类和图像理解等下游任务上进行微调,定量与定性结果均表明CNTLoRA优于标准及现有数据驱动初始化方法。大量分析与消融实验进一步揭示了框架设计选择,提供更快收敛与更高性能的优化方案。

原文摘要 · Abstract (English)

Foundation models are pre-trained on large-scale datasets and subsequently fine-tuned on small-scale datasets using parameter-efficient fine-tuning (PEFT) techniques like low-rank adapters (LoRA). In most previous works, LoRA weight matrices are randomly initialized with a fixed rank across all attachment points. In this paper, we improve convergence and final performance of LoRA fine-tuning, using our proposed data-driven weight initialization method, ConsNoTrainLoRA (CNTLoRA). We express LoRA initialization as a domain shift problem where we use multiple constraints relating the pre-training and fine-tuning activations. By reformulating these constraints, we obtain a closed-form estimate of LoRA weights that depends on pre-training weights and fine-tuning activation vectors and hence requires no training during initialization. This weight estimate is decomposed to initialize the up and down matrices with proposed flexibility of variable ranks. With the proposed initialization method, we fine-tune on downstream tasks such as image generation, image classification and image understanding. Both quantitative and qualitative results demonstrate that CNTLoRA outperforms standard and data-driven weight initialization methods. Extensive analyses and ablations further elucidate the design choices of our framework, providing an optimal recipe for faster convergence and enhanced performance.

LoRA微调初始化

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