Byzantine-Robust Learning on Heterogeneous Datasets via Bucketing
arXiv:2006.09365
Abstract
In Byzantine robust distributed or federated learning, a central server wants to train a machine learning model over data distributed across multiple workers. However, a fraction of these workers may deviate from the prescribed algorithm and send arbitrary messages. While this problem has received significant attention recently, most current defenses assume that the workers have identical data. For realistic cases when the data across workers are heterogeneous (non-iid), we design new attacks which circumvent current defenses, leading to significant loss of performance. We then propose a simple bucketing scheme that adapts existing robust algorithms to heterogeneous datasets at a negligible computational cost. We also theoretically and experimentally validate our approach, showing that combining bucketing with existing robust algorithms is effective against challenging attacks. Our work is the first to establish guaranteed convergence for the non-iid Byzantine robust problem under realistic assumptions.
arXiv commentv5 is the camera-ready version of this paper on ICLR 2022
References in corpus (16)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Robust Aggregation for Federated Learning
- How To Backdoor Federated Learning
- Byzantine-Robust Distributed Learning: Towards Optimal Statistical Rates
- The Hidden Vulnerability of Distributed Learning in Byzantium
- Can You Really Backdoor Federated Learning?
- Geometric median and robust estimation in Banach spaces
- Byzantine Stochastic Gradient Descent
- Error Feedback Fixes SignSGD and other Gradient Compression Schemes
- Robust Federated Learning in a Heterogeneous Environment
- signSGD with Majority Vote is Communication Efficient And Fault Tolerant
- Attack of the Tails: Yes, You Really Can Backdoor Federated Learning
- Zeno: Distributed Stochastic Gradient Descent with Suspicion-based Fault-tolerance
- Robust Training in High Dimensions via Block Coordinate Geometric Median Descent
- Implementation of a Practical Distributed Calculation System with Browsers and JavaScript, and Application to Distributed Deep Learning