Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Multi-Task Learning for Dense Prediction Tasks: A Survey

About

With the advent of deep learning, many dense prediction tasks, i.e. tasks that produce pixel-level predictions, have seen significant performance improvements. The typical approach is to learn these tasks in isolation, that is, a separate neural network is trained for each individual task. Yet, recent multi-task learning (MTL) techniques have shown promising results w.r.t. performance, computations and/or memory footprint, by jointly tackling multiple tasks through a learned shared representation. In this survey, we provide a well-rounded view on state-of-the-art deep learning approaches for MTL in computer vision, explicitly emphasizing on dense prediction tasks. Our contributions concern the following. First, we consider MTL from a network architecture point-of-view. We include an extensive overview and discuss the advantages/disadvantages of recent popular MTL models. Second, we examine various optimization methods to tackle the joint learning of multiple tasks. We summarize the qualitative elements of these works and explore their commonalities and differences. Finally, we provide an extensive experimental evaluation across a variety of dense prediction benchmarks to examine the pros and cons of the different methods, including both architectural and optimization based strategies.

Simon Vandenhende, Stamatios Georgoulis, Wouter Van Gansbeke, Marc Proesmans, Dengxin Dai, Luc Van Gool• 2020

Related benchmarks

TaskDatasetResultRank
Semantic segmentationCityscapes (test)
mIoU70.43
1154
Depth EstimationNYU v2 (test)--
432
Surface Normal EstimationNYU v2 (test)--
224
Semantic segmentationNYUD v2 (test)
mIoU39.9
187
Semantic segmentationNYU Depth V2 (test)
mIoU39.44
183
Depth PredictionCityscapes (test)
RMSE6.797
52
Semantic segmentationSYNTHIA-to-Cityscapes (SYN2CS) 16 classes (val)--
50
Multi-task LearningCityscapes (test)
MR43.6
43
Semantic segmentationVKITTI2 -> Cityscapes 8 classes
mIoU79.63
19
Edge DetectionNYUD v2 (test)--
16
Showing 10 of 15 rows

Other info

Follow for update