Abstract:The problem of imbalanced data amounts between the samples of healthy and faulty conditions degrades the performance of traditional intelligent diagnosis methods for rotating machinery. This article explores a data generation approach to improve the diagnostic performance of key components in the rotating machinery. A new generative adversarial network (GAN) model, called categorical feature GAN (CFGAN), is proposed to generate sufficient synthetic samples for the minority faulty conditions so as to rebalance the datasets. The main contribution is that the quality of the synthetic samples is ensured by considering both consistency and diversity with the real samples, which conduces to machinery fault diagnosis. Specifically, latent categorical features are first learned from the real fault samples via the proposed CFGAN, which is an integration model of autoencoder (AE) and auxiliary classifier GAN (ACGAN). Then, synthetic categorical features are produced by interpolation among the learned categorical features followed by slight noise addition. Finally, synthetic samples are generated using the synthetic categorical features through the generator of the CFGAN. In addition, Wasserstein distance (WD) is applied to the CFGAN to avoid mode collapse in training. Because categorical features rather than random noise are used to generate samples and the interpolation is performed in the feature space instead of the noisy data space, the consistency of the synthetic samples is improved as compared to the traditional methods. The diversity is ensured by the use of diverse synthetic categorical features and the WD. Two machinery datasets under varying imbalance ratios confirmed that the proposed CFGAN model can synthesize high-quality samples and has better generalization capability over the existing data generation methods in machinery fault diagnosis.

Transforming the Open Set Into a Pseudo-Closed Set: A Regularized GAN for Domain Adaptation in Open-Set Fault Diagnosis

Ensemble Data Augmentation for Imbalanced Fault Diagnosis.

Generalized open-set domain adaptation in mechanical fault diagnosis using multiple metric weighting learning network

Open Set Bearing Fault Diagnosis with Domain Adaptive Adversarial Network under Varying Conditions

Multiweight Adversarial Open-Set Domain Adaptation Network for Machinery Fault Diagnosis With Unknown Faults

Open-Set Domain Adaptation in Machinery Fault Diagnostics Using Instance-Level Weighted Adversarial Learning

Open Set Domain Adaptation: Theoretical Bound and Algorithm

A case study of conditional deep convolutional generative adversarial networks in machine fault diagnosis

Adversarial Domain Adaptation With Dual Auxiliary Classifiers for Cross-Domain Open-Set Intelligent Fault Diagnosis

Universal Domain Adaptation in Fault Diagnostics With Hybrid Weighted Deep Adversarial Learning

Single domain generalization method based on anti-causal learning for rotating machinery fault diagnosis

A Transfer Learning Strategy for Rotation Machinery Fault Diagnosis Based on Cycle-Consistent Generative Adversarial Networks

A Novel Multidomain Contrastive-Coding-Based Open-Set Domain Generalization Framework for Machinery Fault Diagnosis

Cross-domain augmentation diagnosis: An adversarial domain-augmented generalization method for fault diagnosis under unseen working conditions

Open Set Domain Adaptation With Soft Unknown-Class Rejection

Bearing fault diagnosis based on partial domain adaptation adversarial network

Open-Set Fault Diagnosis Via Supervised Contrastive Learning with Negative Out-of-Distribution Data Augmentation.

Application of Generative Adversarial Networks for Intelligent Fault Diagnosis

A Deep Transfer Learning Fault Diagnosis Method Based on WGAN and Minimum Singular Value for Non-Homologous Bearing

Categorical Feature GAN for Imbalanced Intelligent Fault Diagnosis of Rotating Machinery

An Intelligent Machinery Fault Diagnosis Method Based on GAN and Transfer Learning under Variable Working Conditions