Gradient Coding in Decentralized Learning for Evading Stragglers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Chengxi, Skoglund, Mikael
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909223804207104
author Li, Chengxi
Skoglund, Mikael
author_facet Li, Chengxi
Skoglund, Mikael
contents In this paper, we consider a decentralized learning problem in the presence of stragglers. Although gradient coding techniques have been developed for distributed learning to evade stragglers, where the devices send encoded gradients with redundant training data, it is difficult to apply those techniques directly to decentralized learning scenarios. To deal with this problem, we propose a new gossip-based decentralized learning method with gradient coding (GOCO). In the proposed method, to avoid the negative impact of stragglers, the parameter vectors are updated locally using encoded gradients based on the framework of stochastic gradient coding and then averaged in a gossip-based manner. We analyze the convergence performance of GOCO for strongly convex loss functions. And we also provide simulation results to demonstrate the superiority of the proposed method in terms of learning performance compared with the baseline methods.
format Preprint
id arxiv_https___arxiv_org_abs_2402_04193
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Gradient Coding in Decentralized Learning for Evading Stragglers
Li, Chengxi
Skoglund, Mikael
Machine Learning
Signal Processing
In this paper, we consider a decentralized learning problem in the presence of stragglers. Although gradient coding techniques have been developed for distributed learning to evade stragglers, where the devices send encoded gradients with redundant training data, it is difficult to apply those techniques directly to decentralized learning scenarios. To deal with this problem, we propose a new gossip-based decentralized learning method with gradient coding (GOCO). In the proposed method, to avoid the negative impact of stragglers, the parameter vectors are updated locally using encoded gradients based on the framework of stochastic gradient coding and then averaged in a gossip-based manner. We analyze the convergence performance of GOCO for strongly convex loss functions. And we also provide simulation results to demonstrate the superiority of the proposed method in terms of learning performance compared with the baseline methods.
title Gradient Coding in Decentralized Learning for Evading Stragglers
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2402.04193