Please wait a minute...
Frontiers of Mechanical Engineering

ISSN 2095-0233

ISSN 2095-0241(Online)

CN 11-5984/TH

Postal Subscription Code 80-975

2018 Impact Factor: 0.989

Front. Mech. Eng.    2021, Vol. 16 Issue (2) : 340-352    https://doi.org/10.1007/s11465-021-0629-3
RESEARCH ARTICLE
Multi-model ensemble deep learning method for intelligent fault diagnosis with high-dimensional samples
Xin ZHANG1, Tao HUANG1, Bo WU1, Youmin HU1(), Shuai HUANG1, Quan ZHOU2, Xi ZHANG1
1. School of Mechanical Science and Engineering, Huazhong University of Science and Technology, Wuhan 430074, China
2. School of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, China
 Download: PDF(2343 KB)   HTML
 Export: BibTeX | EndNote | Reference Manager | ProCite | RefWorks
Abstract

Deep learning has achieved much success in mechanical intelligent fault diagnosis in recent years. However, many deep learning methods cannot fully extract fault information to recognize mechanical health states when processing high-dimensional samples. Therefore, a multi-model ensemble deep learning method based on deep convolutional neural network (DCNN) is proposed in this study to accomplish fault recognition of high-dimensional samples. First, several 1D DCNN models with different activation functions are trained through dimension reduction learning to obtain different fault features from high-dimensional samples. Second, the obtained features are constructed into 2D images with multiple channels through a conversion method. The integrated 2D feature images can effectively represent the fault characteristic contained in raw high-dimension vibration signals. Lastly, a 2D DCNN model with multi-layer convolution and pooling is used to automatically learn features from the 2D images and identify the fault mode of the mechanical equipment by adopting a softmax classifier. The proposed method, which is validated using the bearing public dataset of Case Western Reserve University, USA and a one-stage reduction gearbox dataset, has high recognition accuracy. Compared with other classical deep learning methods, the proposed fault diagnosis method has considerable improvements.

Keywords fault intelligent diagnosis      deep learning      deep convolutional neural network      high-dimensional samples     
Corresponding Author(s): Youmin HU   
Online First Date: 10 May 2021    Issue Date: 15 June 2021
 Cite this article:   
Xin ZHANG,Tao HUANG,Bo WU, et al. Multi-model ensemble deep learning method for intelligent fault diagnosis with high-dimensional samples[J]. Front. Mech. Eng., 2021, 16(2): 340-352.
 URL:  
https://academic.hep.com.cn/fme/EN/10.1007/s11465-021-0629-3
https://academic.hep.com.cn/fme/EN/Y2021/V16/I2/340
Fig.1  Typical architecture of 2D deep convolutional neural network.
Function Expression
relu f( x)={x, x00, x< 0
elu f(α, x)={ x, x0 α( ex 1)x , x <0
tanh f(x )=2/(1+exp(2x))1
Tab.1  Three different types of activation functions used in this study
Fig.2  Main procedure of the proposed multi-model ensemble deep learning method.
Layer Activation function
1st 1D DCNN 2nd 1D DCNN 3rd 1D DCNN 2D DCNN
input
conv1 relu elu tanh tanh
pool1
conv2 relu elu tanh tanh
pool2
conv3 relu elu tanh tanh
pool3
FC1
FC2 relu relu relu relu
output
Tab.2  Activation function settings of the four models
Layer 1D DCNN 2D DCNN
Parameter size Output size Parameter size Output size
input 4096×1 32×32×3
conv1 8×1×4 4096×4 4×4×8 32×32×8
pool1 4×1 1024×4 2×2 16×16×8
conv2 4×1×8 1024×8 4×4×16 16×16×16
pool2 4×1 256×8 2×2 8×8×16
conv3 4×1×16 256×16 2×2×32 8×8×32
pool3 4×1 64×16 2×2 4×4×32
FC1 1024×1 512×1
FC2 1024×1024 1024×1 512×256 256×1
DR 1024×1 256×1
output 1024×10 10×1 256×10 10×1
Tab.3  Details of the 1D DCNN and 2D DCNN models in case study 1
Fig.3  Training and validation accuracies/loss curves of the multi-model ensemble deep learning model: 1D DCNN using (a) relu, (b) elu, and (c) tanh; (d) 2D DCNN.
Fig.4  Visualization effect by t-SNE of the learned features under the test dataset: (a) 1D DCNN, (b) 2D DCNN, (c) DCAE, and (d) MMEDL.
Fig.5  Confusion matrices of the fault diagnosis results under the test dataset: (a) 1D DCNN, (b) 2D DCNN, (c) DCAE, and (d) MMEDL.
Fig.6  Composition of the monitoring platform.
Fig.7  Installation position of three triaxial sensors.
Gear type Teeth number Gear module/mm Teeth width/mm
Driving gear 50 2 20
Driven gear 80 2 20
Tab.4  Parameters of the experimental gears
Layer Parameter size Output size
input 2048×1
conv1 8×1×4 2048×4
pool1 4×1 512×4
conv2 4×1×8 512×8
pool2 4×1 128×8
conv3 4×1×16 128×16
pool3 2×1 64×16
FC1 1024×1
FC2 1024×1024 1024×1
DR 1024×1
output 1024×4 4×1
Tab.5  Details of the 1D DCNN models in case study 2
Fig.8  Fast Fourier transform results the four gear crack severities: (a) Non-crack, (b) 1/4 crack, (c) 1/2 crack, and (d) 3/4 crack.
Fig.9  (a) Training and validation accuracies and (b) loss curves of the 2D deep convolutional neural network in the proposed multi-model ensemble deep learning.
Fig.10  Accuracy comparison of the different methods under the multi-channel signals. DNN: Deep neural network; DBN: Deep belief network; DCNN: Deep convolutional neural network; MMEDL: Multi-model ensemble deep learning.
1D One-dimensional
2D Two-dimensional
conv Convolutional layer
CNN Convolutional neural network
DBN Deep belief network
DCAE Deep convolutional autoencoder
DCNN Deep convolutional neural network
DNN Deep neural network
DR Dropout layer
FC Fully-connected layer
IR Inner race fault state of the bearing in case study 1
MMEDL Multi-model ensemble deep learning
N Normal state of the bearing in case study 1
OR Outer race fault state of the bearing in case study 1
pool Max pooling layer
RE Rolling element fault state of the bearing in case study 1
SVM Support vector machine
t-SNE t-distributed stochastic neighbor embedding
b bias of the neurons in Eq. (1)
dk Length of the trainable filters in Eq. (1)
k Trainable filters in Eq. (1)
Lc Length of the gear crack in case study 2
Rc Radius of the root circle of the main driving wheel in case study 2
rh Radius of the center hole of the main driving wheel in case study 2
x Input of the convolution layer in Eq. (1)
s Nonlinear activation functions in Eq. (1)
  
1 R Liu, B Yang, E Zio, et al.. Artificial intelligence for fault diagnosis of rotating machinery: A review. Mechanical Systems and Signal Processing, 2018, 108: 33–47
https://doi.org/10.1016/j.ymssp.2018.02.016
2 H Zhang. Fault diagnosis and life prediction of mechanical equipment based on artificial intelligence. Journal of Intelligent & Fuzzy Systems, 2018, 37(3): 3535–3544
https://doi.org/10.3233/JIFS-179157
3 F Jia, Y Lei, L Guo, et al.. A neural network constructed by deep learning technique and its application to intelligent fault diagnosis of machines. Neurocomputing, 2018, 272: 619–628
https://doi.org/10.1016/j.neucom.2017.07.032
4 J Liu, Y Hu, Y Wang, et al.. An integrated multi-sensor fusion-based deep feature learning approach for rotating machinery diagnosis. Measurement Science & Technology, 2018, 29(5): 055103
https://doi.org/10.1088/1361-6501/aaaca6
5 M Azamfar, X Li, J Lee. Intelligent ball screw fault diagnosis using a deep domain adaptation methodology. Mechanism and Machine Theory, 2020, 151: 103932
https://doi.org/10.1016/j.mechmachtheory.2020.103932
6 A Ainapure, X Li, J Singh, et al.. Enhancing intelligent cross-domain fault diagnosis performance on rotating machines with noisy health labels. Procedia Manufacturing, 2020, 48: 940–946
https://doi.org/10.1016/j.promfg.2020.05.133
7 X Zhang, S Guo, L Jiang. Semi-supervised fault identification based on improved Laplace feature mapping and constraint seed K-means. Journal of Vibration and Shock, 2019, 38(16): 93–99 (in Chinese)
https://doi.org/10.13465/j.cnki.jvs.2019.16.014
8 A Widodo, B S Yang. Support vector machine in machine condition monitoring and fault diagnosis. Mechanical Systems and Signal Processing, 2007, 21(6): 2560–2574
https://doi.org/10.1016/j.ymssp.2006.12.007
9 Y Sun, S Zhang, C Miao, et al.. Improved BP neural network for transformer fault diagnosis. Journal of China University of Mining and Technology, 2007, 17(1): 138–142
https://doi.org/10.1016/S1006-1266(07)60029-7
10 J Liu, Y Hu, B Wu, et al.. A hybrid generalized hidden markov model-based condition monitoring approach for rolling bearings. Sensors (Basel), 2017, 17(5): 1143
https://doi.org/10.3390/s17051143
11 D T Hoang, H J Kang. A survey on deep learning based bearing fault diagnosis. Neurocomputing, 2019, 335: 327–335
https://doi.org/10.1016/j.neucom.2018.06.078
12 X Li, J Li, Y Qu, et al.. Semi-supervised gear fault diagnosis using raw vibration signal based on deep learning. Chinese Journal of Aeronautics, 2020, 33(2): 418–426
https://doi.org/10.1016/j.cja.2019.04.018
13 W Zhang, C Li, G Peng, et al.. A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load. Mechanical Systems and Signal Processing, 2018, 100: 439–453
https://doi.org/10.1016/j.ymssp.2017.06.022
14 J Schmidhuber. Deep learning in neural networks: An overview. Neural Networks, 2015, 61: 85–117
https://doi.org/10.1016/j.neunet.2014.09.003
15 S Tang, S Yuan, Y Zhu. Deep learning-based intelligent fault diagnosis methods toward rotating machinery. IEEE Access: Practical Innovations, Open Solutions, 2020, 8: 9335–9346
https://doi.org/10.1109/ACCESS.2019.2963092
16 L Jing, T Wang, M Zhao, et al.. An adaptive multi-sensor data fusion method based on deep convolutional neural networks for fault diagnosis of planetary gearbox. Sensors (Basel), 2017, 17(2): 414
https://doi.org/10.3390/s17020414
17 J Zhu, T Hu, B Jiang, et al.. Intelligent bearing fault diagnosis using PCA-DBN framework. Neural Computing & Applications, 2020, 32(14): 10773–10781
https://doi.org/10.1007/s00521-019-04612-z
18 Y Lecun, L Bottou, Y Bengio, et al.. Gradient-based learning applied to document recognition. Proceedings of the IEEE, 1998, 86(11): 2278–2324
https://doi.org/10.1109/5.726791
19 D Zhao, T Wang, F Chu. Deep convolutional neural network based planet bearing fault classification. Computers in Industry, 2019, 107: 59–66
https://doi.org/10.1016/j.compind.2019.02.001
20 H Chen, N Hu, Z Cheng, et al.. A deep convolutional neural network based fusion method of two-direction vibration signal data for health state identification of planetary gearboxes. Measurement, 2019, 146: 268–278
https://doi.org/10.1016/j.measurement.2019.04.093
21 C Wu, P Jiang, C Ding, et al.. Intelligent fault diagnosis of rotating machinery based on one-dimensional convolutional neural network. Computers in Industry, 2019, 108: 53–61
https://doi.org/10.1016/j.compind.2018.12.001
22 A Kumar, Y Zhou, C P Gandhi, et al.. Bearing defect size assessment using wavelet transform based deep convolutional neural network (DCNN). Alexandria Engineering Journal, 2020, 59(2): 999–1012
https://doi.org/10.1016/j.aej.2020.03.034
23 Y Li, X Du, F Wan, et al.. Rotating machinery fault diagnosis based on convolutional neural network and infrared thermal imaging. Chinese Journal of Aeronautics, 2020, 33(2): 427–438
https://doi.org/10.1016/j.cja.2019.08.014
24 L Wen, X Li, L Gao, et al.. A new convolutional neural network-based data-driven fault diagnosis method. IEEE Transactions on Industrial Electronics, 2018, 65(7): 5990–5998
https://doi.org/10.1109/TIE.2017.2774777
25 Y Xue, D Dou, J Yang. Multi-fault diagnosis of rotating machinery based on deep convolution neural network and support vector machine. Measurement, 2020, 156: 107571
https://doi.org/10.1016/j.measurement.2020.107571
26 Z X Hu, Y Wang, M F Ge, et al.. Data-driven fault diagnosis method based on compressed sensing and improved multiscale network. IEEE Transactions on Industrial Electronics, 2020, 67(4): 3216–3225
https://doi.org/10.1109/TIE.2019.2912763
27 H Li, J Huang, S Ji. Bearing fault diagnosis with a feature fusion method based on an ensemble convolutional neural network and deep neural network. Sensors (Basel), 2019, 19(9): 2034
https://doi.org/10.3390/s19092034
28 Q Zhou, Y Li, Y Tian, et al.. A novel method based on nonlinear auto-regression neural network and convolutional neural network for imbalanced fault diagnosis of rotating machinery. Measurement, 2020, 161: 107880
https://doi.org/10.1016/j.measurement.2020.107880
29 N Pezzotti, B Lelieveldt, L Maaten, et al.. Approximated and user steerable tSNE for progressive visual analytics. IEEE Transactions on Visualization and Computer Graphics, 2017, 23(7): 1739–1752
https://doi.org/10.1109/TVCG.2016.2570755
30 W A Smith, R B Randall. Rolling element bearing diagnostics using the case western reserve university data: A benchmark study. Mechanical Systems and Signal Processing, 2015, 64–65: 100–131
https://doi.org/10.1016/j.ymssp.2015.04.021
31 N Srivastava, G Hinton, A Krizhevsky, et al.. Dropout: A simple way to prevent neural networks from overfitting. Journal of Machine Learning Research, 2014, 15(1): 1929–1958
https://doi.org/10.5555/2627435.2670313
32 C E Agostini, M A Sampaio. Probabilistic neural network with bayesian-based, spectral torque imaging and deep convolutional autoencoder for PDC bit wear monitoring. Journal of Petroleum Science Engineering, 2020, 193: 107434
https://doi.org/10.1016/j.petrol.2020.107434
33 F Jia, Y Lei, J Lin, et al.. Deep neural networks: A promising tool for fault characteristic mining and intelligent diagnosis of rotating machinery with massive data. Mechanical Systems and Signal Processing, 2016, 72–73: 303–315
https://doi.org/10.1016/j.ymssp.2015.10.025
34 X Zhang, S Guo, Y Li, et al.. Semi-supervised fault identification based on laplacian eigenmap and deep belief networks. Journal of Mechanical Engineering, 2019, 56(1): 69
https://doi.org/10.3901/JME.2001.01.069
35 W Gong, H Chen, Z Zhang, et al.. A novel deep learning method for intelligent fault diagnosis of rotating machinery based on improved CNN-SVM and multichannel data fusion. Sensors (Basel), 2019, 19(7): 1693
https://doi.org/10.3390/s19071693
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed