Generative adversarial networks for data augmentation in machine fault diagnosis

Abstract

Generative adversarial networks (GANs) have been proved to be able to produce artificial data that are alike the real data, and have been successfully applied to various image generation tasks as a useful tool for data augmentation. In this paper, we develop an auxiliary classifier GAN(ACGAN)-based framework to learn from mechanical sensor signals and generate realistic one-dimensional raw data. The proposed architecture contains two parts, generator and discriminator, and both of them are built by stacking one-dimensional convolution layers to learn local features from the original input. Such stacked structure is able to learn hierarchical representations through convolution operation and easy to train. Batch normalization is performed within generator to avoid the problem of gradient vanishing during training, and category labels are used as the auxiliary information in this framework to help train the model. The proposed approach is designed to produce realistic synthesized signals with labels and the generated signals can be used as augmented data for further applications in machine fault diagnosis. In order to evaluate the performance of the generative model, we introduce a set of assessment to evaluate the quality of generated samples, including statistical characteristics and experimental verification. Finally, induction motor vibration signal datasets are utilized to investigate the effectiveness of the proposed framework.