Learning from History for Byzantine Robust Optimization
Sai Praneeth Karimireddy, Lie He, Martin Jaggi
Abstract
Byzantine robustness has received significant attention recently given its importance for distributed and federated learning. In spite of this, we identify severe flaws in existing algorithms even when the data across the participants is identically distributed. First, we show realistic examples where current state of the art robust aggregation rules fail to converge even in the absence of any Byzantine attackers. Secondly, we prove that even if the aggregation rules may succeed in limiting the influence of the attackers in a single round, the attackers can couple their attacks across time eventually leading to divergence. To address these issues, we present two surprisingly simple strategies: a new robust iterative clipping procedure, and incorporating worker momentum to overcome time-coupled attacks. This is the first provably robust method for the standard stochastic optimization setting. Our code is open sourced at https://github.com/epfml/byzantine-robust-optimizer.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c9ef75c6-2986-4cc0-ae06-23d04a2d17e0Cited by top-tier papers52
- Byzantine-Robust Learning on Heterogeneous Datasets via BucketingSai Praneeth Karimireddy, Lie He, Martin JaggiICLR 2022 · 192 citations
- Breaking the centralized barrier for cross-device federated learningSai Praneeth Karimireddy, Martin Jaggi, Satyen Kale, Mehryar Mohri et al.NeurIPS 2021 · 113 citations
- Byzantine Machine Learning Made Easy By Resilient Averaging of MomentumsSadegh Farhadkhani, Rachid Guerraoui, Nirupam Gupta, Rafael Pinot et al.ICML 2022 · 96 citations
- High-Probability Bounds for Stochastic Optimization and Variational Inequalities: the Case of Unbounded VarianceAbdurakhmon Sadiev, Marina Danilova, Eduard Gorbunov, Samuel Horváth et al.ICML 2023 · 68 citations
- Learning to Attack Federated Learning: A Model-based Reinforcement Learning Attack FrameworkHenger Li, Xiaolin Sun, Zizhan ZhengNeurIPS 2022 · 55 citations
Builds on7
- Practical Secure Aggregation for Privacy-Preserving Machine LearningKallista A. Bonawitz, Vladimir Ivanov, Ben Kreuter, Antonio Marcedone et al.CCS 2017 · 3,936 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Attack of the Tails: Yes, You Really Can Backdoor Federated LearningHongyi Wang, Kartik Sreenivasan, Shashank Rajput, Harit Vishwakarma et al.NeurIPS 2020 · 862 citations
- An Improved Analysis of Stochastic Gradient Descent with MomentumYanli Liu, Yuan Gao, Wotao YinNeurIPS 2020 · 328 citations
- Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient ClippingEduard Gorbunov, Marina Danilova, Alexander V. GasnikovNeurIPS 2020 · 181 citations
Related papers
- Simple Minimax Optimal Byzantine Robust Algorithm for Nonconvex Objectives with Uniform Gradient HeterogeneityTomoya Murata, Kenta Niwa, Takumi Fukami, Iifan TyouICLR 2024
- Byzantine-Robust Learning on Heterogeneous Data via Gradient SplittingYuchen Liu, Chen Chen, Lingjuan Lyu, Fangzhao Wu et al.ICML 2023 · 27 citations
- Delayed Momentum Aggregation: Communication-efficient Byzantine-robust Federated Learning with Partial ParticipationKaoru Otsuka, Yuki Takezawa, Makoto YamadaICML 2026
- Dynamic Byzantine-Robust Learning: Adapting to Switching Byzantine WorkersRon Dorfman, Naseem Yehya, Kfir Yehuda LevyICML 2024 · 5 citations
- On the Tension between Byzantine Robustness and No-Attack Accuracy in Distributed LearningYi-Rui Yang, Chang-Wei Shi, Wu-Jun LiICML 2025
