FedENLC: An End-to-End Noisy Label Correction Framework in Federated Learning

In this paper, we propose FedENLC, an end-to-end noisy label correction model that performs model training and label correction simultaneously to fundamentally mitigate the label noise problem of federated learning (FL). FedENLC consists of two stages. In the first stage, the proposed model employs Symmetric Cross Entropy (SCE), a robust loss function for noisy labels, and label smoothing to prevent the model from being biased by incorrect information in noisy environments. Subsequently, a Bayesian Gaussian Mixture Model (BGMM) is utilized to detect noisy clients. BGMM mitigates extreme parameter bias through its prior distribution, enabling stable and reliable detection in FL environments where data heterogeneity and noisy labels coexist. In the second stage, only the top noisy clients with high noise ratios are selectively included in the label correction process. The selection of top noisy clients is determined dynamically by considering the number of classes, posterior probabilities, and the degree of data heterogeneity. Through this approach, the proposed model prevents performance degradation caused by incorrect detection, while improving both computational efficiency and training stability. Experimental results show that FedENLC achieves significantly improved performance over existing models on the CIFAR-10 and CIFAR-100 datasets under data heterogeneity settings along with four noise settings.