BOBA: Byzantine-Robust Federated Learning with Label Skewness

• Published in: Proceedings of machine learning research Vol. 1; p. 892
• Main Authors: Bao, Wenxuan, Wu, Jun, He, Jingrui
• Format: Journal Article
• Language: English
• Published: United States 2024
• ISSN: 2640-3498; 2640-3498

In federated learning, most existing robust aggregation rules (AGRs) combat Byzantine attacks in the IID setting, where client data is assumed to be independent and identically distributed. In this paper, we address label skewness, a more realistic and challenging non-IID setting, where each client only has access to a few classes of data. In this setting, state-of-the-art AGRs suffer from selection bias, leading to significant performance drop for particular classes; they are also more vulnerable to Byzantine attacks due to the increased variation among gradients of honest clients. To address these limitations, we propose an efficient two-stage method named . Theoretically, we prove the convergence of BOBA with an error of the optimal order. Our empirical evaluations demonstrate BOBA's superior unbiasedness and robustness across diverse models and datasets when compared to various baselines. Our code is available at https://github.com/baowenxuan/BOBA.