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

Attentive Task Interaction Network for Multi-Task Learning

McGill University(麦吉尔大学);Mila - Québec AI Institute(魁北克人工智能研究所(Mila));Department of Electrical & Computer Engineering McGill University(麦吉尔大学电气与计算机工程系)

Dimitrios Sinodinos, Narges Armanfard

英文摘要:

Multitask learning (MTL) has recently gained a lot of popularity as a learning paradigm that can lead to improved per-task performance while also using fewer per-task model parameters compared to single task learning. One of the biggest challenges regarding MTL networks involves how to share features across tasks. To address this challenge, we propose the Attentive Task Interaction Network (ATI-Net). ATI-Net employs knowledge distillation of the latent features for each task, then combines the feature maps to provide improved contextualized information to the decoder. This novel approach to introducing knowledge distillation into an attention based multitask network outperforms state of the art MTL baselines such as the standalone MTAN and PAD-Net, with roughly the same number of model parameters.

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

↑