Federated Learning with Sample-level Client Drift Mitigation

TL;DR

This paper introduces FedBSS, a novel sample-level client drift mitigation method for federated learning that dynamically selects samples based on bias, improving performance under data heterogeneity, feature distribution skew, and noisy labels.

Contribution

The paper proposes a bias-aware sample selection scheme for federated learning that addresses data heterogeneity at the sample level, enhancing scalability and robustness.

Findings

FedBSS outperforms state-of-the-art baselines.

Effective in feature distribution skew and noisy label settings.

Reduces heterogeneity and improves model stability.

Abstract

Federated Learning (FL) suffers from severe performance degradation due to the data heterogeneity among clients. Existing works reveal that the fundamental reason is that data heterogeneity can cause client drift where the local model update deviates from the global one, and thus they usually tackle this problem from the perspective of calibrating the obtained local update. Despite effectiveness, existing methods substantially lack a deep understanding of how heterogeneous data samples contribute to the formation of client drift. In this paper, we bridge this gap by identifying that the drift can be viewed as a cumulative manifestation of biases present in all local samples and the bias between samples is different. Besides, the bias dynamically changes as the FL training progresses. Motivated by this, we propose FedBSS that first mitigates the heterogeneity issue in a sample-level manner, orthogonal to existing methods. Specifically, the core idea of our method is to adopt a bias-aware sample selection scheme that dynamically selects the samples from small biases to large epoch by epoch to train progressively the local model in each round. In order to ensure the stability of training, we set the diversified knowledge acquisition stage as the warm-up stage to avoid the local optimality caused by knowledge deviation in the early stage of the model. Evaluation results show that FedBSS outperforms state-of-the-art baselines. In addition, we also achieved effective results on feature distribution skew and noise label dataset setting, which proves that FedBSS can not only reduce heterogeneity, but also has scalability and robustness.