arXiv:2411.15231cs.LGcs.AI2024-11CVPR被引 13

提出迭代对齐方法,高效融合多任务LoRA而不依赖训练数据。

IterIS: Iterative Inference-Solving Alignment for LoRA Merging

  • 通过迭代推理-求解框架优化目标,缓解特征假设粗糙问题。
  • 仅需1%-5%无标签样本,大幅降低数据需求量。
  • 自适应加权机制平衡不同任务影响,适合多任务模型融合场景。

低秩适配(LoRA)被广泛用于在多个领域对大型模型进行微调以应对特定下游任务。尽管存在针对特定任务的LoRA,但数据隐私和知识产权问题常限制训练数据获取,阻碍了基于梯度训练的多任务模型构建。为此,LoRA融合提供了一种有效方案,在不泄露数据的前提下将多个LoRA合并为统一适配器。现有方法多将其视为优化问题,但存在输入特征假设粗略、需大量样本、优化目标失衡等局限,严重影响性能。本文提出新方法IterIS:1)将LoRA融合建模为更精细的优化问题,并引入迭代推理-求解框架,逐步优化目标以提升效果;2)设计高效正则化项,仅需1%-5%的无标签样本(相比先前方法显著减少);3)采用自适应权重调节优化目标,缓解融合过程中的不平衡问题。实验表明,该方法在文本到图像扩散模型、视觉语言模型及大语言模型的多任务组合上显著优于多个基线与先进方法。此外,其分层算法可在极少步骤内收敛,兼顾内存与计算效率。

原文摘要 · Abstract (English)

Low-rank adaptations (LoRA) are widely used to fine-tune large models across various domains for specific downstream tasks. While task-specific LoRAs are often available, concerns about data privacy and intellectual property can restrict access to training data, limiting the acquisition of a multi-task model through gradient-based training. In response, LoRA merging presents an effective solution by combining multiple LoRAs into a unified adapter while maintaining data privacy. Prior works on LoRA merging primarily frame it as an optimization problem, yet these approaches face several limitations, including the rough assumption about input features utilized in optimization, massive sample requirements, and the unbalanced optimization objective. These limitations can significantly degrade performance. To address these, we propose a novel optimization-based method, named IterIS: 1) We formulate LoRA merging as an advanced optimization problem to mitigate the rough assumption. Additionally, we employ an iterative inference-solving framework in our algorithm. It can progressively refine the optimization objective for improved performance. 2) We introduce an efficient regularization term to reduce the need for massive sample requirements (requiring only 1-5% of the unlabeled samples compared to prior methods). 3) We utilize adaptive weights in the optimization objective to mitigate potential unbalances in LoRA merging process. Our method demonstrates significant improvements over multiple baselines and state-of-the-art methods in composing tasks for text-to-image diffusion, vision-language models, and large language models. Furthermore, our layer-wise algorithm can achieve convergence with minimal steps, ensuring efficiency in both memory and computation.

LoRA融合多任务学习优化方法

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