通信学报2024,Vol.45Issue(3):197-207,11.DOI:10.11959/j.issn.1000-436x.2024069
基于相似度加速的自适应聚类联邦学习
Adaptive clustering federated learning via similarity acceleration
摘要
Abstract
In order to solve the problem of model performance degradation caused by data heterogeneity in the federated learning process,it is necessary to consider personalizing in the federated model.A new adaptively clustering federated learning(ACFL)algorithm via similarity acceleration was proposed,achieving adaptive acceleration clustering based on geometric properties of local updates and the positive feedback mechanism during clients federated training.By dividing clients into different task clusters,clients with similar data distribution in the same cluster was cooperated to improve the performance of federated model.It did not need to determine the number of clusters in advance and iteratively divide the clients,so as to avoid the problems of high computational cost and slow convergence speed in the existing clustering fed-eration methods while ensuring the performance of models.The effectiveness of ACFL was verified by using deep con-volutional neural networks on commonly used datasets.The results show that the performance of ACFL is comparable to the clustered federated learning(CFL)algorithm,it is better than the traditional iterative federated cluster algorithm(IFCA),and has faster convergence speed.关键词
联邦学习/个性化/聚类/几何特性/正向反馈Key words
federated learning/personalization/clustering/geometric characteristic/positive feedback分类
信息技术与安全科学引用本文复制引用
朱素霞,顾玢珂,孙广路..基于相似度加速的自适应聚类联邦学习[J].通信学报,2024,45(3):197-207,11.基金项目
黑龙江省自然科学基金资助项目(No.LH2021F032) (No.LH2021F032)
黑龙江省重点研发计划基金资助项目(No.2022ZX01A34) The Natural Science Foundation of Heilongjiang Province(No.LH2021F032),The Key Research and Develop-ment Program of Heilongjiang Province(No.2022ZX01A34) (No.2022ZX01A34)