Local fusion generative adversarial network with dual-discriminator and parallel multipath and its application in machinery fault diagnosis with imbalanced data

Abstract

Abstract Diagnosing faults in critical machinery components is imperative for effective condition monitoring and real-world datasets often suffer from data imbalance. To address this issue, numerous data generation methods have been developed, such as improved local fusion generative adversarial network (ILoFGAN), variational autoencoding GAN (VAEGAN), etc. However, the existing data generation methods primarily concentrate on global and single-scale features and often ignore local or multi-scale features, which leads to the omission of key features or nuances in the generated data. Therefore, a novel approach called the Local Fusion Generative Adversarial Network with Dual-discriminator and Parallel multipath (LoFGAN-DP) is designed to enhance the fault diagnosis performance in the context of imbalanced data. The LoFGAN-DP features a Parallel Multi-Path (PMP) module along with a dual-discriminator scheme, in which the multipath module facilitates feature extraction at various scales through convolution across paths of diverse sizes, and the dual-discriminator scheme can better improve the quality and diversity of the samples generated by the generator. The PMP module and dual-discriminator scheme enhance the proposed method's robustness against variations in input data. After generating data by LoFGAN-DP, a two-dimensional capsule network is further used to achieve the efficient recognition of fault features. To validate the proposed LoFGAN-DP in the machinery fault diagnosis with imbalanced data, the gear dataset and the self-constructed bearing dataset were utilized. Experimental results show that LoFGAN-DP significantly improves Structural Similarity Index (SSIM), Fréchet Inception Distance (FID), and fault classification accuracy compared to several advanced methods.