面向工业物联网的相似度聚类个性化联邦学习
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP393

基金项目:

国家自然科学基金项目(61602216, 61672270);江苏省研究生科研与实践创新计划项目(SJCX24_1774);江苏高校“青蓝工程”培养对象项目(KYQ22003);江苏省高等学校自然科学研究面上项目(24KJB520007).


Personalized federated learning with similarity clustering for industrial internet of things
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    在工业物联网数据异质性场景中, 现有的联邦学习方案存在通信不稳定以及无关模型聚合导致的负面影响问题. 鉴于此, 提出面向工业物联网的个性化联邦学习“云-边-端”分层架构(FEDI), 并设计基于相似度聚类的个性化联邦学习(PFedSA)算法. 在模型更新机制上, 该算法利用余弦相似度维护关系矩阵, 自适应选取具有高相似参数的个性化云模型进行下载; 在模型聚合策略上, 动态计算权重并引入正则化, 以此聚合更新局部模型. 在MNIST、FMNIST和CIFAR10三个数据集上, 与FedAvg等8种算法进行对比实验和分析, 实验结果表明: 1)准确率方面, PFedSA算法在病态Non-IID等3种经典场景下精度最优(最高可达99.78%)或接近最高精度; 2)通信效率方面, PFedSA算法借助相似度聚类机制加快模型收敛至目标精度, 单轮计算时间较FedRep减少; 3)超参数影响方面, PFedSA算法对于设备掉线率的鲁棒性更好, 对于数据异质性的适应性更强, 能够有效提升模型个性化性能.

    Abstract:

    In the context of data heterogeneity in the industrial internet of things (IIoT), existing federated learning schemes suffer from unstable communication and negative impacts caused by the aggregation of irrelevant models. This paper proposes a “cloud-edge-end” personalized federated learning hierarchical architecture for industrial internet of things (FEDI), and designs a personalized federated learning algorithm based on similarity clustering (PFedSA). In the model update mechanism, the cosine similarity is used to maintain the relationship matrix, and the personalized cloud model with high similarity parameters is adaptively selected and downloaded. In the model aggregation strategy, weights are dynamically calculated and regularization is introduced to aggregate and update local models. On three datasets of MNIST, FMNIST and CIFAR10, compared with 8 algorithms such as FedAvg, the experimental results show that: 1) In terms of accuracy, the PFedSA algorithm has the best accuracy (up to 99.78%) or close to the highest accuracy in three classic scenarios such as pathological non-independent identical distribution (Non-IID); 2) In terms of communication efficiency, the PFedSA algorithm leverages a similarity clustering mechanism to accelerate the model’s convergence to the target accuracy, and the single-round calculation time is reduced compared to the FedRep algorithm; 3) In terms of the impact of hyperparameters, the PFedSA algorithm has better robustness to device dropout rates, stronger adaptability to data heterogeneity, and can effectively improve the personalized performance of the model.

    参考文献
    相似文献
    引证文献
引用本文

张心雨,史培中,古春生,等.面向工业物联网的相似度聚类个性化联邦学习[J].控制与决策,2026,41(4):1044-1054

复制
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-08-12
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-03-24
  • 出版日期: 2026-04-10
文章二维码