通过稀疏训练缓解多任务学习中的梯度冲突,提升模型整体性能。
Proactive Gradient Conflict Mitigation in Multi-Task Learning: A Sparse Training Perspective
- 仅更新部分参数的稀疏训练策略减少梯度冲突
- 实验显示该方法显著提升多任务模型表现
- 可与现有梯度调节技术结合,适合复杂多任务场景
构建通用智能体需要统一模型同时处理多个下游任务,这使得多任务学习的重要性日益凸显。然而,多任务学习中常见的梯度冲突会导致不同任务间相互竞争,常表现为一个任务提升而另一任务下降。尽管已有多种优化方法通过调节任务梯度来实现任务平衡,但无法降低梯度冲突的发生率。本文系统研究了不同方法下的梯度冲突现象,提出一种基于稀疏训练(Sparse Training, ST)的策略:训练时仅更新部分模型参数,其余保持不变。大量实验表明,该方法能有效缓解梯度冲突,并带来更优的性能表现。此外,稀疏训练可轻松与现有梯度操纵技术结合,进一步增强其效果。
原文摘要 · Abstract (English)
Advancing towards generalist agents necessitates the concurrent processing of multiple tasks using a unified model, thereby underscoring the growing significance of simultaneous model training on multiple downstream tasks. A common issue in multi-task learning is the occurrence of gradient conflict, which leads to potential competition among different tasks during joint training. This competition often results in improvements in one task at the expense of deterioration in another. Although several optimization methods have been developed to address this issue by manipulating task gradients for better task balancing, they cannot decrease the incidence of gradient conflict. In this paper, we systematically investigate the occurrence of gradient conflict across different methods and propose a strategy to reduce such conflicts through sparse training (ST), wherein only a portion of the model's parameters are updated during training while keeping the rest unchanged. Our extensive experiments demonstrate that ST effectively mitigates conflicting gradients and leads to superior performance. Furthermore, ST can be easily integrated with gradient manipulation techniques, thus enhancing their effectiveness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。