English

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

Machine Learning 2025-08-07 v3 Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2501.10945,
  title  = {Gradient-Based Multi-Objective Deep Learning: Algorithms, Theories, Applications, and Beyond},
  author = {Weiyu Chen and Baijiong Lin and Xiaoyuan Zhang and Xi Lin and Han Zhao and Qingfu Zhang and James T. Kwok},
  journal= {arXiv preprint arXiv:2501.10945},
  year   = {2025}
}