arXiv:2505.00009cs.CL2025-05中稿 · IEEE International…被引 4

用低秩表示提升多任务学习中知识迁移效率,兼顾性能与参数节省。

Efficient Knowledge Transfer in Multi-Task Learning through Task-Adaptive Low-Rank Representation

  • 基于提示调优,用低秩矩阵建模任务特异性知识。
  • 在16个任务上达到顶尖效果,少样本场景下优势明显。
  • 适合追求高效微调的NLP研究者和工业应用开发者。

预训练语言模型(PLMs)虽具强大智能,但在现实应用中面对未见新任务时表现不佳。为应对此问题,多任务学习(MTL)通过从源任务向目标任务迁移共享知识来解决。作为主流的参数高效微调方法,提示调优(PT)引入可适配向量作为前缀,保留原始提示中的共享知识,同时冻结模型参数。然而,PT因表征能力有限,难以有效捕捉任务间的差异性。为此,本文提出任务自适应低秩表示(TA-LoRA),在PT基础上构建,利用低秩表示建模任务异质性,并采用快慢权重机制:慢权重编码共享知识,快权重捕获任务特异性信息,避免从零训练低秩表示导致的共享与特异知识混杂。此外,引入零初始化注意力机制,在预热阶段最小化不成熟低秩组件对原始提示的干扰。在16项任务上的实验表明,TA-LoRA在全数据与少样本设置下均达到当前最优性能,且保持卓越的参数效率。

原文摘要 · Abstract (English)

Pre-trained language models (PLMs) demonstrate remarkable intelligence but struggle with emerging tasks unseen during training in real-world applications. Training separate models for each new task is usually impractical. Multi-task learning (MTL) addresses this challenge by transferring shared knowledge from source tasks to target tasks. As an dominant parameter-efficient fine-tuning method, prompt tuning (PT) enhances MTL by introducing an adaptable vector that captures task-specific knowledge, which acts as a prefix to the original prompt that preserves shared knowledge, while keeping PLM parameters frozen. However, PT struggles to effectively capture the heterogeneity of task-specific knowledge due to its limited representational capacity. To address this challenge, we propose Task-Adaptive Low-Rank Representation (TA-LoRA), an MTL method built on PT, employing the low-rank representation to model task heterogeneity and a fast-slow weights mechanism where the slow weight encodes shared knowledge, while the fast weight captures task-specific nuances, avoiding the mixing of shared and task-specific knowledge, caused by training low-rank representations from scratch. Moreover, a zero-initialized attention mechanism is introduced to minimize the disruption of immature low-rank components on original prompts during warm-up epochs. Experiments on 16 tasks demonstrate that TA-LoRA achieves state-of-the-art performance in full-data and few-shot settings while maintaining superior parameter efficiency.

多任务学习低秩表示提示调优参数效率

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