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

Weiyu Chen, Baijiong Lin, Xiaoyuan Zhang, Xi Lin, Han Zhao +2 more
1/19/2025
cs.LGstat.ML

Official code: Baijiong-Lin/Awesome-Multi-Objective-Deep-Learning (120 stars, last commit 2025-08-08). 1 more implementation below.

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.

View on arXivView PDF

Code Implementations(2)

A comprehensive list of gradient-based multi-objective optimization algorithms in deep learning.

12014Jan 9, 20251 years ago
multi-objective-learningmulti-objective-optimizationmulti-task-learningmultitask-learning
PyTorchMIT

A PyTorch Library for Multi-Task Learning

2,573232PythonDec 18, 20211 years agoMIT
deep-learningmmoemtlmulti-domain-learningmulti-objective-optimization+6 more

Similar Papers

Cite this paper

@article{chen2025gradientbased,
  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},
  year   = {2025},
  eprint = {2501.10945},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2501.10945v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2501.10945.svg)](https://codeofpaper.com/paper/2501.10945)

Discussion