Alert button

Bandit-based Communication-Efficient Client Selection Strategies for Federated Learning

Dec 14, 2020
Yae Jee Cho, Samarth Gupta, Gauri Joshi, Osman Yağan

Figure 1 for Bandit-based Communication-Efficient Client Selection Strategies for Federated Learning
Figure 2 for Bandit-based Communication-Efficient Client Selection Strategies for Federated Learning
Figure 3 for Bandit-based Communication-Efficient Client Selection Strategies for Federated Learning

Share this with someone who'll enjoy it:

Due to communication constraints and intermittent client availability in federated learning, only a subset of clients can participate in each training round. While most prior works assume uniform and unbiased client selection, recent work on biased client selection has shown that selecting clients with higher local losses can improve error convergence speed. However, previously proposed biased selection strategies either require additional communication cost for evaluating the exact local loss or utilize stale local loss, which can even make the model diverge. In this paper, we present a bandit-based communication-efficient client selection strategy UCB-CS that achieves faster convergence with lower communication overhead. We also demonstrate how client selection can be used to improve fairness.

View paper onarxiv icon

Share this with someone who'll enjoy it: