Federated Multi-Mini-Batch: An Efficient Training Approach to Federated Learning in Non-IID Environments
Machine Learning
2021-07-06 v2 Machine Learning
Abstract
Federated learning has faced performance and network communication challenges, especially in the environments where the data is not independent and identically distributed (IID) across the clients. To address the former challenge, we introduce the federated-centralized concordance property and show that the federated single-mini-batch training approach can achieve comparable performance as the corresponding centralized training in the Non-IID environments. To deal with the latter, we present the federated multi-mini-batch approach and illustrate that it can establish a trade-off between the performance and communication efficiency and outperforms federated averaging in the Non-IID settings.
Cite
@article{arxiv.2011.07006,
title = {Federated Multi-Mini-Batch: An Efficient Training Approach to Federated Learning in Non-IID Environments},
author = {Reza Nasirigerdeh and Mohammad Bakhtiari and Reihaneh Torkzadehmahani and Amirhossein Bayat and Markus List and David B. Blumenthal and Jan Baumbach},
journal= {arXiv preprint arXiv:2011.07006},
year = {2021}
}