Gradient-Based Multi-Objective Deep Learning: Algorithms, Theories, Applications, and Beyond
Not on the map. Why?
Abstract (arXiv)
Many modern deep learning applications require balancing multiple objectives that are often conflicting. Examples include multi-task learning, fairness-aware learning, and the alignment of Large Language Models (LLMs). This leads to multi-objective deep learning, which tries to find optimal trade-offs or Pareto-optimal solutions by adapting mathematical principles from the field of Multi-Objective Optimization (MOO). However, directly applying gradient-based MOO techniques to deep neural networks presents unique challenges, including high computational costs, optimization instability, and the difficulty of effectively incorporating user preferences. This paper provides a comprehensive survey of gradient-based techniques for multi-objective deep learning. We systematically categorize existing algorithms based on their outputs: (i) methods that find a single, well-balanced solution, (ii) methods that generate a finite set of diverse Pareto-optimal solutions, and (iii) methods that learn a continuous Pareto set of solutions. In addition to this taxonomy, the survey covers theoretical analyses, key applications, practical resources, and highlights open challenges and promising directions for future research. A comprehensive list of multi-objective deep learning algorithms is available at https://github.com/Baijiong-Lin/Awesome-Multi-Objective-Deep-Learning.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 143 of its 248 references that are in the map. First posted on arXiv 19 Jan 2025.
Builds on
143 of 248 references.
References · 143
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 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
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,168
- 2016HyperNetworkscited by 1,003
- 2017A Survey on Multi-Task Learningcited by 838
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
38 more
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
- 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 791
- 2020XTREME: A Massively Multilingual Multi-task Benchmark for Evaluating Cross-lingual Generalizationcited by 738
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
- 2018End-to-End Multi-Task Learning with Attentioncited by 533
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
- 2018FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Searchcited by 480
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2023Fine-Grained Human Feedback Gives Better Rewards for Language Model Trainingcited by 399
- 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
- 2023Personalized Soups: Personalized Large Language Model Alignment via Post-hoc Parameter Mergingcited by 271
- 2020A Comprehensive Survey of Neural Architecture Search: Challenges and Solutionscited by 253
- 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
- 2023Efficient Diffusion Training via Min-SNR Weighting Strategycited by 242
- 2024A Survey on Mixture of Experts in Large Language Modelscited by 233
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
- 2024HelpSteer2: Open-source dataset for training top-performing reward modelscited by 232
- 2019Pareto Multi-Task Learningcited by 224
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 199
- 2020Differentiable Expected Hypervolume Improvement for Parallel Multi-Objective Bayesian Optimizationcited by 176
- 2024Rewards-in-Context: Multi-objective Alignment of Foundation Models with Dynamic Preference Adjustmentcited by 171
- 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 167
- 2020Minimax Pareto Fairness: A Multi Objective Perspectivecited by 152
- 2024Arithmetic Control of LLMs for Diverse User Preferences: Directional Preference Alignment with Multi-Objective Rewardscited by 145
- 2023Unexpected Improvements to Expected Improvement for Bayesian Optimizationcited by 144
- 2020Learning the Pareto Front with Hypernetworkscited by 122
- 2023Beyond One-Preference-Fits-All Alignment: Multi-Objective Direct Preference Optimizationcited by 121
- 2021Parallel Bayesian Optimization of Multiple Noisy Objectives with Expected Hypervolume Improvementcited by 115
- 2024Controllable Preference Optimization: Toward Controllable Multi-Objective Alignmentcited by 105
- 2021Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learningcited by 100
- 2022Multi-Objective GFlowNetscited by 100
- 2018A Flexible Framework for Multi-Objective Bayesian Optimization using Random Scalarizationscited by 98
- 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 90
- 2022InvPT: Inverted Pyramid Multi-task Transformer for Dense Scene Understandingcited by 89
50 most cited of 143 All 143 (JSON)
Built on by
Citing papers per year: 2025: 9; 2026: 6.
15 papers.
Sort:
By area:
- 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Drivingcited by 39
- 2025PARM: Multi-Objective Test-Time Alignment via Preference-Aware Autoregressive Reward Modelcited by 27
- 2025Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasetscited by 17
- 2025TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Predictioncited by 9
- 2026Advancing General-Purpose Reasoning Models with Modular Gradient Surgerycited by 9
- 2025Rethinking Parameter Sharing for LLM Fine-Tuning with Multiple LoRAscited by 3
- 2025Addressing Gradient Misalignment in Data-Augmented Training for Robust Speech Deepfake Detectioncited by 2
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspectivecited by 2
- 2025SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbationcited by 1
- 2025A Framework for Controllable Multi-objective Learning with Annealed Stein Variational Hypernetworkscited by 0
- 2026Barrier-enforced multi-objective optimization for direct point and sharp interval forecastingcited by 0
- 2026MGDA-Decoupled: Geometry-Aware Multi-Objective Optimisation for DPO-based LLM Alignmentcited by 0
- 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Level13 Aug 2026
- 2026MAdam: Metric-Aware Multi-Objective Adam2 Jun 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026MGDA-Decoupled: Geometry-Aware Multi-Objective Optimisation for DPO-based LLM Alignment22 Apr 2026
- 2026Barrier-enforced multi-objective optimization for direct point and sharp interval forecasting20 Apr 2026
- 2026Advancing General-Purpose Reasoning Models with Modular Gradient Surgery2 Feb 2026
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspective21 Oct 2025
- 2025Rethinking Parameter Sharing for LLM Fine-Tuning with Multiple LoRAs29 Sep 2025
- 2025Addressing Gradient Misalignment in Data-Augmented Training for Robust Speech Deepfake Detection25 Sep 2025
- 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Driving22 Sep 2025
- 2025TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Prediction6 Aug 2025
- 2025SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbation10 Jul 2025
3 more
- 2025A Framework for Controllable Multi-objective Learning with Annealed Stein Variational Hypernetworks7 Jun 2025
- 2025Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasets17 May 2025
- 2025PARM: Multi-Objective Test-Time Alignment via Preference-Aware Autoregressive Reward Model6 May 2025
- 2025Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasetscited by 17
- 2025TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Predictioncited by 9
- 2025Addressing Gradient Misalignment in Data-Augmented Training for Robust Speech Deepfake Detectioncited by 2
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspectivecited by 2
- 2025SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbationcited by 1
- 2025A Framework for Controllable Multi-objective Learning with Annealed Stein Variational Hypernetworkscited by 0
- 2026Barrier-enforced multi-objective optimization for direct point and sharp interval forecastingcited by 0
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimizationcited by 0
- 2026MAdam: Metric-Aware Multi-Objective Adamcited by 0
- 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Levelcited by 0
- 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Level13 Aug 2026
- 2026MAdam: Metric-Aware Multi-Objective Adam2 Jun 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026Barrier-enforced multi-objective optimization for direct point and sharp interval forecasting20 Apr 2026
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspective21 Oct 2025
- 2025Addressing Gradient Misalignment in Data-Augmented Training for Robust Speech Deepfake Detection25 Sep 2025
- 2025TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Prediction6 Aug 2025
- 2025SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbation10 Jul 2025
- 2025A Framework for Controllable Multi-objective Learning with Annealed Stein Variational Hypernetworks7 Jun 2025
- 2025Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasets17 May 2025
Similar, not cited
Close in wording, no citation either way
- 2024A Decade of Deep Learning: A Survey on The Magnificent Seven
- 2025Pareto Multi-Objective Alignment for Language Models
- 2025Aligned Multi Objective Optimization
- 2026Gradient Smoothing: Coupling Layer-wise Updates for Improved Optimization
- 2026Multi-Objective Hierarchical Optimization with Large Language Models
- 2025An LLM-Empowered Adaptive Evolutionary Algorithm For Multi-Component Deep Learning Systems
- 2023An Invitation to Deep Reinforcement Learning
- 2026Multi-Objective Learning for Diffusion Models: A Statistical Theory under Semi-Supervised Learning