A Gated Recurrent Generative Transfer Learning Network for Fault Diagnostics Considering Imbalanced Data and Variable Working Conditions
Zhuorui Li, Jun Ma, Jiande Wu, Pak Kin Wong, Xiaodong Wang, Xiang Li
Abstract
Transfer learning (TL) and generative adversarial networks (GANs) have been widely applied to intelligent fault diagnosis under imbalanced data and different working conditions. However, the existing data synthesis methods focus on the overall distribution alignment between the generated data and real data, and ignore the fault-sensitive features in the time domain, which results in losing convincing temporal information for the generated signal. For this reason, a novel gated recurrent generative TL network (GRGTLN) is proposed. First, a smooth conditional matrix-based gated recurrent generator is proposed to extend the imbalanced dataset. It can adaptively increase the attention of fault-sensitive features in the generated sequence. Wasserstein distance (WD) is introduced to enhance the construction of mapping relationships to promote data generation ability and transfer performance of the fault diagnosis model. Then, an iterative "generation-transfer" co-training strategy is developed for continuous parallel training of the model and the parameter optimization. Finally, comprehensive case studies demonstrate that GRGTLN can generate high-quality data and achieve satisfactory cross-domain diagnosis accuracy.