Data synthesis using deep feature enhanced generative adversarial networks for rolling bearing imbalanced fault diagnosis

• Published in: Mechanical systems and signal processing 2022-01, Vol.163, p.108139, Article 108139
• Main Authors: Liu, Shaowei, Jiang, Hongkai, Wu, Zhenghong, Li, Xingqiu
• Format: Article
• Language: English
• Subjects: Data synthesis; Deep feature enhanced generative adversarial networks; Fault diagnosis; Generative adversarial networks; Imbalanced data; Performance enhancement; Roller bearings; Rolling bearing; Rotating machinery; Signal quality; Synthesis
• ISSN: 0888-3270; 1096-1216
• DOI: 10.1016/j.ymssp.2021.108139

•A pull-away function is combined to design a new loss function of the generator.•The self-attention module is used in the networks to enhance deep features.•An automatic data filter is established to ensure the quality of generated data. Rolling bearing fault diagnosis is of great significance to the stable operation of rotating machinery systems. However, the fault data collected in practical engineering is seriously imbalanced, which degrades the diagnosis performance. In this paper, a novel data synthesis method called deep feature enhanced generative adversarial network is proposed to improve the performance of imbalanced fault diagnosis. Firstly, to avoid the mode collapse phenomenon and improve the stability of the generative adversarial networks, a pull-away function is integrated to design a new objective function of the generator. Secondly, a self-attention module is utilized in the networks to enhance the deep features of real signals, thereby the quality of synthesized data is improved. Finally, an automatic data filter is established to timely ensure the accuracy and diversity of synthesized samples. Experiments are implemented on two rolling bearing datasets. The results indicate that the proposed method outperforms other intelligent methods and shows great potential in imbalanced fault diagnosis.