A Novel Image-Based Diagnosis Method Using Improved DCGAN for Rotating Machinery

Rotating machinery plays an important role in industrial systems, and faults in the machinery may damage the system health. A novel image-based diagnosis method using improved deep convolutional generative adversarial networks (DCGAN) is proposed for the feature recognition and fault classification of rotating machinery. First, vibration signal data from the rotating machinery is transformed into time–frequency feature 2-D image data by a continuous wavelet transform and used for fault classification with the neural network method. The adaptive deep convolution neural network (ADCNN) is then combined with the generative adversarial networks (GANs) to improve the performance of the feature self-learning ability from input data. Compared with different fault diagnosis methods, the proposed method has better performance for image feature classification in rotating machinery.