arXiv:2505.15471cs.CL2025-05ACL被引 9

提出CoLA框架,让低秩适配更高效,尤其适合数据少的任务。

CoLA: Collaborative Low-Rank Adaptation

  • 通过协同优化矩阵A和B的初始化,提升低秩适配灵活性。
  • 在少量样本下超越现有方法,性能提升显著,鲁棒性强。
  • 适合资源有限或多任务场景,代码开源可复现。

大语言模型的缩放定律揭示了模型规模增大时性能增益递减的幂律关系。尽管从头训练大模型成本高昂,但微调预训练模型已成为实用方案。全量微调(FFT)虽性能强,却计算开销大且效率低。参数高效微调(PEFT)方法如LoRA通过冻结预训练模型并添加轻量级任务模块来缓解问题。然而,传统LoRA在多任务场景中易受任务干扰。近期如专家混合(MOE)和非对称LoRA等方法虽有所改进,但仍受限于固定结构带来的样本稀缺与噪声干扰。为此,本文提出更灵活的LoRA架构CoLA,结合高效初始化策略,并引入三种协同机制以更好利用矩阵A与B之间的数量关系。实验表明,CoLA在多种设置下均优于现有方法,尤其在低样本场景中表现突出。相关数据与代码已公开于https://github.com/zyy-2001/CoLA。

原文摘要 · Abstract (English)

The scaling law of Large Language Models (LLMs) reveals a power-law relationship, showing diminishing return on performance as model scale increases. While training LLMs from scratch is resource-intensive, fine-tuning a pre-trained model for specific tasks has become a practical alternative. Full fine-tuning (FFT) achieves strong performance; however, it is computationally expensive and inefficient. Parameter-efficient fine-tuning (PEFT) methods, like LoRA, have been proposed to address these challenges by freezing the pre-trained model and adding lightweight task-specific modules. LoRA, in particular, has proven effective, but its application to multi-task scenarios is limited by interference between tasks. Recent approaches, such as Mixture-of-Experts (MOE) and asymmetric LoRA, have aimed to mitigate these issues but still struggle with sample scarcity and noise interference due to their fixed structure. In response, we propose CoLA, a more flexible LoRA architecture with an efficient initialization scheme, and introduces three collaborative strategies to enhance performance by better utilizing the quantitative relationships between matrices $A$ and $B$. Our experiments demonstrate the effectiveness and robustness of CoLA, outperforming existing PEFT methods, especially in low-sample scenarios. Our data and code are fully publicly available at https://github.com/zyy-2001/CoLA.

低秩适配参数高效多任务学习

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