arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.06129cs.CVcs.LG
收藏

实例级任务参数:一个鲁棒的多任务加权框架

Instance-Level Task Parameters: A Robust Multi-task Weighting Framework

Apple Inc.(苹果公司)

Pavan Kumar Anasosalu Vasu, Shreyas Saxena, Oncel Tuzel

摘要:

最近的研究表明,深度神经网络通过在多个相关任务之间学习共享表示来受益于多任务学习。然而,这种系统的性能取决于训练期间涉及的各种损失之间的相对权重。损失加权方案的先前工作假设实例对于所有任务都是同样容易或困难的。为了打破这个假设,我们让训练过程决定数据集中每个实例的任务的最佳权重。更具体地说,我们为数据集中的每个实例配备了一组可学习的参数(实例级任务参数),其中基数等于模型学习的任务数量。这些参数为实例的每个任务的权重建模。它们通过梯度下降更新,不需要手工制定规则。我们在SURREAL和CityScapes数据集上进行了广泛的实验,用于人体形状和姿势估计,深度估计和语义分割任务。在这些任务中,我们的方法优于最近的动态损失加权方法,例如在SURREAL上将表面估计误差减少了8.97%。当应用于一个或多个任务可能具有噪声注释的数据集时,所提出的方法学习优先考虑从给定任务的干净标签中学习,例如将表面估计误差减少高达60%。我们还表明,我们可以可靠地检测损坏的标签为一个给定的任务作为一个副产品,从学习的实例级任务参数。

英文摘要:

Recent works have shown that deep neural networks benefit from multi-task learning by learning a shared representation across several related tasks. However, performance of such systems depend on relative weighting between various losses involved during training. Prior works on loss weighting schemes assume that instances are equally easy or hard for all tasks. In order to break this assumption, we let the training process dictate the optimal weighting of tasks for every instance in the dataset. More specifically, we equip every instance in the dataset with a set of learnable parameters (instance-level task parameters) where the cardinality is equal to the number of tasks learned by the model. These parameters model the weighting of each task for an instance. They are updated by gradient descent and do not require hand-crafted rules. We conduct extensive experiments on SURREAL and CityScapes datasets, for human shape and pose estimation, depth estimation and semantic segmentation tasks. In these tasks, our approach outperforms recent dynamic loss weighting approaches, e.g. reducing surface estimation errors by 8.97% on SURREAL. When applied to datasets where one or more tasks can have noisy annotations, the proposed method learns to prioritize learning from clean labels for a given task, e.g. reducing surface estimation errors by up to 60%. We also show that we can reliably detect corrupt labels for a given task as a by-product from learned instance-level task parameters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑