LLM2D
基于对抗性联邦共识学习的工业物联网数据异构表面缺陷分类
Adversarial Federated Consensus Learning for Surface Defect Classification Under Data Heterogeneity in IIoT
发布日期: 9/25/2024
arXiv ID: oai:arXiv.org:2409.15711v1

摘要

数据稀缺性阻碍了深度学习在工业表面缺陷分类 (SDC) 中的应用,因为由于隐私问题,很难从工业物联网 (IIoT) 中的各个实体收集和集中足够的训练数据。联邦学习 (FL) 通过允许在客户端之间协作训练全局模型,同时保持隐私,提供了一种解决方案。然而,由于数据异质性——客户端之间数据分布的差异,性能可能会受到影响。在本文中,我们针对 SDC 中不同客户端之间数据异质性的挑战,提出了一种名为对抗性联邦共识学习 (AFedCL) 的新型个性化联邦学习 (PFL) 方法。首先,我们开发了一种动态共识构建策略,以减轻数据异质性造成的性能下降。通过对抗性训练,来自不同客户端的本地模型利用全局模型作为桥梁来实现分布对齐,从而缓解全局知识遗忘问题。作为对该策略的补充,我们提出了一种共识感知聚合机制。它根据不同客户端在全局知识学习中的有效性为它们分配聚合权重,从而增强全局模型的泛化能力。最后,我们设计了一个自适应特征融合模块,以进一步提高全局知识利用效率。针对每个客户端,个性化融合权重会逐渐调整,以最佳地平衡全局和局部特征,从而适应其各自的全局知识学习有效性。与 FedALA 等最先进的 FL 方法相比,所提出的 AFedCL 方法在三个 SDC 数据集上实现了高达 5.67% 的准确率提升。