Personalized federated learning based on dual adaptive aggregation in the IoT environments
摘要
Personalized federated learning is a method to solve the problem of non-independent and identically distributed data in the Internet of Things and provide personalized services to users. The extant personalized federated learning methods typically prioritize optimization at either the client or server. Our objective is to develop a mechanism that enables collaboration between these two parties. Therefore, we propose a personalized federated learning framework based on dual adaptive aggregation. Specifically, we design aggregation strategy based on dynamic similarity, which adaptively updates the cluster model on the server by exploiting the correlation between client and cluster distributions. Then, based on the differences in the impact of features on the feature-level aggregation results of local models, we construct feature-level local model aggregation strategy. Subsequently, Extensive experimentation was conducted on computer vision and Internet of Things benchmark datasets, and the results demonstrate that our method effectively enhances the accuracy of personalized local models.