Efficient Pareto Manifold Learning with Low-Rank Structure

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chen, Weiyu, Kwok, James T.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909274201915392
author Chen, Weiyu
Kwok, James T.
author_facet Chen, Weiyu
Kwok, James T.
contents Multi-task learning, which optimizes performance across multiple tasks, is inherently a multi-objective optimization problem. Various algorithms are developed to provide discrete trade-off solutions on the Pareto front. Recently, continuous Pareto front approximations using a linear combination of base networks have emerged as a compelling strategy. However, it suffers from scalability issues when the number of tasks is large. To address this issue, we propose a novel approach that integrates a main network with several low-rank matrices to efficiently learn the Pareto manifold. It significantly reduces the number of parameters and facilitates the extraction of shared features. We also introduce orthogonal regularization to further bolster performance. Extensive experimental results demonstrate that the proposed approach outperforms state-of-the-art baselines, especially on datasets with a large number of tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2407_20734
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient Pareto Manifold Learning with Low-Rank Structure
Chen, Weiyu
Kwok, James T.
Machine Learning
Multi-task learning, which optimizes performance across multiple tasks, is inherently a multi-objective optimization problem. Various algorithms are developed to provide discrete trade-off solutions on the Pareto front. Recently, continuous Pareto front approximations using a linear combination of base networks have emerged as a compelling strategy. However, it suffers from scalability issues when the number of tasks is large. To address this issue, we propose a novel approach that integrates a main network with several low-rank matrices to efficiently learn the Pareto manifold. It significantly reduces the number of parameters and facilitates the extraction of shared features. We also introduce orthogonal regularization to further bolster performance. Extensive experimental results demonstrate that the proposed approach outperforms state-of-the-art baselines, especially on datasets with a large number of tasks.
title Efficient Pareto Manifold Learning with Low-Rank Structure
topic Machine Learning
url https://arxiv.org/abs/2407.20734