MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning
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 pretrained 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.
Authors
- Yunhai Tong (ORCID: https://orcid.org/0000-0001-8735-2516)
- Dilxat Muhtar
- Yaming Yang (ORCID: https://orcid.org/0000-0002-8830-1208)
- Weizhu Chen
- Yuefeng Zhan
- Weiwei Deng (ORCID: https://orcid.org/0000-0002-8331-9249)
- Qi Zhang (ORCID: https://orcid.org/0000-0001-5303-9804)
- Yujing Wang (ORCID: https://orcid.org/0000-0002-6334-245X)
- Yelong Shen
- Hao Sun (ORCID: https://orcid.org/0000-0002-1314-4957)
- Jianfeng Liu (ORCID: https://orcid.org/0000-0002-3054-4426)
- Feng Sun
Institutions
- Peking University (CN)
- Microsoft (Finland) (FI)
- Nanjing University (CN)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i20.35509
- Citations
- 7
- Primary Topic
- Domain Adaptation and Few-Shot Learning
- Type
- article
- Field-Weighted Citation Impact
- 6.21