A Federated Adversarial Fault Diagnosis Method Driven by Fault Information Discrepancy

Abstract

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%.