Fault Diagnosis of Induction Motors with Imbalanced Data Using Deep Convolutional Generative Adversarial Network

Abstract

A homemade defective model of an induction motor was created by the laboratory team to acquire the vibration acceleration signals of five operating states of an induction motor under different loads. Two major learning models, namely a deep convolutional generative adversarial network (DCGAN) and a convolutional neural network, were applied for fault diagnosis of the induction motor to the problem of an imbalanced training dataset. Two datasets were studied and analyzed: a sufficient and balanced training dataset and insufficient and imbalanced training data. When the training datasets were adequate and balanced, time–frequency analysis was advantageous for fault diagnosis at different loads, with the diagnostic accuracy achieving 95.06% and 96.38%. For the insufficient and imbalanced training dataset, regardless of the signal preprocessing method, the more imbalanced the training dataset, the lower the diagnostic accuracy was for the testing dataset. Samples generated by DCGAN were found to exhibit 80% similarity with the actual data through comparison. By oversampling the imbalanced dataset, DCGAN achieved a 90% diagnostic accuracy, close to the accuracy achieved using a balanced dataset. Among all oversampling techniques, the pro-balanced method yielded the optimal result. The diagnostic accuracy reached 85% in the cross-load test, indicating that the generated data had successfully learned the different fault features that validate the DCGAN’s ability to learn parts of input signals.