PMGDA: A Preference-based Multiple Gradient Descent Algorithm

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Xiaoyuan, Lin, Xi, Zhang, Qingfu
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914681252216832
author Zhang, Xiaoyuan
Lin, Xi
Zhang, Qingfu
author_facet Zhang, Xiaoyuan
Lin, Xi
Zhang, Qingfu
contents It is desirable in many multi-objective machine learning applications, such as multi-task learning with conflicting objectives and multi-objective reinforcement learning, to find a Pareto solution that can match a given preference of a decision maker. These problems are often large-scale with available gradient information but cannot be handled very well by the existing algorithms. To tackle this critical issue, this paper proposes a novel predict-and-correct framework for locating a Pareto solution that fits the preference of a decision maker. In the proposed framework, a constraint function is introduced in the search progress to align the solution with a user-specific preference, which can be optimized simultaneously with multiple objective functions. Experimental results show that our proposed method can efficiently find a particular Pareto solution under the demand of a decision maker for standard multiobjective benchmark, multi-task learning, and multi-objective reinforcement learning problems with more than thousands of decision variables. Code is available at: https://github.com/xzhang2523/pmgda. Our code is current provided in the pgmda.rar attached file and will be open-sourced after publication.}
format Preprint
id arxiv_https___arxiv_org_abs_2402_09492
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PMGDA: A Preference-based Multiple Gradient Descent Algorithm
Zhang, Xiaoyuan
Lin, Xi
Zhang, Qingfu
Machine Learning
It is desirable in many multi-objective machine learning applications, such as multi-task learning with conflicting objectives and multi-objective reinforcement learning, to find a Pareto solution that can match a given preference of a decision maker. These problems are often large-scale with available gradient information but cannot be handled very well by the existing algorithms. To tackle this critical issue, this paper proposes a novel predict-and-correct framework for locating a Pareto solution that fits the preference of a decision maker. In the proposed framework, a constraint function is introduced in the search progress to align the solution with a user-specific preference, which can be optimized simultaneously with multiple objective functions. Experimental results show that our proposed method can efficiently find a particular Pareto solution under the demand of a decision maker for standard multiobjective benchmark, multi-task learning, and multi-objective reinforcement learning problems with more than thousands of decision variables. Code is available at: https://github.com/xzhang2523/pmgda. Our code is current provided in the pgmda.rar attached file and will be open-sourced after publication.}
title PMGDA: A Preference-based Multiple Gradient Descent Algorithm
topic Machine Learning
url https://arxiv.org/abs/2402.09492