论文标题
使用表示相似性指导多任务功能共享的遏制任务干扰
Curbing Task Interference using Representation Similarity-Guided Multi-Task Feature Sharing
论文作者
论文摘要
通过共享编码器和解码器而不是仅共享编码器,对密集预测任务的多任务学习提供了一个有吸引力的方面,以提高准确性和计算效率。当任务相似时,共享解码器将作为额外的归纳偏见,为任务提供更多的互补信息的空间。但是,增加的共享暴露于任务干扰的更多参数,这可能会阻碍概括和鲁棒性。在利用共享解码器的归纳偏见的同时,遏制这种干扰的有效方法仍然是一个开放的挑战。为了应对这一挑战,我们建议进行渐进解码器融合(PDF),以根据任务间表示相似性逐步组合任务解码器。我们表明,此过程导致了一个多任务网络,具有更好地概括到分发和分布数据以及对对抗性攻击的鲁棒性。此外,我们观察到,该多任务网络的不同任务的预测彼此更加一致。
Multi-task learning of dense prediction tasks, by sharing both the encoder and decoder, as opposed to sharing only the encoder, provides an attractive front to increase both accuracy and computational efficiency. When the tasks are similar, sharing the decoder serves as an additional inductive bias providing more room for tasks to share complementary information among themselves. However, increased sharing exposes more parameters to task interference which likely hinders both generalization and robustness. Effective ways to curb this interference while exploiting the inductive bias of sharing the decoder remains an open challenge. To address this challenge, we propose Progressive Decoder Fusion (PDF) to progressively combine task decoders based on inter-task representation similarity. We show that this procedure leads to a multi-task network with better generalization to in-distribution and out-of-distribution data and improved robustness to adversarial attacks. Additionally, we observe that the predictions of different tasks of this multi-task network are more consistent with each other.