Information-Theoretic Requirements for Gradient-Based Task Affinity Estimation in Multi-Task Learning
Not on the map. Why?
Abstract (arXiv)
Multi-task learning shows strikingly inconsistent results -- sometimes joint training helps substantially, sometimes it actively harms performance -- yet the field lacks a principled framework for predicting these outcomes. We identify a fundamental but unstated assumption underlying gradient-based task analysis: tasks must share training instances for gradient conflicts to reveal genuine relationships. When tasks are measured on the same inputs, gradient alignment reflects shared mechanistic structure; when measured on disjoint inputs, any apparent signal conflates task relationships with distributional shift. We discover this sample overlap requirement exhibits a sharp phase transition: below 30% overlap, gradient-task correlations are statistically indistinguishable from noise; above 40%, they reliably recover known biological structure. Comprehensive validation across multiple datasets achieves strong correlations and recovers biological pathway organization. Standard benchmarks systematically violate this requirement -- MoleculeNet operates at <5% overlap, TDC at 8-14% -- far below the threshold where gradient analysis becomes meaningful. This provides the first principled explanation for seven years of inconsistent MTL results.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 28 references that are in the map. First posted on arXiv 9 Apr 2026.
Builds on
19 of 28 references.
References · 19
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,358
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,051
- 2019Strategies for Pre-training Graph Neural Networkscited by 971
- 2017A Survey on Multi-Task Learningcited by 840
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
- 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 803
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 634
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
7 more
- 2019Analyzing Learned Molecular Representations for Property Predictioncited by 375
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 375
- 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 264
- 2018Characterizing and Avoiding Negative Transfercited by 229
- 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 222
- 2022Multi-Task Learning as a Bargaining Gamecited by 215
- 2019Task2Vec: Task Embedding for Meta-Learningcited by 205
Built on by
Not cited yet.