arXiv:2503.04046cs.LGcs.AI2025-03NeurIPS被引 4

用对称跃迁优化多任务学习中的冲突与不平衡问题。

Continual Optimization with Symmetry Teleportation for Multi-Task Learning

  • 通过低秩适配器在损失曲面中寻找等效点,缓解优化冲突。
  • 在多个主流数据集上显著提升现有多任务学习方法性能。
  • 无需修改原模型,可即插即用,适合各类多任务场景。

多任务学习(MTL)通过单个模型同时学习多个任务,但优化冲突和任务不平衡问题仍未有效解决。现有方法多依赖重加权任务损失或梯度来缓解冲突,而本文提出一种基于连续优化与对称跃迁(COST)的新方法。当优化冲突发生时,COST通过低秩适配器(LoRA)在损失曲面上寻找损失值不变的替代点,实现高效跃迁。同时引入历史轨迹复用策略,持续利用先进优化器的优势。在多个主流数据集上的实验表明,COST能显著提升多种现有MTL方法的表现,且为即插即用方案,兼容性强。

原文摘要 · Abstract (English)

Multi-task learning (MTL) is a widely explored paradigm that enables the simultaneous learning of multiple tasks using a single model. Despite numerous solutions, the key issues of optimization conflict and task imbalance remain under-addressed, limiting performance. Unlike existing optimization-based approaches that typically reweight task losses or gradients to mitigate conflicts or promote progress, we propose a novel approach based on Continual Optimization with Symmetry Teleportation (COST). During MTL optimization, when an optimization conflict arises, we seek an alternative loss-equivalent point on the loss landscape to reduce conflict. Specifically, we utilize a low-rank adapter (LoRA) to facilitate this practical teleportation by designing convergent, loss-invariant objectives. Additionally, we introduce a historical trajectory reuse strategy to continually leverage the benefits of advanced optimizers. Extensive experiments on multiple mainstream datasets demonstrate the effectiveness of our approach. COST is a plug-and-play solution that enhances a wide range of existing MTL methods. When integrated with state-of-the-art methods, COST achieves superior performance.

多任务学习优化方法低秩适配

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