Rolling bearing fault diagnosis method based on 2D grayscale images and Wasserstein Generative Adversarial Nets under unbalanced sample condition

Abstract

Accurate diagnosis of rolling bearing faults plays a crucial role in ensuring the stable operation of rotating machinery systems. However, in actual engineering applications, a significant disparity between the volume of normal data and the quantity of fault data collected impairs diagnostic performance. Bearing fault diagnosis under sample imbalance conditions is an engineering challenge encountered in the field of fault diagnosis. To improve the fault diagnosis accuracy under unbalanced sample conditions, a rolling bearing fault diagnosis method based on 2D grayscale images and Wasserstein Generative Adversarial Networks (WGAN) is proposed. The method consists of three main steps. First, the acquired bearing vibration signals are transformed into 2D grayscale images. Second, the WGAN generation model is used to generate more fault samples. Finally, both the original samples and the generated samples are used to train the Convolutional Neural Networks classification model. The validity and effectiveness of the proposed method are evaluated and compared to other bearing fault diagnosis approaches using the Case Western Reserve University Bearing Data Center dataset. The experimental results demonstrate the superior quality of the generated samples and the improved fault identification accuracy achieved by the proposed method.

Keywords

Rolling bearing / fault diagnosis / unbalanced sample / 2D grayscale images / Wasserstein Generative Adversarial Network

References

[1] Diez-olivan A,Galar D.Data fusion and machine learning for industrial prognosis: trends and perspectives towards industry 4.0.Inf Fusion2019;50:92-111

[2] Xing S,Jia F.Intelligent fault diagnosis of rotating machinery using locally connected restricted boltzmann machine in big data era. In: 2017 IEEE International Conference on Industrial Engineering and Engineering Management (IEEM); 2017. pp. 1930-34.

[3] Jia F,Lin J,Lu N.Deep neural networks: A promising tool for fault characteristic mining and intelligent diagnosis of rotating machinery with massive data.Mech Syst Signal Process2016;72-3:303-15

[4] Shao H,Zhang H,Liang T.Rolling bearing fault feature learning using improved convolutional deep belief network with compressed sensing.Mech Syst Signal Process2018;100:743-65

[5] Chawla NV,Hall LO.SMOTE: synthetic minority over-sampling technique.J Artif Intell Res2002;16:321-57

[6] Jia F,Lu N.Deep normalized convolutional neural network for imbalanced fault classification of machinery and its understanding via visualization.Mech Syst Signal Process2018;110:349-67

[7] Sun Y,Wong AK.Cost-sensitive boosting for classification of imbalanced data.Pattern Recognit2007;40:3358-78

[8] Deng Y,Shiyao J,Zhiyuan X.Prognostic study of ball screws by ensemble data-driven particle filters.J Manuf Syst2020;56:359-72

[9] Deng Y,Du S,Zhao C.A double-layer attention based adversarial network for partial transfer learning in machinery fault diagnosis.Comput Ind2021;127:103399

[10] Jia S,Lv J,Xie Z.Joint distribution adaptation with diverse feature aggregation: a new transfer learning framework for bearing diagnosis across different machines.Measurement2022;187:110332

[11] Deng Y,Wang D,Huang D.A calibration-based hybrid transfer learning framework for RUL prediction of rolling bearing across different machines.IEEE Trans Instrum Meas2023;72:1-15

[12] Deng Y,Huang D.Combining the theoretical bound and deep adversarial network for machinery open-set diagnosis transfer.Neurocomputing2023;548:126391

[13] Goodfellow I,Mirza M.Generative adversarial networks.Commun ACM2020;63:139-44

[14] Wang Z,Wang Y.An intelligent diagnosis scheme based on generative adversarial learning deep neural networks and its application to planetary gearbox fault pattern recognition.Neurocomputing2018;310:213-22

[15] Lee YO,Hwang J.Application of deep neural network and generative adversarial network to industrial maintenance: a case study of induction motor fault detection. In: 2017 IEEE international conference on big data (big data). 2017. pp. 3248-53.

[16] Zhou F,Fujita H,Wen C.Deep learning fault diagnosis method based on global optimization GAN for unbalanced data.Knowl Based Syst2020;187:104837

[17] Radford A,Chintala S. Unsupervised representation learning with deep convolutional generative adversarial networks. 2016. Available from: https://arxiv.org/abs/1511.06434 [Last accessed on 14 August 2023].

[18] Arjovsky M,Bottou L. Wasserstein generative adversarial networks. 2017. pp.214-23. Available from: https://arxiv.org/abs/1701.07875 [Last accessed on 14 August 2023].

[19] Wen L,Gao L.A new convolutional neural network-based data-driven fault diagnosis method.IEEE Trans Ind Electron2018;65:5990-8