The paper proposes a novel approach called Weighted Federated Averaging and Client Selection (WeiAvgCS) to address the issue of data heterogeneity and diversity in federated learning (FL). The key insights are:
Clients with more diverse data can improve the performance of the federated learning model. The diversity of a client's data is quantified by the variance of the label distribution.
To mitigate privacy concerns, the paper introduces an estimation of data diversity using a projection-based method, which has a strong correlation with the actual variance.
WeiAvgCS assigns higher weights to updates from high-diversity clients and retains them to participate in more rounds of training. This emphasizes the influence of high-diversity clients and diminishes the impact of low-diversity clients.
Extensive experiments on FashionMNIST and CIFAR10 datasets demonstrate the effectiveness of WeiAvgCS. It can converge 46% faster on FashionMNIST and 38% faster on CIFAR10 than its benchmarks on average.
WeiAvgCS is orthogonal to other state-of-the-art algorithms like FedProx, MOON, and Scaffold, and can be combined with them to further improve performance.
The paper also identifies limitations of WeiAvgCS, where the correlation between projection and variance decreases under severe under-fitting, leading to poorer performance compared to FedAvg.
Til et andet sprog
fra kildeindhold
arxiv.org
Vigtigste indsigter udtrukket fra
by Fan Dong,Ali... kl. arxiv.org 04-11-2024
https://arxiv.org/pdf/2305.16351.pdfDybere Forespørgsler