Byzantine Machine Learning Made Easy by Resilient Averaging of Momentums
This work addresses a critical reliability issue in distributed ML systems, offering a unified solution that is more practical and easier to verify compared to existing fragmented approaches.
The paper tackles the challenge of ensuring convergence in distributed machine learning despite misbehaving (Byzantine) workers by introducing RESAM, a framework that simplifies achieving optimal Byzantine resilience using only standard assumptions, and demonstrates its effectiveness through theoretical convergence guarantees and empirical evaluation.
Byzantine resilience emerged as a prominent topic within the distributed machine learning community. Essentially, the goal is to enhance distributed optimization algorithms, such as distributed SGD, in a way that guarantees convergence despite the presence of some misbehaving (a.k.a., {\em Byzantine}) workers. Although a myriad of techniques addressing the problem have been proposed, the field arguably rests on fragile foundations. These techniques are hard to prove correct and rely on assumptions that are (a) quite unrealistic, i.e., often violated in practice, and (b) heterogeneous, i.e., making it difficult to compare approaches. We present \emph{RESAM (RESilient Averaging of Momentums)}, a unified framework that makes it simple to establish optimal Byzantine resilience, relying only on standard machine learning assumptions. Our framework is mainly composed of two operators: \emph{resilient averaging} at the server and \emph{distributed momentum} at the workers. We prove a general theorem stating the convergence of distributed SGD under RESAM. Interestingly, demonstrating and comparing the convergence of many existing techniques become direct corollaries of our theorem, without resorting to stringent assumptions. We also present an empirical evaluation of the practical relevance of RESAM.