"It's a Match!" -- A Benchmark of Task Affinity Scores for Joint Learning
Not on the map. Why?
Abstract (arXiv)
While the promises of Multi-Task Learning (MTL) are attractive, characterizing the conditions of its success is still an open problem in Deep Learning. Some tasks may benefit from being learned together while others may be detrimental to one another. From a task perspective, grouping cooperative tasks while separating competing tasks is paramount to reap the benefits of MTL, i.e., reducing training and inference costs. Therefore, estimating task affinity for joint learning is a key endeavor. Recent work suggests that the training conditions themselves have a significant impact on the outcomes of MTL. Yet, the literature is lacking of a benchmark to assess the effectiveness of tasks affinity estimation techniques and their relation with actual MTL performance. In this paper, we take a first step in recovering this gap by (i) defining a set of affinity scores by both revisiting contributions from previous literature as well presenting new ones and (ii) benchmarking them on the Taskonomy dataset. Our empirical campaign reveals how, even in a small-scale scenario, task affinity scoring does not correlate well with actual MTL performance. Yet, some metrics can be more indicative than others.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 18 of its 25 references that are in the map. First posted on arXiv 7 Jan 2023.
Builds on
18 of 25 references.
References · 18
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,320
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
- 2020Gradient Surgery for Multi-Task Learningcited by 1,210
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
- 2016Cross-stitch Networks for Multi-task Learningcited by 466
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
- 2016Not Just a Black Box: Learning Important Features Through Propagating Activation Differencescited by 295
- 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 218
- 2019Task2Vec: Task Embedding for Meta-Learningcited by 197
- 2019AdaShare: Learning What To Share For Efficient Deep Multi-Task Learningcited by 159
- 2019Representation Similarity Analysis for Efficient Task taxonomy & Transfer Learningcited by 86
6 more
- 2017Identifying beneficial task relations for multi-task learning in deep neural networkscited by 59
- 2018A Modulation Module for Multi-task Learning with Applications in Image Retrievalcited by 49
- 2020Dynamic Task Weighting Methods for Multi-task Networks in Autonomous Driving Systemscited by 34
- 2019Deep Model Transferability from Attribution Mapscited by 28
- 2020Maximum Roaming Multi-Task Learningcited by 11
- 2021Disentangling Transfer and Interference in Multi-Domain Learningcited by 3
Built on by
Citing papers per year: 2024: 1.
1 paper.