Enhancing Generalization Robustness of Federated Learning in Highly Heterogeneous Environments

Wei WAN,Shengshan HU,Jianrong LU,Minghui LI,Ziqi ZHOU,Hai JIN
DOI: https://doi.org/10.1360/ssi-2023-0107
2024-01-01
Abstract:Federated learning (FL) is a distributed processing network that focuses on protecting client privacy data, providing a promising solution for addressing privacy leakage issues.However, a major quagmire in FL is to train clients' models over significantly non-independent and identically distributed (non-IID) data, which would lead to a low-performance global model.Although this issue has been investigated by many previous works, this paper finds that they have little or no performance improvement over the standard baseline FedAvg when facing highly non-IID data, unstable client participation, and deep models, seriously hindering the privacy protection application value of FL.To address this issue, a new solution called FedUp has been proposed.FedUp is a robust optimization solution for non-IID FL that improves the generalization robustness of the global model while retaining the privacy protection characteristics of FL.FedUp minimizes the upper bound of the global empirical loss function to ensure that the models exhibit smaller generalization errors.Simulation experiments show that FedUp achieves significant advantages over state-of-the-art methods, and is robust to highly non-IID data as well as unstable and large-cohort client participation.This solution has the potential to improve the performance of FL and make it more practical for privacy protection applications.
What problem does this paper attempt to address?