用数据敏感度指导LoRA初始化,提升微调效果
Learning in the Fisher Subspace: A Guided Initialization for LoRA Fine-Tuning
- 基于下游数据的费舍尔信息选择低秩更新方向
- 在多任务多模态上性能显著优于传统方法
- 适合需要高效微调大模型的研究者
LoRA通过限制预训练权重的低秩子空间更新来适应大语言模型,虽大幅降低训练成本,但效果高度依赖初始化时子空间的选择:若将容量分配给与任务无关的方向,会严重损害下游性能。现有初始化策略主要依赖预训练权重的内在特性,隐含假设权重几何能反映任务相关性,但忽略了模型与下游数据分布的交互。本文将LoRA初始化问题建模为评估参数空间方向在目标数据分布下的影响程度,认为应以数据感知的敏感性而非仅权重大小决定适配子空间。基于此,提出一种费舍尔引导框架,利用下游数据诱导的曲率信息刻画参数扰动对模型预测的影响,从而提供一种原则性、任务相关的方向选择准则。实验证明,该方法在多种任务和模态下均显著优于现有方法。
原文摘要 · Abstract (English)
LoRA adapts large language models (LLMs) by restricting updates to low-rank subspaces of pre-trained weights. While this substantially reduces training cost, the effectiveness of adaptation critically depends on which subspace is chosen at initialization: a poor initialization that allocates capacity to task-irrelevant directions can severely hinder downstream performance. Existing initialization strategies primarily rely on the intrinsic properties of pre-trained weights, implicitly assuming that weight geometry alone reflects task relevance. However, such criteria overlook how the model interacts with the downstream data distribution. In this work, we formulate LoRA initialization as identifying the degree of impact of directions in parameter space under the target data distribution. We argue that data-aware sensitivity, rather than weight-only magnitude, should govern the choice of adaptation subspaces. Building on this perspective, we propose a Fisher-guided framework that leverages curvature information induced by downstream data to characterize how parameter perturbations influence model predictions. This perspective yields a principled, task-dependent criterion for selecting LoRA directions that better align adaptation with the target objective. Empirical results across diverse tasks and modalities demonstrate that data-aware initialization consistently and significantly improves downstream performance over existing approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。