Multi-task Learning by Leveraging the Semantic Information
Not on the map. Why?
Abstract (arXiv)
One crucial objective of multi-task learning is to align distributions across tasks so that the information between them can be transferred and shared. However, existing approaches only focused on matching the marginal feature distribution while ignoring the semantic information, which may hinder the learning performance. To address this issue, we propose to leverage the label information in multi-task learning by exploring the semantic conditional relations among tasks. We first theoretically analyze the generalization bound of multi-task learning based on the notion of Jensen-Shannon divergence, which provides new insights into the value of label information in multi-task learning. Our analysis also leads to a concrete algorithm that jointly matches the semantic distribution and controls label distribution divergence. To confirm the effectiveness of the proposed method, we first compare the algorithm with several baselines on some benchmarks and then test the algorithms under label space shift conditions. Empirical results demonstrate that the proposed method could outperform most baselines and achieve state-of-the-art performance, particularly showing the benefits under the label shift conditions.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 28 of its 57 references that are in the map. First posted on arXiv 3 Mar 2021.
Builds on
28 of 57 references.
References · 28
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
- 2019Domain Generalization via Model-Agnostic Learning of Semantic Featurescited by 405
- 2018Detecting and Correcting for Label Shift with Black Box Predictorscited by 364
- 2018Recent Advances in Open Set Recognition: A Surveycited by 309
- 2017Distral: Robust Multitask Reinforcement Learningcited by 231
- 2015The Benefit of Multitask Representation Learningcited by 194
- 2019Domain Generalization Using a Mixture of Multiple Latent Domainscited by 170
16 more
- 2018Partial Adversarial Domain Adaptationcited by 127
- 2019Domain-Symmetric Networks for Adversarial Domain Adaptationcited by 120
- 2019Regularized Learning for Domain Adaptation under Label Shiftscited by 118
- 2017Few-Shot Adversarial Domain Adaptationcited by 111
- 2016Theoretical Analysis of Domain Adaptation with Optimal Transportcited by 105
- 2017Adversarial Multi-task Learning for Text Classificationcited by 91
- 2017Identifying beneficial task relations for multi-task learning in deep neural networkscited by 59
- 2017Label Efficient Learning of Transferable Representations across Domains and Taskscited by 49
- 2018A Modulation Module for Multi-task Learning with Applications in Image Retrievalcited by 49
- 2019A Principled Approach for Learning Task Similarity in Multitask Learningcited by 32
- 2020Domain Generalization via Optimal Transport with Metric Similarity Learningcited by 27
- 2018Multi-task Learning of Hierarchical Vision-Language Representationcited by 26
- 2017Deep Learning for Multi-Task Medical Image Segmentation in Multiple Modalitiescited by 25
- 2016Multi-Task Learning with Labeled and Unlabeled Taskscited by 17
- 2020Beyond H-Divergence: Domain Adaptation Theory With Jensen-Shannon Divergencecited by 15
- 2014Heterogeneous Multi-task Learning for Human Pose Estimation with Deep Convolutional Neural Networkcited by 11
Built on by
Citing papers per year: 2021: 1; 2022: 2; 2023: 2.
5 papers.
- 2021Learning to Transfer for Traffic Forecasting via Multi-task Learningcited by 7
- 2023Personalised Federated Learning On Heterogeneous Feature Spacescited by 5
- 2022Gap Minimization for Knowledge Sharing and Transfercited by 4
- 2022Compressed Hierarchical Representations for Multi-Task Learning and Task Clusteringcited by 2
- 2023Learning Easily Updated General Purpose Text Representations with Adaptable Task-Specific Prefixescited by 0
- 2023Learning Easily Updated General Purpose Text Representations with Adaptable Task-Specific Prefixes22 May 2023
- 2023Personalised Federated Learning On Heterogeneous Feature Spaces26 Jan 2023
- 2022Compressed Hierarchical Representations for Multi-Task Learning and Task Clustering31 May 2022
- 2022Gap Minimization for Knowledge Sharing and Transfer26 Jan 2022
- 2021Learning to Transfer for Traffic Forecasting via Multi-task Learning27 Nov 2021