arXiv:2410.09437cs.LGcs.AI2024-10AAAI被引 33

改进LoRA,让多任务学习更高效且不互相干扰。

MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning

  • 为LoRA增加任务自适应参数,区分不同任务特征。
  • 在多个基准上表现优于原版LoRA,参数更少却效果更好。
  • 适合需要同时适配多个下游任务的轻量级模型优化场景。

参数高效微调(PEFT)广泛用于领域适应,其中LoRA因简单有效而备受关注。但在多任务学习(MTL)中,LoRA会将不同任务的稀疏高维特征投影到同一稠密低维内在空间,导致任务混淆和性能下降。为此,我们提出MTL-LoRA,保留低秩适配优势的同时显著提升多任务能力。该方法通过引入额外的任务自适应参数,区分任务特异性信息,并在低维空间中捕捉跨任务共享知识。这使得预训练模型能在少量可训练参数下联合适配多个目标领域。大量实验表明,包括自然语言理解、常识推理、图文理解等公开学术基准及真实工业文本广告相关数据集,MTL-LoRA在多任务设置下均优于LoRA及其变体,且参数量相当甚至更少。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) has been widely employed for domain adaptation, with LoRA being one of the most prominent methods due to its simplicity and effectiveness. However, in multi-task learning (MTL) scenarios, LoRA tends to obscure the distinction between tasks by projecting sparse high-dimensional features from different tasks into the same dense low-dimensional intrinsic space. This leads to task interference and suboptimal performance for LoRA and its variants. To tackle this challenge, we propose MTL-LoRA, which retains the advantages of low-rank adaptation while significantly enhancing MTL capabilities. MTL-LoRA augments LoRA by incorporating additional task-adaptive parameters that differentiate task-specific information and capture shared knowledge across various tasks within low-dimensional spaces. This approach enables pre-trained models to jointly adapt to different target domains with a limited number of trainable parameters. Comprehensive experimental results, including evaluations on public academic benchmarks for natural language understanding, commonsense reasoning, and image-text understanding, as well as real-world industrial text Ads relevance datasets, demonstrate that MTL-LoRA outperforms LoRA and its various variants with comparable or even fewer learnable parameters in MTL setting.

多任务学习LoRA参数高效微调

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