Sun Jiechen, Zhou Funa, Chen Jie, Wang Chaoge, Hu Xiong, Wang Tianzhen
School of Logistic Engineering, Shanghai Maritime University, Shanghai 201306, China.
Entropy (Basel). 2024 Aug 23;26(9):718. doi: 10.3390/e26090718.
Federated learning (FL) facilitates the collaborative optimization of fault diagnosis models across multiple clients. However, the performance of the global model in the federated center is contingent upon the effectiveness of the local models. Low-quality local models participating in the federation can result in negative transfer within the FL framework. Traditional regularization-based FL methods can partially mitigate the performance disparity between local models. Nevertheless, they do not adequately address the inconsistency in model optimization directions caused by variations in fault information distribution under different working conditions, thereby diminishing the applicability of the global model. This paper proposes a federated adversarial fault diagnosis method driven by fault information discrepancy (FedAdv_ID) to address the challenge of constructing an optimal global model under multiple working conditions. A consistency evaluation metric is introduced to quantify the discrepancy between local and global average fault information, guiding the federated adversarial training mechanism between clients and the federated center to minimize feature discrepancy across clients. In addition, an optimal aggregation strategy is developed based on the information discrepancies among different clients, which adaptively learns the aggregation weights and model parameters needed to reduce global feature discrepancy, ultimately yielding an optimal global model. Experiments conducted on benchmark and real-world motor-bearing datasets demonstrate that FedAdv_ID achieves a fault diagnosis accuracy of 93.09% under various motor operating conditions, outperforming model regularization-based FL methods by 17.89%.
联邦学习(FL)有助于跨多个客户端对故障诊断模型进行协同优化。然而,联邦中心中全局模型的性能取决于局部模型的有效性。参与联邦的低质量局部模型可能会在FL框架内导致负迁移。传统的基于正则化的FL方法可以部分缓解局部模型之间的性能差异。然而,它们没有充分解决不同工作条件下故障信息分布变化导致的模型优化方向不一致问题,从而降低了全局模型的适用性。本文提出了一种由故障信息差异驱动的联邦对抗故障诊断方法(FedAdv_ID),以应对在多种工作条件下构建最优全局模型的挑战。引入了一种一致性评估指标来量化局部和全局平均故障信息之间的差异,指导客户端与联邦中心之间的联邦对抗训练机制,以最小化客户端之间的特征差异。此外,基于不同客户端之间的信息差异开发了一种最优聚合策略,该策略自适应地学习减少全局特征差异所需的聚合权重和模型参数,最终产生一个最优全局模型。在基准和实际电机轴承数据集上进行的实验表明,FedAdv_ID在各种电机运行条件下实现了93.09%的故障诊断准确率,比基于模型正则化的FL方法高出17.89%。