Understanding Generalization of Federated Learning via Stability: Heterogeneity Matters

Zhenyu Sun, Xiaochun Niu, Ermin Wei

Generalization performance is a key metric in evaluating machine learning models when applied to real-world applications. Good generalization indicates the model can predict unseen…