GAN-Based Bearing Fault Diagnosis Method for Short and Imbalanced Vibration Signal

Abstract

Bearings are vital components of a rotary machine. Real-time fault diagnosis of bearings has great significance in the maintenance of equipment. Current vibration-based fault diagnosis methods rely on the usage of long time series data to reduce the influence of noise but often suffer from imbalances in datasets, significantly affecting diagnosis accuracy. In this article, a fault diagnosis method based on intertemporal return plot (IRP) and data augmentation is proposed. The IRP is employed to transform 1-D time series data into 2-D images. The Wasserstein generative adversarial network (WGAN) is employed to generate synthetic images for data argumentation. This approach helps reduce the influence of data imbalance, thereby improving the accuracy and convergence speed of the fault diagnosis. Comparisons with existing well-known methods are performed to demonstrate the effectiveness of the proposed method. The results show that the proposed method achieves a high accuracy of fault diagnosis for short time series data. An inverse transformation is also proposed to convert the images to time series data.