77
v1v2 (latest)

Fast and Robust Distributed Learning in High Dimension

Abstract

Could a gradient aggregation rule (GAR) for distributed machine learning be both robust and fast? This paper answers by the affirmative through multi-Bulyan. Given nn workers, ff of which are arbitrary malicious (Byzantine) and m=nfm=n-f are not, we prove that multi-Bulyan can ensure a strong form of Byzantine resilience, as well as an mn{\frac{m}{n}} slowdown, compared to averaging, the fastest (but non Byzantine resilient) rule for distributed machine learning. When mnm \approx n (almost all workers are correct), multi-Bulyan reaches the speed of averaging. We also prove that multi-Bulyan's cost in local computation is O(d)O(d) (like averaging), an important feature for ML where dd commonly reaches 10910^9, while robust alternatives have at least quadratic cost in dd. Our theoretical findings are complemented with an experimental evaluation which, in addition to supporting the linear O(d)O(d) complexity argument, conveys the fact that multi-Bulyan's parallelisability further adds to its efficiency.

View on arXiv
Comments on this paper