FedCO: Communication-Efficient Federated Learning via Clustering Optimization †
2022 (English)In: Future Internet, E-ISSN 1999-5903, Vol. 14, no 12, article id 377
Article in journal (Refereed) Published
Abstract [en]
Federated Learning (FL) provides a promising solution for preserving privacy in learning shared models on distributed devices without sharing local data on a central server. However, most existing work shows that FL incurs high communication costs. To address this challenge, we propose a clustering-based federated solution, entitled Federated Learning via Clustering Optimization (FedCO), which optimizes model aggregation and reduces communication costs. In order to reduce the communication costs, we first divide the participating workers into groups based on the similarity of their model parameters and then select only one representative, the best performing worker, from each group to communicate with the central server. Then, in each successive round, we apply the Silhouette validation technique to check whether each representative is still made tight with its current cluster. If not, the representative is either moved into a more appropriate cluster or forms a cluster singleton. Finally, we use split optimization to update and improve the whole clustering solution. The updated clustering is used to select new cluster representatives. In that way, the proposed FedCO approach updates clusters by repeatedly evaluating and splitting clusters if doing so is necessary to improve the workers’ partitioning. The potential of the proposed method is demonstrated on publicly available datasets and LEAF datasets under the IID and Non-IID data distribution settings. The experimental results indicate that our proposed FedCO approach is superior to the state-of-the-art FL approaches, i.e., FedAvg, FedProx, and CMFL, in reducing communication costs and achieving a better accuracy in both the IID and Non-IID cases. © 2022 by the authors.
Place, publisher, year, edition, pages
MDPI, 2022. Vol. 14, no 12, article id 377
Keywords [en]
clustering, communication efficiency, convolutional neural network, federated learning, Internet of Things, Convolutional neural networks, Cost reduction, Learning systems, Privacy-preserving techniques, Central servers, Clustering optimizations, Clusterings, Communication cost, Optimization approach, Shared model, Workers'
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:bth-24176DOI: 10.3390/fi14120377ISI: 000901037100001Scopus ID: 2-s2.0-85144590253OAI: oai:DiVA.org:bth-24176DiVA, id: diva2:1725283
Note
open access
2023-01-102023-01-102023-08-03Bibliographically approved