Abstract:We investigate the generalization error of statistical learning models in a Federated Learning (FL) setting. Specifically, we study the evolution of the generalization error with the number of communication rounds $R$ between $K$ clients and a parameter server (PS), i.e., the effect on the generalization error of how often the clients' local models are aggregated at PS. In our setup, the more the clients communicate with PS the less data they use for local training in each round, such that the amount of training data per client is identical for distinct values of $R$. We establish PAC-Bayes and rate-distortion theoretic bounds on the generalization error that account explicitly for the effect of the number of rounds $R$, in addition to the number of participating devices $K$ and individual datasets size $n$. The bounds, which apply to a large class of loss functions and learning algorithms, appear to be the first of their kind for the FL setting. Furthermore, we apply our bounds to FL-type Support Vector Machines (FSVM); and derive (more) explicit bounds in this case. In particular, we show that the generalization bound of FSVM increases with $R$, suggesting that more frequent communication with PS diminishes the generalization power. This implies that the population risk decreases less fast with $R$ than does the empirical risk. Moreover, our bound suggests that the generalization error of FSVM decreases faster than that of centralized learning by a factor of $\mathcal{O}(\sqrt{\log(K)/K})$. Finally, we provide experimental results obtained using neural networks (ResNet-56) which show evidence that not only may our observations for FSVM hold more generally but also that the population risk may even start to increase beyond some value of $R$.

Lessons from Generalization Error Analysis of Federated Learning: You May Communicate Less Often!

FedPSE: Personalized Sparsification with Element-wise Aggregation for Federated Learning

Improved Generalization Bounds for Communication Efficient Federated Learning

Goal-Oriented Communications in Federated Learning via Feedback on Risk-Averse Participation

A Non-parametric View of FedAvg and FedProx: Beyond Stationary Points

Communication-Efficient Federated Learning: A Variance-Reduced Stochastic Approach with Adaptive Sparsification.

FedSpeed: Larger Local Interval, Less Communication Round, and Higher Generalization Accuracy

Can We Theoretically Quantify the Impacts of Local Updates on the Generalization Performance of Federated Learning?

Advocating for the Silent: Enhancing Federated Generalization for Non-Participating Clients

Sparse Training for Federated Learning with Regularized Error Correction

Non-IID Federated Learning with Sharper Risk Bound.

Adaptive Block-Wise Regularization and Knowledge Distillation for Enhancing Federated Learning

FLCP: federated learning framework with communication-efficient and privacy-preserving

Communication-Efficient Model Aggregation with Layer Divergence Feedback in Federated Learning

Generalization Bounds for Federated Learning: Fast Rates, Unparticipating Clients and Unbounded Losses

User-Centric Federated Learning: Trading off Wireless Resources for Personalization

Data-Quality-Driven Federated Learning for Optimizing Communication Costs

Achieving Statistical Optimality of Federated Learning: Beyond Stationary Points

Federated Learning via Variational Bayesian Inference: Personalization, Sparsity and Clustering

FedSL: A Communication Efficient Federated Learning with Split Layer Aggregation

Privacy-Preserving and Reliable Decentralized Federated Learning