让大模型同时学多个任务,不加结构还高效。
ThanoRA: Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation
- 按任务差异分配低秩空间,实现精准知识注入。
- 训练中保持任务多样性,避免干扰和退化。
- 比单独调优还强,适合实际部署的多任务场景。
低秩适应(LoRA)因高效且推理无开销,被广泛用于基础模型的下游微调。现实应用常需模型同时专精多个任务,推动高效多任务微调需求。现有方法主要分为两类:基于模型合并的LoRA在无训练场景有优势,但整体性能落后于多任务训练;基于MoE的LoRA虽提升性能,却引入路由器,阻碍参数合并并增加推理开销,限制实用性。为此,我们提出ThanoRA——一种任务异质性感知的多任务低秩适应框架,无需额外结构即可实现高效统一的多任务微调。ThanoRA通过初始化时依据任务间异质性动态分配子空间维度,并在训练中施加多样性保持正则,实现细粒度知识注入与抗干扰能力,充分挖掘低秩潜力。在跨模态与纯文本基准上的大量实验表明,ThanoRA在不同多任务组合下均持续优于强基线,甚至超越单独任务微调,且无额外结构或推理开销。代码将公开于:https://github.com/LiangJian24/ThanoRA。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) is widely adopted for downstream fine-tuning of foundation models due to its efficiency and zero additional inference cost. Many real-world applications require foundation models to specialize in several specific tasks simultaneously, motivating the need for efficient multi-task downstream adaptation. To address this need, existing studies have primarily explored two directions: Model Merging with LoRA, which shows advantages in training-free scenarios but still lags behind multi-task training in overall performance; and MoE-based LoRA approaches, which improve multi-task learning performance but introduce routers that hinder the mergeability of LoRA parameters and incur considerable inference overhead, thereby limiting real-world deployment practicality. To this end, we propose ThanoRA, a Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation framework that enables effective, efficient and unified multi-task downstream adaptation without introducing additional structure. ThanoRA performs multi-task learning by tailoring subspace allocation at initialization and enforcing diversity preservation throughout training: it allocates varying dimensions to construct task-specific low-rank subspaces driven by inter-task heterogeneity, enabling fine-grained knowledge injection, while diversity-preserving regularization mitigates task interference and subspace collapse, thereby fully exploiting the low-rank capacity. Extensive experiments across multimodal and text-only benchmarks under varying multi-task mixtures demonstrate that ThanoRA consistently outperforms strong baselines, surpassing even separate task-specific fine-tuning, while introducing no additional structures or inference overhead. Our code will be publicly available at: https://github.com/LiangJian24/ThanoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。