mlsanityData to 7 Oct 2026

Gradient-Based Multi-Objective Deep Learning: Algorithms, Theories, Applications, and Beyond

arXiv 2501.10945 ↗First posted 19 Jan 2025cs.LG stat.MLcited by 15

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

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
  3. 2014Deep Learning Face Attributes in the Wildcited by 3,679
  4. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
  5. 2018DARTS: Differentiable Architecture Searchcited by 1,782
  6. 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
  7. 2020Gradient Surgery for Multi-Task Learningcited by 1,210
  8. 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
  9. 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,168
  10. 2016HyperNetworkscited by 1,003
  11. 2017A Survey on Multi-Task Learningcited by 838
  12. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
38 more
  1. 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
  2. 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 791
  3. 2020XTREME: A Massively Multilingual Multi-task Benchmark for Evaluating Cross-lingual Generalizationcited by 738
  4. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
  5. 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
  6. 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
  7. 2018End-to-End Multi-Task Learning with Attentioncited by 533
  8. 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
  9. 2018FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Searchcited by 480
  10. 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
  11. 2023Fine-Grained Human Feedback Gives Better Rewards for Language Model Trainingcited by 399
  12. 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
  13. 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
  14. 2023Personalized Soups: Personalized Large Language Model Alignment via Post-hoc Parameter Mergingcited by 271
  15. 2020A Comprehensive Survey of Neural Architecture Search: Challenges and Solutionscited by 253
  16. 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
  17. 2023Efficient Diffusion Training via Min-SNR Weighting Strategycited by 242
  18. 2024A Survey on Mixture of Experts in Large Language Modelscited by 233
  19. 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
  20. 2024HelpSteer2: Open-source dataset for training top-performing reward modelscited by 232
  21. 2019Pareto Multi-Task Learningcited by 224
  22. 2022Multi-Task Learning as a Bargaining Gamecited by 209
  23. 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 199
  24. 2020Differentiable Expected Hypervolume Improvement for Parallel Multi-Objective Bayesian Optimizationcited by 176
  25. 2024Rewards-in-Context: Multi-objective Alignment of Foundation Models with Dynamic Preference Adjustmentcited by 171
  26. 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 167
  27. 2020Minimax Pareto Fairness: A Multi Objective Perspectivecited by 152
  28. 2024Arithmetic Control of LLMs for Diverse User Preferences: Directional Preference Alignment with Multi-Objective Rewardscited by 145
  29. 2023Unexpected Improvements to Expected Improvement for Bayesian Optimizationcited by 144
  30. 2020Learning the Pareto Front with Hypernetworkscited by 122
  31. 2023Beyond One-Preference-Fits-All Alignment: Multi-Objective Direct Preference Optimizationcited by 121
  32. 2021Parallel Bayesian Optimization of Multiple Noisy Objectives with Expected Hypervolume Improvementcited by 115
  33. 2024Controllable Preference Optimization: Toward Controllable Multi-Objective Alignmentcited by 105
  34. 2021Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learningcited by 100
  35. 2022Multi-Objective GFlowNetscited by 100
  36. 2018A Flexible Framework for Multi-Objective Bayesian Optimization using Random Scalarizationscited by 98
  37. 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 90
  38. 2022InvPT: Inverted Pyramid Multi-task Transformer for Dense Scene Understandingcited by 89

50 most cited of 143 All 143 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 9; 2026: 6.

15 papers.

Sort:

By area:

  1. 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Drivingcited by 39
  2. 2025PARM: Multi-Objective Test-Time Alignment via Preference-Aware Autoregressive Reward Modelcited by 27
  3. 2025Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasetscited by 17
  4. 2025TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Predictioncited by 9
  5. 2026Advancing General-Purpose Reasoning Models with Modular Gradient Surgerycited by 9
  6. 2025Rethinking Parameter Sharing for LLM Fine-Tuning with Multiple LoRAscited by 3
  7. 2025Addressing Gradient Misalignment in Data-Augmented Training for Robust Speech Deepfake Detectioncited by 2
  8. 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspectivecited by 2
  9. 2025SAMO: A Lightweight Sharpness-Aware Approach for Multi-Task Optimization with Joint Global-Local Perturbationcited by 1
  10. 2025A Framework for Controllable Multi-objective Learning with Annealed Stein Variational Hypernetworkscited by 0
  11. 2026Barrier-enforced multi-objective optimization for direct point and sharp interval forecastingcited by 0
  12. 2026MGDA-Decoupled: Geometry-Aware Multi-Objective Optimisation for DPO-based LLM Alignmentcited by 0
3 more
  1. 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimizationcited by 0
  2. 2026MAdam: Metric-Aware Multi-Objective Adamcited by 0
  3. 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Levelcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2024A Decade of Deep Learning: A Survey on The Magnificent Seven
  2. 2025Pareto Multi-Objective Alignment for Language Models
  3. 2025Aligned Multi Objective Optimization
  4. 2026Gradient Smoothing: Coupling Layer-wise Updates for Improved Optimization
  5. 2026Multi-Objective Hierarchical Optimization with Large Language Models
  6. 2025An LLM-Empowered Adaptive Evolutionary Algorithm For Multi-Component Deep Learning Systems
  7. 2023An Invitation to Deep Reinforcement Learning
  8. 2026Multi-Objective Learning for Diffusion Models: A Statistical Theory under Semi-Supervised Learning