Federated User Clustering for non-IID Federated Learning

Pacheco, Lucas; Rosário, Denis; Cerqueira, Eduardo; Braun, Torsten (September 2021). Federated User Clustering for non-IID Federated Learning. In: International Conference on Networked Systems 2021 (NetSys 2021). Lübeck, Germany. 13-16 September 2021.

Text GroupClustering_NetSys_Camera_Ready.pdf - Published Version Restricted to registered users only Available under License Publisher holds Copyright. Download (561kB)

Federated Learning (FL) is one of the leading learning paradigms for enabling a more significant presence of intelligent applications in networking considering highly distributed environments while preserving user privacy. However, FL has the significant shortcoming of requiring user data to be Independent Identically Distributed (IID) to make reliable predictions for a given group of users. We present a Neural Network-based Federated Clustering mechanism capable of clustering the local models trained by users of the network with no access to their raw data. We also present an alternative to the FedAvg aggregation algorithm used in traditional FL, which significantly increases the aggregated models’ reliability in Mean Square Error by creating several training models over IID users.

Interest & Impact

Downloads

5 since deposited on 22 Apr 2021
3 in the past 12 months

Citations

5 Citations in Web of Science ®
6 Citations in Scopus

Search

in Google Scholar™

Services

Actions (login required)

Edit item

Actions (login required)

Edit item