Abstract
In practical industrial application, the fault samples collected from rotating machinery are frequently unbalanced, which will create difficulties when it comes to diagnosis. Besides, the variation of working conditions and noise factors will further reduce the diagnosis’s accuracy and stability. Considering the above problems, we established a model based on deep Wasserstein generative adversarial network with gradient penalty (DWGANGP). In this model, the unbalanced fault data set will first be trained by the sample generation network to generate synthetic samples, which will be used to restore the balance. A one-dimensional convolutional neural network with a specific structure is then used as the fault diagnosis network to classify the reconstructed equilibrium samples. The experimental results show that the proposed sample generation network can generate high-quality synthetic samples under highly imbalanced data, and the diagnostic network has a fast training convergence. Compared to the combination methods of support vector machines, back propagation neural network and deep belief network, our method has a 74% average accuracy in all unbalanced experimental conditions, which has 64%, 69% and 87% averages leading, respectively.