Sep, 2024

基于梯度投影的多任务学习任务权重调整

TL;DR本研究解决了多任务学习中任务梯度冲突导致的模型训练性能下降问题。我们提出了一种新的方法,将梯度投影算法与任务优先级调整结合起来,通过概率分布来决定在冲突情况下哪些任务梯度被投影。这一方法在多个数据集上的实验结果显示,与传统的梯度投影相比,能够显著提升多任务的性能指标。