Federated Adaptive Asynchronous Clustering Algorithm for Wireless Mesh Networks
Cheng Qiao, Kenneth N. Brown, Fan Zhang, Zhihong Tian
Abstract
It is a challenge to generate an accurate machine learning model in a distributed network due to the increased concern in data privacy and high cost in gathering all raw data. This paper presents an adaptive asynchronous distributed clustering algorithm for agents in wireless network to learn the global models, while the privacy is protected. Moreover, the communication cost and clustering quality can be adaptively balanced. The proposed clustering algorithm does not require the number of clusters to be pre-defined. To improve the accuracy of the global model, we propose a bounding boxes based method to fully utilize the shape information of clusters. In addition, we consider different knowledge levels of agent and different requirements about the global model. In experiments on randomly generated network topologies, we demonstrate that methods which do more extensive clustering in each cycle, and which exchange descriptions of cluster shape and density instead of just centroids and data counts, achieve more consistent clustering, in significantly shorter elapsed time. We also show that the proposed methods can learn the same number of clusters as the ground truth when clusters are well separated from each other.