End-to-end multi-task learning with attention
Author(s)
Johns, Edward
Liu, Shikun
Davison, Andrew
Type
Conference Paper
Abstract
We propose a novel multi-task learning architecture, which allows learning of task-specific feature-level attention. Our design, the Multi-Task Attention Network (MTAN), consists of a single shared network containing a global feature pool, together with a soft-attention module for each task. These modules allow for learning of task-specific features from the global features, whilst simultaneously allowing for features to be shared across different tasks. The architecture can be trained end-to-end and can be built upon any feed-forward neural network, is simple to implement, and is parameter efficient. We evaluate our approach on a variety of datasets, across both image-to-image predictions and image classification tasks. We show that our architecture is state-of-the-art in multi-task learning compared to existing methods, and is also less sensitive to various weighting schemes in the multi-task loss function. Code is available at https://github.com/lorenmt/mtan.
Date Issued
2020-01-09
Date Acceptance
2019-03-11
Citation
2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020
Publisher
IEEE
Journal / Book Title
2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
Copyright Statement
© 2019 IEEE.
Source
The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2019
Publication Status
Published
Start Date
2019-06-16
Finish Date
2019-06-20
Coverage Spatial
Long Beach, California, USA