NeRCC: Nested-Regression Coded Computing for Resilient Distributed Prediction Serving Systems

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Moradi, Parsa, Maddah-Ali, Mohammad Ali
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929238090711040
author Moradi, Parsa
Maddah-Ali, Mohammad Ali
author_facet Moradi, Parsa
Maddah-Ali, Mohammad Ali
contents Resilience against stragglers is a critical element of prediction serving systems, tasked with executing inferences on input data for a pre-trained machine-learning model. In this paper, we propose NeRCC, as a general straggler-resistant framework for approximate coded computing. NeRCC includes three layers: (1) encoding regression and sampling, which generates coded data points, as a combination of original data points, (2) computing, in which a cluster of workers run inference on the coded data points, (3) decoding regression and sampling, which approximately recovers the predictions of the original data points from the available predictions on the coded data points. We argue that the overall objective of the framework reveals an underlying interconnection between two regression models in the encoding and decoding layers. We propose a solution to the nested regressions problem by summarizing their dependence on two regularization terms that are jointly optimized. Our extensive experiments on different datasets and various machine learning models, including LeNet5, RepVGG, and Vision Transformer (ViT), demonstrate that NeRCC accurately approximates the original predictions in a wide range of stragglers, outperforming the state-of-the-art by up to 23%.
format Preprint
id arxiv_https___arxiv_org_abs_2402_04377
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle NeRCC: Nested-Regression Coded Computing for Resilient Distributed Prediction Serving Systems
Moradi, Parsa
Maddah-Ali, Mohammad Ali
Machine Learning
Distributed, Parallel, and Cluster Computing
Information Theory
Resilience against stragglers is a critical element of prediction serving systems, tasked with executing inferences on input data for a pre-trained machine-learning model. In this paper, we propose NeRCC, as a general straggler-resistant framework for approximate coded computing. NeRCC includes three layers: (1) encoding regression and sampling, which generates coded data points, as a combination of original data points, (2) computing, in which a cluster of workers run inference on the coded data points, (3) decoding regression and sampling, which approximately recovers the predictions of the original data points from the available predictions on the coded data points. We argue that the overall objective of the framework reveals an underlying interconnection between two regression models in the encoding and decoding layers. We propose a solution to the nested regressions problem by summarizing their dependence on two regularization terms that are jointly optimized. Our extensive experiments on different datasets and various machine learning models, including LeNet5, RepVGG, and Vision Transformer (ViT), demonstrate that NeRCC accurately approximates the original predictions in a wide range of stragglers, outperforming the state-of-the-art by up to 23%.
title NeRCC: Nested-Regression Coded Computing for Resilient Distributed Prediction Serving Systems
topic Machine Learning
Distributed, Parallel, and Cluster Computing
Information Theory
url https://arxiv.org/abs/2402.04377