Saved in:
Bibliographic Details
Main Authors: Even, Mathieu, Massoulié, Laurent
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2501.18975
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915148965347328
author Even, Mathieu
Massoulié, Laurent
author_facet Even, Mathieu
Massoulié, Laurent
contents Motivated by multi-task and meta-learning approaches, we consider the problem of learning structure shared by tasks or users, such as shared low-rank representations or clustered structures. While all previous works focus on well-specified linear regression, we consider more general convex objectives, where the structural low-rank and cluster assumptions are expressed on the optima of each function. We show that under mild assumptions such as \textit{Hessian concentration} and \textit{noise concentration at the optimum}, rank and clustered regularized estimators recover such structure, provided the number of samples per task and the number of tasks are large enough. We then study the problem of recovering the subspace in which all the solutions lie, in the setting where there is only a single sample per task: we show that in that case, the rank-constrained estimator can recover the subspace, but that the number of tasks needs to scale exponentially large with the dimension of the subspace. Finally, we provide a polynomial-time algorithm via nuclear norm constraints for learning a shared linear representation in the context of convex learning objectives.
format Preprint
id arxiv_https___arxiv_org_abs_2501_18975
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Meta-learning of shared linear representations beyond well-specified linear regression
Even, Mathieu
Massoulié, Laurent
Machine Learning
Motivated by multi-task and meta-learning approaches, we consider the problem of learning structure shared by tasks or users, such as shared low-rank representations or clustered structures. While all previous works focus on well-specified linear regression, we consider more general convex objectives, where the structural low-rank and cluster assumptions are expressed on the optima of each function. We show that under mild assumptions such as \textit{Hessian concentration} and \textit{noise concentration at the optimum}, rank and clustered regularized estimators recover such structure, provided the number of samples per task and the number of tasks are large enough. We then study the problem of recovering the subspace in which all the solutions lie, in the setting where there is only a single sample per task: we show that in that case, the rank-constrained estimator can recover the subspace, but that the number of tasks needs to scale exponentially large with the dimension of the subspace. Finally, we provide a polynomial-time algorithm via nuclear norm constraints for learning a shared linear representation in the context of convex learning objectives.
title Meta-learning of shared linear representations beyond well-specified linear regression
topic Machine Learning
url https://arxiv.org/abs/2501.18975