Multi-Task Learning as a Bargaining Game
Not on the map. Why?
Abstract (arXiv)
In Multi-task learning (MTL), a joint model is trained to simultaneously make predictions for several tasks. Joint training reduces computation costs and improves data efficiency; however, since the gradients of these different tasks may conflict, training a joint model for MTL often yields lower performance than its corresponding single-task counterparts. A common method for alleviating this issue is to combine per-task gradients into a joint update direction using a particular heuristic. In this paper, we propose viewing the gradients combination step as a bargaining game, where tasks negotiate to reach an agreement on a joint direction of parameter update. Under certain assumptions, the bargaining problem has a unique solution, known as the Nash Bargaining Solution, which we propose to use as a principled approach to multi-task learning. We describe a new MTL optimization procedure, Nash-MTL, and derive theoretical guarantees for its convergence. Empirically, we show that Nash-MTL achieves state-of-the-art results on multiple MTL benchmarks in various domains.
Key facts. Cited by 209 papers in the map (data through 7 Oct 2026). Builds on 38 of its 65 references that are in the map. First posted on arXiv 2 Feb 2022.
Builds on
38 of 65 references.
References · 38
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2015SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentationcited by 1,483
- 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
- 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,168
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
- 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 791
26 more
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2018End-to-End Multi-Task Learning with Attentioncited by 533
- 2016Cross-stitch Networks for Multi-task Learningcited by 466
- 2019Provably Powerful Graph Networkscited by 385
- 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
- 2015Order Matters: Sequence to sequence for setscited by 236
- 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 199
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 167
- 2019Attentive Single-Tasking of Multiple Taskscited by 148
- 2020Multi-Task Reinforcement Learning with Soft Modularizationcited by 138
- 2020Learning the Pareto Front with Hypernetworkscited by 122
- 2015Instance-aware Semantic Segmentation via Multi-task Network Cascadescited by 120
- 2020Self-Supervised Learning for Domain Adaptation on Point-Cloudscited by 99
- 2018Adapting Auxiliary Losses Using Gradient Similaritycited by 98
- 2019Self-Supervised Generalisation with Meta Auxiliary Learningcited by 88
- 2016A Joint Many-Task Model: Growing a Neural Network for Multiple NLP Taskscited by 62
- 2018A Modulation Module for Multi-task Learning with Applications in Image Retrievalcited by 49
- 2020Auxiliary Learning by Implicit Differentiationcited by 45
- 2019Ray Interference: a Source of Plateaus in Deep Reinforcement Learningcited by 40
- 2016Learning to Push by Grasping: Using multiple tasks for effective learningcited by 36
- 2019Regularizing Deep Multi-Task Networks using Orthogonal Gradientscited by 36
- 2019First-order methods almost always avoid saddle points: the case of vanishing step-sizescited by 35
- 2021Fair Exploration via Axiomatic Bargainingcited by 21
- 2019The Kalai-Smorodinski solution for many-objective Bayesian optimizationcited by 13
Built on by
Citing papers per year: 2022: 5; 2023: 29; 2024: 54; 2025: 60; 2026: 61.
209 papers.
Sort:
By area:
- 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 114
- 2024Representation Surgery for Multi-Task Model Mergingcited by 95
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 74
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2023Independent Component Alignment for Multi-Task Learningcited by 69
- 2023FedDisco: Federated Learning with Discrepancy-Aware Collaborationcited by 66
- 2023Revisiting Scalarization in Multi-Task Learning: A Theoretical Perspectivecited by 51
- 2024Smooth Tchebycheff Scalarization for Multi-Objective Optimizationcited by 50
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networkscited by 43
- 2023FAME-ViL: Multi-Tasking Vision-Language Model for Heterogeneous Fashion Taskscited by 37
- 2024Multimodal Instruction Tuning with Conditional Mixture of LoRAcited by 36
- 2024Fair Resource Allocation in Multi-Task Learningcited by 34
38 more
- 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 33
- 2023Addressing Negative Transfer in Diffusion Modelscited by 33
- 2022Pareto Manifold Learning: Tackling multiple tasks via ensembles of single-task modelscited by 29
- 2023ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learningcited by 28
- 2023Denoising Task Routing for Diffusion Modelscited by 28
- 2023Dual-Balancing for Multi-Task Learningcited by 27
- 2024Robust Multi-Task Learning with Excess Riskscited by 23
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
- 2025CAT Merging: A Training-Free Approach for Resolving Conflicts in Model Mergingcited by 22
- 2024E2E-MFD: Towards End-to-End Synchronous Multimodal Fusion Detectioncited by 21
- 2024Dual Cone Gradient Descent for Training Physics-Informed Neural Networkscited by 21
- 2023Trust Region-Based Safe Distributional Reinforcement Learning for Multiple Constraintscited by 20
- 2024Are NeRFs ready for autonomous driving? Towards closing the real-to-simulation gapcited by 20
- 2024Jacobian Descent for Multi-Objective Optimizationcited by 18
- 2024MUNBa: Machine Unlearning via Nash Bargainingcited by 17
- 2022Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspectivecited by 15
- 2024Few for Many: Tchebycheff Set Scalarization for Many-Objective Optimizationcited by 15
- 2025Gradient-Based Multi-Objective Deep Learning: Algorithms, Theories, Applications, and Beyondcited by 15
- 2025Selective Task Group Updates for Multi-Task Optimizationcited by 15
- 2025SplitFlow: Flow Decomposition for Inversion-Free Text-to-Image Editingcited by 15
- 2022Mitigating Gradient Bias in Multi-objective Learning: A Provably Convergent Stochastic Approachcited by 14
- 2023Fairness in Multi-Task Learning via Wasserstein Barycenterscited by 14
- 2024Pareto Low-Rank Adapters: Efficient Multi-Task Learning with Preferencescited by 13
- 2023Auxiliary Learning as an Asymmetric Bargaining Gamecited by 12
- 2024Exploring Training on Heterogeneous Data with Mixture of Low-rank Adapterscited by 12
- 2024SurgeryV2: Bridging the Gap Between Model Merging and Multi-Task Learning with Deep Representation Surgerycited by 12
- 2024Understanding Forgetting in LLM Supervised Fine-Tuning and Preference Learning -- A Convex Optimization Perspectivecited by 12
- 2024Proactive Gradient Conflict Mitigation in Multi-Task Learning: A Sparse Training Perspectivecited by 12
- 2026DanceOPD: On-Policy Generative Field Distillationcited by 12
- 2024Bayesian Uncertainty for Gradient Aggregation in Multi-Task Learningcited by 11
- 2024GO4Align: Group Optimization for Multi-Task Alignmentcited by 11
- 2025Task Arithmetic in Trust Region: A Training-Free Model Merging Approach to Navigate Knowledge Conflictscited by 11
- 2025MTL-UE: Learning to Learn Nothing for Multi-Task Learningcited by 11
- 2024Quantifying Task Priority for Multi-Task Optimizationcited by 10
- 2023Improvable Gap Balancing for Multi-Task Learningcited by 9
- 2024Safe and Balanced: A Framework for Constrained Multi-Objective Reinforcement Learningcited by 9
- 2024Efficient Pareto Manifold Learning with Low-Rank Structurecited by 9
- 2025Commute Your Domains: Trajectory Optimality Criterion for Multi-Domain Learningcited by 9
50 most cited of 209 All 209 (JSON)
- 2026ElectrolyteFM: Unifying Electrolyte Property Prediction through Cross-Property Knowledge Learning30 Sep 2026
- 2026Beyond Spatial-Domain Supervision: A Relation Constrained Space for Multi-Modal Image Fusion30 Sep 2026
- 2026Beyond Layers: Position-Resolved Gradient Conflict and Position-Aware Modulation for Unified Multimodal Models29 Sep 2026
- 2026Does Gradient Conflict Predict the Understanding--Generation Trade-off? A Controlled Audit of Conflict-Metric Validity in Unified Multimodal Models29 Sep 2026
- 2026When Sparse Reward Meets Dense Distillation: Training Dynamics of On-Policy Distillation28 Sep 2026
- 2026Multi-Scale Semantic Mapping in Urban Environments via Observation Calibration and Policy Dependence Regularization28 Sep 2026
- 2026Learning Through Game: Skewed Transfer of Tabular Knowledge to Strengthen Image Model26 Sep 2026
- 2026Gradient Surgery for Physics-Informed Neural Networks25 Sep 2026
- 2026On the Potential of Multi-Task Learning in Predictive Process Monitoring11 Sep 2026
- 2026SIMS: Scale-Invariant Merit-Function-Based Scalarization for Multi-Task Learning11 Sep 2026
- 2026Not All Variables Agree: Reliability-Aware Variable-Wise Gradient Surgery for Multivariate Time-Series Forecasting8 Sep 2026
- 2026One Step, One Lead: Mitigating Higher-Order Interference in Multi-Domain Reinforcement Learning via Cross-Step Control6 Sep 2026
38 more
- 2026An Exploratory Study of Frequency-Aware Task Weighting for YOLOv8-Based Unified Driving Perception31 Aug 2026
- 2026Bounded Precision-Geometry Scaling for Robust Multi-Task Learning under Loss Scale Mismatch21 Aug 2026
- 2026MOON: Multi-Objective OrthoNormalized Updates for Multitask Learning12 Aug 2026
- 2026The Free-Recipe Limit: Every Recipe Effect Measures Which Premise of an Idealised Learner Broke9 Aug 2026
- 2026EMAN: Optimization-Driven Capacity Growth through Path Emergence in Multi-Task Learning5 Aug 2026
- 2026End-to-End Fairness Optimization with Fair Decision-Focused Learning31 Jul 2026
- 2026Rosetta: Composable Native Multimodal Pretraining1 Jul 2026
- 2026T3R: Deeper Test-Time Adaptation for Graph Neural Networks via Gradient Rotation29 Jun 2026
- 2026Reducing Per-Sample Interference in Stochastic Optimization28 Jun 2026
- 2026Not All Objectives Are Born Equal: Priority-Constrained Descent for Hierarchical Multi-Objective Optimization28 Jun 2026
- 2026DanceOPD: On-Policy Generative Field Distillation25 Jun 2026
- 2026The Geometry of Sequential Learning: Lie-Bracket Prediction of Transfer Order23 Jun 2026
- 2026HadBalance: A Plug-and-Play Unified Global Geometric Prior Framework for Generalizable Biomedical Segmentation14 Jun 2026
- 2026De novo molecular generation with optical property preconditioning at the token level6 Jun 2026
- 2026SAW: Stage-Aware Dynamic Weighting for Multi-Objective Reinforcement Learning in Large Language Models5 Jun 2026
- 2026Multilingual Fine-Tuning via Localized Gradient Conflict Resolution4 Jun 2026
- 2026Who Needs Labels? Adapting Vision Foundation Models With the Metadata You Already Have3 Jun 2026
- 2026MAdam: Metric-Aware Multi-Objective Adam2 Jun 2026
- 2026CORE-MTL: Rethinking Gradient Balancing via Causal Orthogonal Representations1 Jun 2026
- 2026A Unified Framework for Gradient Aggregation in Multi-Objective Optimization28 May 2026
- 2026GAC: Noise-Aware Adaptive Mixing for Hybrid SFT-RL Post-Training25 May 2026
- 2026The Double Dilemma in Multi-Task Radiology Report Generation: A Gradient Dynamics Analysis and Solution21 May 2026
- 2026S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHF15 May 2026
- 2026Distributionally Robust Multi-Task Reinforcement Learning via Adaptive Task Sampling14 May 2026
- 2026Low-Rank Adapters Initialization via Gradient Surgery for Continual Learning12 May 2026
- 2026TOPPO: Rethinking PPO for Multi-Task Reinforcement Learning with Critic Balancing12 May 2026
- 2026Per-Loss Adapters for Gradient Conflict in Physics-Informed Neural Networks11 May 2026
- 2026Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learning8 May 2026
- 2026MARBLE: Multi-Aspect Reward Balance for Diffusion RL7 May 2026
- 2026B3-Net: Controlled Posterior Bridge Learning for Multi-Task Dense Prediction7 May 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026Artificial Jagged Intelligence as Uneven Optimization Energy Allocation Capability Concentration, Redistribution, and Optimization Governance2 May 2026
- 2026Conflict-Aware Harmonized Rotational Gradient for Multiscale Kinetic Regimes27 Apr 2026
- 2026Modeling LLM Unlearning as an Asymmetric Two-Task Learning Problem16 Apr 2026
- 2026Delve into the Applicability of Advanced Optimizers for Multi-Task Learning10 Apr 2026
- 2026Information-Theoretic Requirements for Gradient-Based Task Affinity Estimation in Multi-Task Learning9 Apr 2026
- 2026A Unified Foundation Model for All-in-One Multi-Modal Remote Sensing Image Restoration and Fusion with Language Prompting7 Apr 2026
- 2026HyperCT: Low-Rank Hypernet for Unified Chest CT Analysis3 Apr 2026
50 newest of 209 All 209 (JSON)
- 2024Understanding Forgetting in LLM Supervised Fine-Tuning and Preference Learning -- A Convex Optimization Perspectivecited by 12
- 2025AutoMixAlign: Adaptive Data Mixing for Multi-Task Preference Optimization in LLMscited by 8
- 2026The Geometry of Sequential Learning: Lie-Bracket Prediction of Transfer Ordercited by 8
- 2026SAW: Stage-Aware Dynamic Weighting for Multi-Objective Reinforcement Learning in Large Language Modelscited by 3
- 2026Uncovering Cross-Objective Interference in Multi-Objective Alignmentcited by 2
- 2026S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHFcited by 0
- 2026The Geometry of Sequential Learning: Lie-Bracket Prediction of Transfer Order23 Jun 2026
- 2026SAW: Stage-Aware Dynamic Weighting for Multi-Objective Reinforcement Learning in Large Language Models5 Jun 2026
- 2026S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHF15 May 2026
- 2026Uncovering Cross-Objective Interference in Multi-Objective Alignment6 Feb 2026
- 2025AutoMixAlign: Adaptive Data Mixing for Multi-Task Preference Optimization in LLMs31 May 2025
- 2024Understanding Forgetting in LLM Supervised Fine-Tuning and Preference Learning -- A Convex Optimization Perspective20 Oct 2024
- 2023Addressing Negative Transfer in Diffusion Modelscited by 33
- 2023Denoising Task Routing for Diffusion Modelscited by 28
- 2025SplitFlow: Flow Decomposition for Inversion-Free Text-to-Image Editingcited by 15
- 2026DanceOPD: On-Policy Generative Field Distillationcited by 12
- 2025FairHuman: Boosting Hand and Face Quality in Human Image Generation with Minimum Potential Delay Fairness in Diffusion Modelscited by 4
- 2026MARBLE: Multi-Aspect Reward Balance for Diffusion RLcited by 3
- 2026DanceOPD: On-Policy Generative Field Distillation25 Jun 2026
- 2026MARBLE: Multi-Aspect Reward Balance for Diffusion RL7 May 2026
- 2025SplitFlow: Flow Decomposition for Inversion-Free Text-to-Image Editing29 Oct 2025
- 2025FairHuman: Boosting Hand and Face Quality in Human Image Generation with Minimum Potential Delay Fairness in Diffusion Models3 Jul 2025
- 2023Denoising Task Routing for Diffusion Models11 Oct 2023
- 2023Addressing Negative Transfer in Diffusion Models1 Jun 2023
- 2026TSEmbed: Unlocking Task Scaling in Universal Multimodal Embeddingscited by 6
- 2025Beyond Reasoning Gains: Mitigating General-Capability Forgetting in Large Reasoning Modelscited by 5
- 2026One Step, One Lead: Mitigating Higher-Order Interference in Multi-Domain Reinforcement Learning via Cross-Step Controlcited by 1
- 2026One Step, One Lead: Mitigating Higher-Order Interference in Multi-Domain Reinforcement Learning via Cross-Step Control6 Sep 2026
- 2026TSEmbed: Unlocking Task Scaling in Universal Multimodal Embeddings5 Mar 2026
- 2025Beyond Reasoning Gains: Mitigating General-Capability Forgetting in Large Reasoning Models24 Oct 2025
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
- 2023HarmonyDream: Task Harmonization Inside World Models30 Sep 2023
- 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 114
- 2024Representation Surgery for Multi-Task Model Mergingcited by 95
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 74
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2023Independent Component Alignment for Multi-Task Learningcited by 69
- 2023FedDisco: Federated Learning with Discrepancy-Aware Collaborationcited by 66
- 2023Revisiting Scalarization in Multi-Task Learning: A Theoretical Perspectivecited by 51
- 2024Smooth Tchebycheff Scalarization for Multi-Objective Optimizationcited by 50
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networkscited by 43
- 2024Fair Resource Allocation in Multi-Task Learningcited by 34
- 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 33
- 2022Pareto Manifold Learning: Tackling multiple tasks via ensembles of single-task modelscited by 29
12 most cited of 179 All 179 (JSON)
- 2026ElectrolyteFM: Unifying Electrolyte Property Prediction through Cross-Property Knowledge Learning30 Sep 2026
- 2026Beyond Spatial-Domain Supervision: A Relation Constrained Space for Multi-Modal Image Fusion30 Sep 2026
- 2026Beyond Layers: Position-Resolved Gradient Conflict and Position-Aware Modulation for Unified Multimodal Models29 Sep 2026
- 2026Does Gradient Conflict Predict the Understanding--Generation Trade-off? A Controlled Audit of Conflict-Metric Validity in Unified Multimodal Models29 Sep 2026
- 2026When Sparse Reward Meets Dense Distillation: Training Dynamics of On-Policy Distillation28 Sep 2026
- 2026Multi-Scale Semantic Mapping in Urban Environments via Observation Calibration and Policy Dependence Regularization28 Sep 2026
- 2026Learning Through Game: Skewed Transfer of Tabular Knowledge to Strengthen Image Model26 Sep 2026
- 2026Gradient Surgery for Physics-Informed Neural Networks25 Sep 2026
- 2026On the Potential of Multi-Task Learning in Predictive Process Monitoring11 Sep 2026
- 2026SIMS: Scale-Invariant Merit-Function-Based Scalarization for Multi-Task Learning11 Sep 2026
- 2026Not All Variables Agree: Reliability-Aware Variable-Wise Gradient Surgery for Multivariate Time-Series Forecasting8 Sep 2026
- 2026An Exploratory Study of Frequency-Aware Task Weighting for YOLOv8-Based Unified Driving Perception31 Aug 2026
12 newest of 179 All 179 (JSON)