Skip to yearly menu bar Skip to main content


Poster

Federated Learning: Lessons from Generalization Error Analysis

Milad Sefidgaran · Romain Chor · Abdellatif Zaidi · Yijun Wan


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 the PS. In our setup, the more the clients communicate with the PS the less data they use for local training in each round, in a manner 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 in their generality for 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 we 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 the 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 for every $R$, the generalization error of FSVM decreases faster than that of centralized learning by a factor of $\mathcal{O}(\sqrt{\log(K)/K})$. Furthermore, we also provide results of experiments that are 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$.

Live content is unavailable. Log in and register to view live content