Open-set federated adversarial domain adaptation based cross-domain fault diagnosis

Abstract

Abstract Data-driven fault diagnosis techniques utilizing deep learning have achieved widespread success. However, their diagnostic capability and application possibility are significantly reduced in real-world scenarios where fault modes are not fully covered and labels are lacking. Owing to potential conflicts of interest and legal risks, industrial equipment fault data usually exist in the form of isolated islands, making it difficult to carry out large-scale centralized model training. This paper proposes open-set federated adversarial domain adaptation (OS-FADA) to achieve collaborative evolution of fault diagnosis capabilities among cross-domain data owners while protecting privacy. The OS-FADA is a general fault diagnosis framework that employs two-phase adversarial learning. First, faced with the data distribution shift caused by variable working conditions, a generative adversarial feature extractor training strategy is designed to achieve domain-invariant fault feature extraction by approximating the feature distributions of clients to a unified generated distribution. Second, considering the label distribution shift of unknown faults occurring in the target client, an adversarial learning method is proposed to establish decision boundaries between known and unknown faults. Ultimately, the co-evolution of fault diagnosis models between clients is achieved by combining two-phase adversarial learning and federated aggregation. Results from an industrial gearbox case demonstrate that our proposed method achieves over 20% diagnostic accuracy improvement and has excellent potential for cross-domain fault diagnosis tasks with unknown faults when the data silos problem cannot be ignored.