Adaptive Clustered Federated Learning for Heterogeneous Data in Edge Computing

Biyao Gong,Tianzhang Xing,Zhidan Liu,Junfeng Wang,Xiuya Liu
DOI: https://doi.org/10.1007/s11036-022-01978-8
2022-05-07
Mobile Networks and Applications
Abstract:Although federated learning has been widely used in collaborative training of machine learning models, its practical uses are still challenged by heterogeneous data across clients. To alleviate the impact of non-IID data issue, we present an adaptive clustered federated learning approach, , which can classify clients into suitable clusters according to their local data distribution and train a specialized model for the clients of each cluster. By exploiting the implicit connection between local model weights and data distribution on clients, relies on partial selected model weights to measure the data similarity between clients and adaptively groups them into the optimal number of clusters. Experimental results on three benchmark datasets with various non-IID data settings demonstrate that achieves comparably high model accuracy as the state-of-the-art works, yet with a significant reduction on the communication cost.
computer science, information systems,telecommunications, hardware & architecture
What problem does this paper attempt to address?