Adaptive collaboration for online personalized distributed learning with heterogeneous clients

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Philippenko, Constantin, Bars, Batiste Le, Scaman, Kevin, Massoulié, Laurent
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918087563935744
author Philippenko, Constantin
Bars, Batiste Le
Scaman, Kevin
Massoulié, Laurent
author_facet Philippenko, Constantin
Bars, Batiste Le
Scaman, Kevin
Massoulié, Laurent
contents We study the problem of online personalized decentralized learning with $N$ statistically heterogeneous clients collaborating to accelerate local training. An important challenge in this setting is to select relevant collaborators to reduce gradient variance while mitigating the introduced bias. To tackle this, we introduce a gradient-based collaboration criterion, allowing each client to dynamically select peers with similar gradients during the optimization process. Our criterion is motivated by a refined and more general theoretical analysis of the All-for-one algorithm, proved to be optimal in Even et al. (2022) for an oracle collaboration scheme. We derive excess loss upper-bounds for smooth objective functions, being either strongly convex, non-convex, or satisfying the Polyak-Lojasiewicz condition; our analysis reveals that the algorithm acts as a variance reduction method where the speed-up depends on a sufficient variance. We put forward two collaboration methods instantiating the proposed general schema; and we show that one variant preserves the optimality of All-for-one. We validate our results with experiments on synthetic and real datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2507_06844
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive collaboration for online personalized distributed learning with heterogeneous clients
Philippenko, Constantin
Bars, Batiste Le
Scaman, Kevin
Massoulié, Laurent
Machine Learning
We study the problem of online personalized decentralized learning with $N$ statistically heterogeneous clients collaborating to accelerate local training. An important challenge in this setting is to select relevant collaborators to reduce gradient variance while mitigating the introduced bias. To tackle this, we introduce a gradient-based collaboration criterion, allowing each client to dynamically select peers with similar gradients during the optimization process. Our criterion is motivated by a refined and more general theoretical analysis of the All-for-one algorithm, proved to be optimal in Even et al. (2022) for an oracle collaboration scheme. We derive excess loss upper-bounds for smooth objective functions, being either strongly convex, non-convex, or satisfying the Polyak-Lojasiewicz condition; our analysis reveals that the algorithm acts as a variance reduction method where the speed-up depends on a sufficient variance. We put forward two collaboration methods instantiating the proposed general schema; and we show that one variant preserves the optimality of All-for-one. We validate our results with experiments on synthetic and real datasets.
title Adaptive collaboration for online personalized distributed learning with heterogeneous clients
topic Machine Learning
url https://arxiv.org/abs/2507.06844