Towards Bias Correction of FedAvg over Nonuniform and Time-Varying Communications

TL;DR

This paper addresses the challenge of communication failures in federated learning with non-uniform, time-varying links, proposing a new algorithm FedPBC that ensures convergence despite these issues.

Contribution

The paper introduces FedPBC, a simple variant of FedAvg that postpones broadcasting to handle non-uniform, time-varying communication failures, ensuring convergence.

Findings

FedPBC converges to a stationary point of the original objective.

Postponing broadcasts enables implicit gossiping among clients.

Theoretical bounds on model perturbation are established.

Abstract

Federated learning (FL) is a decentralized learning framework wherein a parameter server (PS) and a collection of clients collaboratively train a model via minimizing a global objective. Communication bandwidth is a scarce resource; in each round, the PS aggregates the updates from a subset of clients only. In this paper, we focus on non-convex minimization that is vulnerable to non-uniform and time-varying communication failures between the PS and the clients. Specifically, in each round $t$, the link between the PS and client $i$ is active with probability $p_i^t$, which is $\textit{unknown}$ to both the PS and the clients. This arises when the channel conditions are heterogeneous across clients and are changing over time. We show that when the $p_i^t$'s are not uniform, $\textit{Federated Average}$ (FedAvg) -- the most widely adopted FL algorithm -- fails to minimize the global objective. Observing this, we propose $\textit{Federated Postponed Broadcast}$ (FedPBC) which is a simple variant of FedAvg. It differs from FedAvg in that the PS postpones broadcasting the global model till the end of each round. We show that FedPBC converges to a stationary point of the original objective. The introduced staleness is mild and there is no noticeable slowdown. Both theoretical analysis and numerical results are provided. On the technical front, postponing the global model broadcasts enables implicit gossiping among the clients with active links at round $t$. Despite $p_i^t$'s are time-varying, we are able to bound the perturbation of the global model dynamics via the techniques of controlling the gossip-type information mixing errors.