Dynamic Clustering in Federated LearningShow others and affiliations
2021 (English)In: ICC 2021 - IEEE International Conference on Communications, Institute of Electrical and Electronics Engineers (IEEE) , 2021Conference paper, Published paper (Refereed)
Abstract [en]
In the resource management of wireless networks, Federated Learning has been used to predict handovers. However, non-independent and identically distributed data degrade the accuracy performance of such predictions. To overcome the problem, Federated Learning can leverage data clustering algorithms and build a machine learning model for each cluster. However, traditional data clustering algorithms, when applied to the handover prediction, exhibit three main limitations: the risk of data privacy breach, the fixed shape of clusters, and the non-adaptive number of clusters. To overcome these limitations, in this paper, we propose a three-phased data clustering algorithm, namely: generative adversarial network-based clustering, cluster calibration, and cluster division. We show that the generative adversarial network-based clustering preserves privacy. The cluster calibration deals with dynamic environments by modifying clusters. Moreover, the divisive clustering explores the different number of clusters by repeatedly selecting and dividing a cluster into multiple clusters. A baseline algorithm and our algorithm are tested on a time series forecasting task. We show that our algorithm improves the performance of forecasting models, including cellular network handover, by 43%.
Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE) , 2021.
Series
IEEE International Conference on Communications, ISSN 1550-3607
Keywords [en]
clustering, Federated Learning, GAN, non-HD, handover prediction
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-306508DOI: 10.1109/ICC42927.2021.9500877ISI: 000719386003137Scopus ID: 2-s2.0-85115675502OAI: oai:DiVA.org:kth-306508DiVA, id: diva2:1621785
Conference
IEEE International Conference on Communications (ICC), JUN 14-23, 2021, Conference Location: Montreal, QC, Canada
Note
QC 20211220
Part of proceeding: ISBN 978-1-7281-7122-7
2021-12-202021-12-202024-03-18Bibliographically approved