dSTAR: Straggler Tolerant and Byzantine Resilient Distributed SGD

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yan, Jiahe, Chaudhari, Pratik, Kleinrock, Leonard
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912149984509952
author Yan, Jiahe
Chaudhari, Pratik
Kleinrock, Leonard
author_facet Yan, Jiahe
Chaudhari, Pratik
Kleinrock, Leonard
contents Distributed model training needs to be adapted to challenges such as the straggler effect and Byzantine attacks. When coordinating the training process with multiple computing nodes, ensuring timely and reliable gradient aggregation amidst network and system malfunctions is essential. To tackle these issues, we propose \textit{dSTAR}, a lightweight and efficient approach for distributed stochastic gradient descent (SGD) that enhances robustness and convergence. \textit{dSTAR} selectively aggregates gradients by collecting updates from the first \(k\) workers to respond, filtering them based on deviations calculated using an ensemble median. This method not only mitigates the impact of stragglers but also fortifies the model against Byzantine adversaries. We theoretically establish that \textit{dSTAR} is (\(α, f\))-Byzantine resilient and achieves a linear convergence rate. Empirical evaluations across various scenarios demonstrate that \textit{dSTAR} consistently maintains high accuracy, outperforming other Byzantine-resilient methods that often suffer up to a 40-50\% accuracy drop under attack. Our results highlight \textit{dSTAR} as a robust solution for training models in distributed environments prone to both straggler delays and Byzantine faults.
format Preprint
id arxiv_https___arxiv_org_abs_2412_07151
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle dSTAR: Straggler Tolerant and Byzantine Resilient Distributed SGD
Yan, Jiahe
Chaudhari, Pratik
Kleinrock, Leonard
Distributed, Parallel, and Cluster Computing
Distributed model training needs to be adapted to challenges such as the straggler effect and Byzantine attacks. When coordinating the training process with multiple computing nodes, ensuring timely and reliable gradient aggregation amidst network and system malfunctions is essential. To tackle these issues, we propose \textit{dSTAR}, a lightweight and efficient approach for distributed stochastic gradient descent (SGD) that enhances robustness and convergence. \textit{dSTAR} selectively aggregates gradients by collecting updates from the first \(k\) workers to respond, filtering them based on deviations calculated using an ensemble median. This method not only mitigates the impact of stragglers but also fortifies the model against Byzantine adversaries. We theoretically establish that \textit{dSTAR} is (\(α, f\))-Byzantine resilient and achieves a linear convergence rate. Empirical evaluations across various scenarios demonstrate that \textit{dSTAR} consistently maintains high accuracy, outperforming other Byzantine-resilient methods that often suffer up to a 40-50\% accuracy drop under attack. Our results highlight \textit{dSTAR} as a robust solution for training models in distributed environments prone to both straggler delays and Byzantine faults.
title dSTAR: Straggler Tolerant and Byzantine Resilient Distributed SGD
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2412.07151