Abstract
Vibration signals can be used to extract effective fault features for fault diagnosis. However, traditional supervised learning requires considerable manpower and time to mark samples manually, and this process is difficult to apply to practical fault diagnosis. Deep reinforcement learning which combines the perception ability of deep learning with the decision-making ability of reinforcement learning, can independently extract hidden fault features and effectively improve the accuracy of fault diagnosis. Semi-supervised learning can reduce the proportion of labeled samples to decrease the learning cost while improving the recognition accuracy with unlabeled samples. In this study, we propose a novel semi-supervised deep reinforcement learning method. A semi-supervised generative adversarial network combined with the improved actor-critic algorithm is proposed to perform fault diagnosis when the labeled sample size is small. In the experiment of rolling bearing fault and engineering application, three-channel time-frequency graphs extracted from raw signals with the wavelet packet are compressed into single channel gray graphs. Then, to simulate the less labeled sample dataset, 2%, 5%, 20%, 50% and 100% sample labels are set by dislodging partial label from the processing sample. The results of the proposed method and other intelligent methods are listed to demonstrate that the proposed method could provide better performance over other methods even if the size of labeled sample is small in compound fault diagnosis.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.References
Shao H, Jiang H, Li X, Liang T (2018) Rolling bearing fault detection using continuous deep belief network with locally linear embedding. Comput Ind 96:27–39
Kong D, Chen Y, Li N, Duan C, Lu L, Chen D (2019) Relevance vector machine for tool wear prediction. Mech Syst Sig Process 127:573–594
Zhang K, Tang B, Qin Y, Deng L (2019) Fault diagnosis of planetary gearbox using a novel semi-supervised method of multiple association layers networks. Mech Syst Sig Process 131:243–260
Gao Z, Cecati C, Ding SX (2015) A survey of fault diagnosis and fault-tolerant techniques part i: Fault diagnosis with model-based and signal-based approaches. IEEE Trans Ind Electron 62(6):3757–3767. https://doi.org/10.1109/TIE.2015.2417501
Kim Kyusung, Parlos AG (2002) Induction motor fault diagnosis based on neuropredictors and wavelet signal processing. IEEE/ASME Trans Mech 7(2):201–219. https://doi.org/10.1109/TMECH.2002.1011258
Zhao R, Yan R, Chen Z, Mao K, Wang P, Gao RX (2019) Deep learning and its applications to machine health monitoring. Mech Syst Sig Process 115:213–237
Singh J, Darpe A, Singh S (2017) Bearing damage assessment using jensen-rnyi divergence based on eemd. Mech Syst Sig Process 87:307–339
Shao H, Jiang H, Lin Y, Li X (2018) A novel method for intelligent fault diagnosis of rolling bearings using ensemble deep auto-encoders. Mech Syst Sig Process 102:278–297
Shao H, Jiang H, Zhao H, Wang F (2017) A novel deep autoencoder feature learning method for rotating machinery fault diagnosis. Mech Syst Sig Process 95:187–204
Mnih V, Kavukcuoglu K, Silver D, Rusu AA, Veness J, Bellemare MG, Graves A, Riedmiller M, Fidjeland AK, Ostrovski G, Petersen S, Beattie C, Sadik A, Antonoglou I, King H, Kumaran D, Wierstra D, Legg S, Hassabis D (2015) Human-level control through deep reinforcement learning. Nature 518(7540):529–533. https://doi.org/10.1038/nature14236
Liu K, Wan Q, Li Y (2018) A deep reinforcement learning algorithm with expert demonstrations and supervised loss and its application in autonomous driving. In: 2018 37th Chinese Control Conference (CCC), Technical Committee on Control Theory, Chinese Association of Automation, pp 2944–2949, https://doi.org/10.23919/ChiCC.2018.8482790
Sangiovanni B, Incremona GP, Piastra M, Ferrara A (2021) Self-configuring robot path planning with obstacle avoidance via deep reinforcement learning. IEEE Control Syst Lett 5(2):397–402. https://doi.org/10.1109/LCSYS.2020.3002852
Schmidhuber J (2015) Deep learning in neural networks: an overview. Neural Netw 61:85–117
LeCun Y, Bengio Y, Hinton G (2015) Deep learning. Nature (London) 521(7553):436–444. https://doi.org/10.1038/nature14539
Mishkin D, Sergievskiy N, Matas J (2017) Systematic evaluation of convolution neural network advances on the imagenet. Comput Vis Image Understand 161:11–19. https://doi.org/10.1016/j.cviu.2017.05.007
Hoang DT, Kang HJ (2019) A survey on deep learning based bearing fault diagnosis. Neurocomputing 335:327–335. https://doi.org/10.1016/j.neucom.2018.06.078
Chen R, Huang X, Yang L, Xu X, Zhang Y, Zhang X (2019) Intelligent fault diagnosis method of planetary gearboxes based on convolution neural network and discrete wavelet transform. Comput Ind 106:48–59. https://doi.org/10.1016/j.compind.2018.11.003
Appana DK, Prosvirin A, Kim JM (2018) Reliable fault diagnosis of bearings with varying rotational speeds using envelope spectrum and convolution neural networks. Soft Comput 22(20):6719–6729. https://doi.org/10.1007/s00500-018-3256-0
Dey D, Chatterjee B, Dalai S, Munshi S, Chakravorti S (2017) A deep learning framework using convolution neural network for classification of impulse fault patterns in transformers with increased accuracy. IEEE Trans Dielectric Electric Insulation 24(6):3894–3897. https://doi.org/10.1109/TDEI.2017.006793
Zhang D, Lin Z, Gao Z (2018) A novel fault detection with minimizing the noise-signal ratio using reinforcement learning. Sensors 18(9):3087. https://doi.org/10.3390/s18093087
Jaafra Y, Luc Laurent J, Deruyver A, Saber Naceur M (2019) Reinforcement learning for neural architecture search: a review. Image and Vis Comput 89:57–66. https://doi.org/10.1016/j.imavis.2019.06.005
Feng X, Jiao Y, Lv C, Zhou D (2016) Label consistent semi-supervised non-negative matrix factorization for maintenance activities identification. Eng Appl Artificial Intell 52:161–167. https://doi.org/10.1016/j.engappai.2016.02.016
Hinton G, Salakhutdinov R (2006) Reducing the dimensionality of data with neural networks. Science (New York, NY) 313:504–7. https://doi.org/10.1126/science.1127647
Salimans T, Goodfellow I, Zaremba W, Cheung V, Radford A, Chen X (2016) Improved techniques for training gans. 1606.03498
Goodfellow IJ, Pouget-Abadie J, Mirza M, Xu B, Warde-Farley D, Ozair S, Courville A, Bengio Y (2014) Generative adversarial nets. In: Ghahramani Z, Welling M, Cortes C, Lawrence ND, Weinberger KQ (eds) Advances in Neural Information Processing Systems, Advances in Neural Information Processing Systems, vol 27. NEURAL INFORMATION PROCESSING SYSTEMS (NIPS), LA JOLLA, pp 2672–2680
Eren L, Ince T, Kiranyaz S (2019) A generic intelligent bearing fault diagnosis system using compact adaptive 1d cnn classifier. J Signal Process Syst 91(2):179C – 189
Zhao Z, Xu Q, Jia M (2016) Improved shuffled frog leaping algorithm-based bp neural network and its application in bearing early fault diagnosis. Neural Comput Appl 27(2):375–385
Ng A, et al. (2011) Sparse autoencoder. CS294A Lecture notes 72(2011):1–19
Lee H, Grosse R, Ranganath R, Ng AY (2009) Convolutional deep belief networks for scalable unsupervised learning of hierarchical representations. In: Proceedings of the 26th annual international conference on machine learning, pp 609–616
Yang M, Liu N, Zuo L, Feng Y, Liu M, Gong H, Liu M (2021) Dynamic charging scheme problem with actor-critic reinforcement learning. IEEE Internet of Things J 8(1):370–380. https://doi.org/10.1109/JIOT.2020.3005598
Zhong S, Liu Q, Fu Q, Franco L (2016) Efficient actor-critic algorithm with hierarchical model learning and planning. Comput Intell Neurosci. https://doi.org/10.1155/2016/4824072
Chamie ME, Yu Y, Acikmese B, Ono M (2019) Controlled markov processes with safety state constraints. IEEE Trans Autom Control 64(3):1003–1018. https://doi.org/10.1109/TAC.2018.2849556
Yan T, Zhang W, Yang S, Yu L (2019) Soft actor-critic reinforcement learning for robotic manipulator with hindsight experience replay. Int J Robotics Autom 34
Wang Z, Xuan J (2021) Intelligent fault recognition framework by using deep reinforcement learning with one dimension convolution and improved actor-critic algorithm. Adv Eng Inform. https://doi.org/10.1016/j.aei.2021.101315
Zhuang Z, Lv H, Xu J, Huang Z, Qin W (2019) A deep learning method for bearing fault diagnosis through stacked residual dilated convolutions. Appl Sci 9(9):18–23. https://doi.org/10.3390/app9091823
Gan M, Wang C, Zhu CA (2016) Construction of hierarchical diagnosis network based on deep learning and its application in the fault pattern recognition of rolling element bearings. Mech Syst Sig Process 72–73:92–104. https://doi.org/10.1016/j.ymssp.2015.11.014
Zhao M, Kang M, Tang B, Pecht M (2018) Deep residual networks with dynamically weighted wavelet coefficients for fault diagnosis of planetary gearboxes. IEEE Trans Ind Electron 65(5):4290–4300. https://doi.org/10.1109/TIE.2017.2762639
Pathak RS, Singh A (2016) Mexican hat wavelet transform of distributions. Integral Trans Special Functions 27(6):468–483. https://doi.org/10.1080/10652469.2016.1155569
Radford A, Metz L, Chintala S (2016) Unsupervised representation learning with deep convolutional generative adversarial networks. 1511.06434
Duan J, Shi T, Duan J, Xuan J, Zhang Y (2018) A narrowband envelope spectra fusion method for fault diagnosis of rolling element bearings. Measurement Sci Technol 29(12):125106
Acknowledgements
This research is supported by the National Key R&D Program of China(Grant No. 2020YFB2007700), and the National Natural Science Foundation of China (Grant No. 52175094).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Wang, Z., Xuan, J. & Shi, T. A novel semi-supervised generative adversarial network based on the actor-critic algorithm for compound fault recognition. Neural Comput & Applic 34, 10787–10805 (2022). https://doi.org/10.1007/s00521-022-07011-z
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00521-022-07011-z