Blockchain-Enabled Clustered Federated Learning in Fog Computing Networks
Xiaoge Huang, Chen Zhi, Qianbin Chen, Jie Zhang · 2021 IEEE 94th Vehicular Technology Conference (VTC2021-Fall) · 2021
In mobile computing scenarios, federation learning allows users to jointly train global models in a decentralized manner without exposing private data. However, due to the heterogeneity of the network and devices, the traditional global model often fails to fit the user data distribution, which is inconsistent with the primary condition of federation learning, resulting in accuracy decreasing of global models. Besides, the security of federated learning is decreasing with the increase of malicious attacks. To address the aforementioned issues, in this paper, we explore the cosine similarity of model gradients and design a clustered mechanism to improve learning efficiency. Furthermore, we combine the clustered federated learning with the blockchain-supported fog computing networks, which could verify local models uploaded by users and generate the traceable global models to improve the learning efficiency. Finally, we conduct experiments on several frameworks with the real-world dataset FEMNIST, and the experimental results demonstrate the efficiency and robustness of the blockchain-enabled clustered federated learning framework.