FedFN: Feature Normalization for Alleviating Data Heterogeneity Problem in Federated Learning

Seongyoon Kim,Gihun Lee,Jaehoon Oh,Se-Young Yun
2023-11-22
Abstract:Federated Learning (FL) is a collaborative method for training models while preserving data privacy in decentralized settings. However, FL encounters challenges related to data heterogeneity, which can result in performance degradation. In our study, we observe that as data heterogeneity increases, feature representation in the FedAVG model deteriorates more significantly compared to classifier weight. Additionally, we observe that as data heterogeneity increases, the gap between higher feature norms for observed classes, obtained from local models, and feature norms of unobserved classes widens, in contrast to the behavior of classifier weight norms. This widening gap extends to encompass the feature norm disparities between local and the global models. To address these issues, we introduce Federated Averaging with Feature Normalization Update (FedFN), a straightforward learning method. We demonstrate the superior performance of FedFN through extensive experiments, even when applied to pretrained ResNet18. Subsequently, we confirm the applicability of FedFN to foundation models.
Machine Learning,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper attempts to address the issue of performance degradation caused by data heterogeneity in Federated Learning (FL). Specifically, the authors observe that as data heterogeneity increases, the degradation of feature representation in the FedAvg model is more significant than that of the classifier weight. Additionally, they find that with increasing data heterogeneity, the gap between the feature norms of observed and unobserved classes in local models gradually widens, and this gap also extends to the feature norm differences between local models and the global model. To tackle these issues, the authors introduce a new method—Federated Averaging with Feature Normalization Update (FedFN). Through extensive experiments, the authors demonstrate the superior performance of FedFN in handling highly heterogeneous data and validate its applicability in both pre-trained and base models.