Aug, 2023

具有门控机制的可变形混合变压器用于密集预测的多任务学习

TL;DR通过结合可变形卷积神经网络(deformable CNN)和基于查询的Transformer的优点,并采用共享门控机制,我们提出了一种新颖的多任务学习模型DeMTG。该模型具有任务特定的强大灵活性,相较于传统的多任务学习方法,其成本更低、复杂度更小、参数更少。实验证明,DeMTG在三个密集预测数据集上使用更少的GFLOPs,显著优于基于Transformer和CNN的竞争模型,且我们的代码和模型可在给出的网址获取。