Improved Generative Adversarial Networks with Filtering Mechanism for Fault Data Augmentation

Few-shot fault diagnosis (i.e., fault diagnosis with few samples) is a challenging issue in practice because fault samples are scarce and difficult to obtain. Data augmentation based on generative adversarial networks (GANs) has proven to be an effective solution. However, it often encounters problems such as difficult model training and low quality of generated samples. In this article, an improved GAN with filtering mechanism is developed for fault data augmentation, which introduces the self-attention mechanism and instance normalization (IN) into the GAN structure and utilizes a filtering mechanism. The implementation process comprises two parts, sample generation and abnormal sample filtering. The self-attention mechanism and IN adopted in sample generation can make the generative model easy to train and have better quality of the generated samples. For abnormal sample filtering, the isolated forest (IF) algorithm is used for detecting low-quality generated samples. The effectiveness of the proposed fault data augmentation method is verified using two public datasets for fault diagnosis purposes, and the results show that the proposed method can have better performance over the state-of-art GAN-based data augmentation methods.