Organizations: Department of Information Engineering, The Chinese University of Hong Kong · National Key Laboratory on Wireless Communications, University of Electronic Science and Technology of China
Noisy labels in distributed datasets induce severe local overfitting and consequently compromise the global model in federated learning (FL). Most existing solutions rely on selecting clean devices or aligning with public clean datasets, rather than endowing the model itself with robustness. In this paper, we propose FedCova, a dependency-free federated covariance learning framework that eliminates such external reliances by enhancing the model's intrinsic robustness via a new perspective on feature covariances. Specifically, FedCova encodes data into a discriminative but resilient feature space to tolerate label noise. Built on mutual information maximization, we design a novel objective for federated lossy feature encoding that relies solely on class feature covariances with an error tolerance term. Leveraging feature subspaces characterized by covariances, we construct a subspace-augmented federated classifier. FedCova unifies three key processes through the covariance: (1) training the network for feature encoding, (2) constructing a classifier directly from the learned features, and (3) correcting noisy labels based on feature subspaces. We implement FedCova across both symmetric and asymmetric noisy settings under heterogeneous data distribution. Experimental results on CIFAR-10/100 and real-world noisy dataset Clothing1M demonstrate the superior robustness of FedCova compared with the state-of-the-art methods.
Figures & tables
Figure 1: Overview of the FedCova framework. The green data is clean, while the red data is mislabeled. The pink rectangle indicates covariance-aware feature learning. The gray rectangle with the circle of blue arrows I - IV indicates the conventional FL processes, around which the circle of red arrows 1 - 6 in FedCova constructs a fortress to guard against label noise under the flow of the covariances Σ . Specifically, the server first broadcasts I the global model and 1 global classifier to edge devices. 2 Label correction can be conducted then, after which devices II 3 perform local feature learning and update III the local models and 4 5 local classifiers to the server. The latter then IV 6 aggregates them for the next rounds of iterations.
Configuration
Accuracy (%)
FedCova
80.71 ± 0.56
FedCova w/o ext. corrector
77.88 ± 1.47
FedCova w/o subs. augment.
79.58 ± 0.60
FedCova w/o classifier align.
76.86 ± 0.30
FedCova w/o error tolerance
69.47 ± 0.51
FedCova w/o zero mean
76.93 ± 0.34
Table 5: Effect of FedCova’s key components on CIFAR-10 (non-i.i.d.) under the noise level (ρ,τ)=(0.6,0.7) .
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 2: Effect of adding error tolerance term to the covariance matrix on the probability distributions.
Figure 3: Effect of ϵ2 illustrated as sample distribution and axis projections.
Figure 5: Visualization of different bilevel noise settings.
Figure 6: Accuracy curves versus communication rounds over baselines.
Figure 7: Orthogonality and accuracy curves versus communication rounds.
Figure 8: Correction performance and accuracy versus communication rounds.
Figure 9: Correction performance on devices.
Method
FedAvg
CoteachingFL
DivideMixFL
RoFL
FedCorr
FedNoRo
Runtime
290
1217
1508
383
643
664
Ratio
1.0 ×
4.2 ×
5.2 ×
1.3 ×
2.2 ×
2.3 ×
Appendix
Table 6: Total runtime comparison of different methods (minutes).
Figure 10: Accuracy under different error tolerance ϵ2 .
Figure 11: Accuracy under different augmentation coefficient α .
Figure 12: Accuracy under different feature dimension d .
Figure 13: Correction performance and accuracy under different data heterogeneity settings.
Federated learning (FL) enables collaborative model training without sharing raw data; however, the presence of noisy labels across distributed clients can severely degrade the learning performance. In this paper, we propose FedSIR, a multi-stage framework for robust FL under noisy labels. Different from existing approaches that mainly rely on designing noise-tolerant loss functions or exploiting loss dynamics during training, our method leverages the spectral structure of client feature representations to identify and mitigate label noise. Our framework consists of three key components. First, we identify clean and noisy clients by analyzing the spectral consistency of class-wise feature subspaces with minimal communication overhead. Second, clean clients provide spectral references that enable noisy clients to relabel potentially corrupted samples using both dominant class directions and residual subspaces. Third, we employ a noise-aware training strategy that integrates logit-adjusted loss, knowledge distillation, and distance-aware aggregation to further stabilize federated optimization. Extensive experiments on standard FL benchmarks demonstrate that FedSIR consistently outperforms state-of-the-art methods for FL with noisy labels. The code is available at https://github.com/sinagh72/FedSIR.
Sina Gholami, Abdulmoneam Ali, Tania Haghighi +2
University of North Carolina at Charlotte Charlotte, NC, USA
Federated learning (FL) suffers from performance degradation due to the inevitable presence of noisy annotations in distributed scenarios. Existing approaches have advanced in distinguishing noisy samples from the dataset for label correction by leveraging loss values. However, noisy samples recognition relying on scalar loss lacks reliability for FL under heterogeneous scenarios. In this paper, we rethink this paradigm from a representation perspective and propose \method~(Federated under Representation Gemometry), which follows the principle of ``representation geometry priority'' to recognize noisy labels. Firstly, \methodcreates label-agnostic spherical representations by using self-supervision. It then iteratively fits a spherical von Mises-Fisher (vMF) mixture model to this geometry using previously identified clean samples to capture semantic clusters. This geometric evidence is integrated with a semantic-label soft mapping mechanism to derive a distribution divergence between the label-free and annotated label-conditioned feature space, which robustly identifies noisy samples and updates the vMF mixture model with the newly separated clean dataset. Lastly, we employ an additional personalized noise absorption matrix on noisy labels to achieve robust optimization. Extensive experimental results demonstrate that \methodsignificantly outperforms state-of-the-art methods for FL with data heterogeneity under diverse noisy clients scenarios.
Tian Wen, Zhiqin Yang, Yonggang Zhang +4
Institute of Computing Technology, Chinese Academy of Sciences · TMLR Group, Hong Kong Baptist University · The Hong Kong University of Science and Technology +1
Personalized Federated Learning (PFL) aims to learn multiple task-specific models rather than a single global model across heterogeneous data distributions. Existing PFL approaches typically rely on iterative optimization-such as model update trajectories-to cluster users that need to accomplish the same tasks together. However, these learning-dynamics-based methods are inherently vulnerable to low-quality data and noisy labels, as corrupted updates distort clustering decisions and degrade personalization performance. To tackle this, we propose FB-NLL, a feature-centric framework that decouples user clustering from iterative training dynamics. By exploiting the intrinsic heterogeneity of local feature spaces, FB-NLL characterizes each user through the spectral structure of the covariances of their feature representations and leverages subspace similarity to identify task-consistent user groupings. This geometry-aware clustering is label-agnostic and is performed in a one-shot manner prior to training, significantly reducing communication overhead and computational costs compared to iterative baselines. Complementing this, we introduce a feature-consistency-based detection and correction strategy to address noisy labels within clusters. By leveraging directional alignment in the learned feature space and assigning labels based on class-specific feature subspaces, our method mitigates corrupted supervision without requiring estimation of stochastic noise transition matrices. In addition, FB-NLL is model-independent and integrates seamlessly with existing noise-robust training techniques. Extensive experiments across diverse datasets and noise regimes demonstrate that our framework consistently outperforms state-of-the-art baselines in terms of average accuracy and performance stability.
Abdulmoneam Ali, Ahmed Arafa
Department of Electrical and Computer Engineering, University of North Carolina at Charlotte, NC 28223, USA