计算机与现代化Issue(4)：5-11,7.DOI:10.3969/j.issn.1006-2475.2024.04.002
基于AP聚类算法的联邦学习聚合算法
Federated Learning Aggregation Algorithm Based on AP Clustering Algorithm
摘要
Abstract
In traditional federation learning,multiple clients'local models are trained independently from their private data,and the central server generates a shared global model by aggregating the local models.However,due to statistical heterogeneity such as non-independent identically distributed(Non-IID)data,a global model often cannot be adapted to each client.To ad-dress this problem,this paper proposes an AP clustering algorithm-based federation learning aggregation algorithm(APFL)for Non-IID data.In APFL,the server calculates the similarity matrix between each client based on the data characteristics of the clients,and then uses the AP clustering algorithm to divide the clients into different clusters and construct a polycentric frame-work to calculate the suitable personalized model weights for each client.This algorithm is experimented on FMINST dataset and CIFAR10 dataset,and APFL improves 1.88 percentage points on FMNIST dataset and 6.08 percentage points on CIFAR10 data-set compared with traditional Federated Learning FedAvg.The results show that the proposed APFL improves the accuracy perfor-mance of Federated Learning on Non-IID data in this paper.关键词
联邦学习/非独立同分布/AP聚类算法Key words
federal learning/non-independent identical distribution/AP clustering algorithm分类
信息技术与安全科学引用本文复制引用
敖博超,范冰冰..基于AP聚类算法的联邦学习聚合算法[J].计算机与现代化,2024,(4):5-11,7.基金项目
广东省重大科技专项(2016B030305003) （2016B030305003）
