Article thumbnail

Multi-Task Learning by a Top-Down Control Network

By Hila Levi and Shimon Ullman

Abstract

A general problem that received considerable recent attention is how to perform multiple tasks in the same network, maximizing both prediction accuracy and efficiency of training. Recent approaches address this problem by branching networks, or by a channel-wise modulation of the feature-maps with task specific vectors. We propose a novel architecture that uses a top-down network to modify the main network according to the task in a channel-wise, as well as spatial-wise, image-dependent computation scheme. We show the effectiveness of our scheme by achieving better results than alternative state-of-the-art approaches to multi-task learning. We also demonstrate our advantages in terms of task selectivity, scaling the number of tasks, learning from fewer examples and interpretability

Topics: Computer Science - Machine Learning, Computer Science - Computer Vision and Pattern Recognition, Statistics - Machine Learning
Year: 2020
OAI identifier: oai:arXiv.org:2002.03335

Suggested articles


To submit an update or takedown request for this paper, please submit an Update/Correction/Removal Request.