Probabilistic Gradient Coding via Structure-Preserving Sparsification

TL;DR

This paper introduces two new probabilistic gradient coding schemes, SG and EP, that extend the applicability of gradient coding in distributed systems by preserving key structures and spectral properties.

Contribution

The paper proposes two novel probabilistic gradient codes, SG and EP, which overcome limitations of existing BIBD codes by enabling broader system parameter ranges.

Findings

Both codes achieve error performance comparable to BIBD codes.

They significantly extend the feasible system parameters beyond existing codes.

Experimental results demonstrate practical effectiveness in large-scale distributed tasks.

Abstract

Gradient coding is a distributed computing technique aiming to provide robustness against slow or non-responsive computing nodes, known as stragglers, while balancing the computational load for responsive computing nodes. Among existing gradient codes, a construction based on combinatorial designs, called BIBD gradient code, achieves the best trade-off between robustness and computational load in the worst-case adversarial straggler setting. However, the range of system parameters for which BIBD gradient codes exist is limited. In this paper, we overcome these limitations by proposing two new probabilistic gradient codes, termed the \emph{Sparse Gaussian} (SG) gradient code and the \emph{Expansion-Preserving} (EP) gradient code. Through probabilistic constructions, the former preserves the combinatorial structure of BIBDs, while the latter preserves key spectral properties. Both codes are based on a common two-step framework: first generating a random matrix and then applying distinct sparsification procedures. The SG gradient code constructs its encoding matrix from a correlated multivariate Gaussian distribution masked by Bernoulli random variables, while the EP gradient code derives its encoding matrix from sparsified expander-like graph structures that preserve key spectral properties. Experimentally, both codes achieve worst-case error performance comparable to that of the BIBD gradient code (when such a code with the same parameters exists). Moreover, they substantially extend the feasible range of system parameters beyond BIBD and soft BIBD gradient codes, offering practical and theoretically grounded solutions for large-scale distributed computing tasks.