基于谱聚类的傅里叶个性化联邦学习研究

来源：电子与信息学报杂志2023年第6期北京时间：

作者：金彤, 陈思光

单位：南京邮电大学物联网学院 南京 210003

摘要：为了缓解联邦学习中跨不同用户终端数据非独立同分布(non-IID)引起的负面影响，该文提出一种基于谱聚类的傅里叶个性化联邦学习算法。具体地，构建一个面向图像分类识别的云边端协同个性化联邦学习模型，提出在云端协同下通过谱聚类将用户终端划分为多个聚类域，以充分利用相似用户终端学到的知识提升模型性能。其次，设计边端协同的局部联邦学习方法，通过代理模型在用户终端对个性化局部模型执行恢复与再更新的操作，可有效恢复聚合过程中丢失的本地知识。进一步地，设计云边协同的傅里叶个性化联邦学习方法，即云服务器通过傅里叶变换将局部模型参数转换到频域空间上进行聚合，为每个边缘节点定制高质量的个性化局部模型，可使全局模型更适用于各个分布式用户终端。最后，实验结果表明，与现有相关算法相比，所提算法收敛速度更快，准确率提高了3%～13%。

关键词：边缘计算, 联邦学习, 谱聚类, 傅里叶变换

基金资助：国家自然科学基金(61971235)，中国博士后科学基金(2018M630590)，江苏省“333高层次人才培养工程”，江苏博士后科研资助计划(2021K501C)，南邮“1311”人才计划和江苏研究生科研创新计划(KYCX22_1029)