A new deep auto-encoder method with fusing discriminant information for bearing fault diagnosis

Abstract In recent years, deep learning techniques have been proved a promising tool for bearing fault diagnosis. However, to extract deep features with better representative ability, how to introduce discriminant information about different fault types into the deep learning model is still challenging. Moreover, as deep learning techniques heavily rely on mass of measuring data, relatively small amounts of data may cause over-fitting and reduce model stability as well. To solve such problems, a new deep auto-encoder method with fusing discriminant information about multiple fault types is proposed for bearing fault diagnosis. First, a new loss function is designed by introducing structural discriminant information. Specifically, to improve the feature’s representative ability, a new discriminant regularizer is designed in the loss function by using maximum correlation entropy. And to represent the structural information among multiple fault types, a relation matrix for fault types is introduced, then a new regularizer with a symmetric constraint on this matrix is constructed. Second, a gradient descent method is provided to optimise this loss function, and the optimal deep features, as well as fault relatedness, are learned simultaneously. Experimental results on CWRU and IMS bearing data sets show that, compared to several state-of-the-art diagnosis methods, the proposed method can effectively improve the diagnostic accuracy with acceptable time efficiency. And the results on the Kruskal–Wallis Test indicate the proposed method has better numerical stability.

[1]  Wei Jiang,et al.  Fault diagnosis of rolling bearings with recurrent neural network-based autoencoders. , 2018, ISA transactions.

[2]  Yuan Li,et al.  Bearing fault diagnosis with auto-encoder extreme learning machine: A comparative study , 2017 .

[3]  Peter V. Gehler,et al.  Learning Output Kernels with Block Coordinate Descent , 2011, ICML.

[4]  Wentao Ma,et al.  Maximum correntropy criterion based sparse adaptive filtering algorithms for robust channel estimation under non-Gaussian environments , 2015, J. Frankl. Inst..

[5]  Boualem Boashash,et al.  1-D CNNs for structural damage detection: Verification on a structural health monitoring benchmark data , 2018, Neurocomputing.

[6]  Qinkai Han,et al.  Deep residual learning with demodulated time-frequency features for fault diagnosis of planetary gearbox under nonstationary running conditions , 2019, Mechanical Systems and Signal Processing.

[7]  Shuilong He,et al.  A hybrid approach to fault diagnosis of roller bearings under variable speed conditions , 2017 .

[8]  W. Briggs.  Statistical Methods in the Atmospheric Sciences , 2007 .

[9]  Ran He,et al.  Robust Principal Component Analysis Based on Maximum Correntropy Criterion , 2011, IEEE Transactions on Image Processing.

[10]  Thomas W. Rauber,et al.  Heterogeneous Feature Models and Feature Selection Applied to Bearing Fault Diagnosis , 2015, IEEE Transactions on Industrial Electronics.

[11]  Jun Wang,et al.  An automatic and robust features learning method for rotating machinery fault diagnosis based on contractive autoencoder , 2018, Eng. Appl. Artif. Intell..

[12]  Lei Xing,et al.  Incorporating prior knowledge via volumetric deep residual network to optimize the reconstruction of sparsely sampled MRI. , 2020, Magnetic resonance imaging.

[13]  Minqiang Xu,et al.  A fault diagnosis scheme for planetary gearboxes using modified multi-scale symbolic dynamic entropy and mRMR feature selection , 2017 .

[14]  Yaguo Lei,et al.  Deep normalized convolutional neural network for imbalanced fault classification of machinery and its understanding via visualization , 2018, Mechanical Systems and Signal Processing.

[15]  Donghua Zhou,et al.  Dominant trend based logistic regression for fault diagnosis in nonstationary processes , 2017 .

[16]  Huijun Gao,et al.  A convolutional neural network based on a capsule network with strong generalization for bearing fault diagnosis , 2019, Neurocomputing.

[17]  Hongli Gao,et al.  A new bearing fault diagnosis method based on modified convolutional neural networks , 2020 .

[18]  Junsheng Cheng,et al.  An improved deep convolutional neural network with multi-scale information for bearing fault diagnosis , 2019, Neurocomputing.

[19]  Jun Wang,et al.  Crude oil price prediction model with long short term memory deep learning based on prior knowledge data transfer , 2019, Energy.

[20]  Andrew Ball,et al.  Fault feature extraction for rolling element bearing diagnosis based on a multi-stage noise reduction method , 2019, Measurement.

[21]  Liang Guo,et al.  A neural network constructed by deep learning technique and its application to intelligent fault diagnosis of machines , 2018, Neurocomputing.

[22]  Haidong Shao,et al.  Rolling bearing fault feature learning using improved convolutional deep belief network with compressed sensing , 2018 .

[23]  Min Lei,et al.  Fault Detection for Vibration Signals on Rolling Bearings Based on the Symplectic Entropy Method , 2017, Entropy.

[24]  Haidong Shao,et al.  A novel deep autoencoder feature learning method for rotating machinery fault diagnosis , 2017 .

[25]  Zhixin Yang,et al.  Fault diagnosis of rotating machinery based on multiple probabilistic classifiers , 2018, Mechanical Systems and Signal Processing.

[26]  Jie Zhang,et al.  A priori trust inference with context-aware stereotypical deep learning , 2015, Knowl. Based Syst..

[27]  Ying Zhang,et al.  An enhanced convolutional neural network for bearing fault diagnosis based on time–frequency image , 2020, Measurement.

[28]  Wentao Huang,et al.  Periodic feature oriented adapted dictionary free OMP for rolling element bearing incipient fault diagnosis , 2019 .

[29]  Chen Lu,et al.  Fault diagnosis of rotary machinery components using a stacked denoising autoencoder-based health state identification , 2017, Signal Process..

[30]  Haidong Shao,et al.  Rolling bearing fault diagnosis using adaptive deep belief network with dual-tree complex wavelet packet. , 2017, ISA transactions.

[31]  Wentao Mao,et al.  A novel deep output kernel learning method for bearing fault structural diagnosis , 2019, Mechanical Systems and Signal Processing.

[32]  Pascal Vincent,et al.  Contractive Auto-Encoders: Explicit Invariance During Feature Extraction , 2011, ICML.

[33]  Minping Jia,et al.  A novel optimized SVM classification algorithm with multi-domain feature and its application to fault diagnosis of rolling bearing , 2018, Neurocomputing.

[34]  Diego Cabrera,et al.  Fault diagnosis in spur gears based on genetic algorithm and random forest , 2016 .

[35]  Hee-Jun Kang,et al.  A survey on Deep Learning based bearing fault diagnosis , 2019, Neurocomputing.

[36]  Takehisa Yairi,et al.  A review on the application of deep learning in system health management , 2018, Mechanical Systems and Signal Processing.