Rolling bearing fault mode recognition based on 2D image and CNN-BiGRU

PDF(2549 KB)

Journal of Vibration and Shock ›› 2021, Vol. 40 ›› Issue (23) : 194-201.

ZHANG Xunjie1, ZHANG Min1,2, LI Xianjun1

Author information +

History +

Abstract

In order to ensure the effectiveness of fault diagnosis of rolling bearings, combining the advantages of convolutional neural network in image feature extraction and classification recognition, the one-dimensional vibration signals of rolling bearings were converted into two-dimensional image data by using the gramian angular fields（GAF）, which not only retained the complete information of the signal, but also maintained the signal's time dependence. And on this basis, a diagnosis model based on the convolutional neural network and bidirectional gated recurrent unit（GRU） was proposed. First, the two-dimensional image was taken as the input data of the model, and the spatial features of the image were extracted by the convolutional neural network. Then the time features were filtered by the bidirectional gating loop unit, and finally the pattern recognition was completed by the classifier. Through the diagnosis experiments of different fault degree and different fault position of rolling bearings, the accuracy is 99.63% and 99.28% respectively, and their effects are better than other commonly used algorithms, which proves the feasibility of the proposed method.

Key words

rolling bearing / fault diagnosis / gramian angular fields（GAF） / two-dimensional image / convolutional neural network / bidirectional gated recurrent unit（GRU）

Cite this article

EndNote

Ris (Procite)

Bibtex

Download Citations

ZHANG Xunjie1, ZHANG Min1,2, LI Xianjun1. Rolling bearing fault mode recognition based on 2D image and CNN-BiGRU[J]. Journal of Vibration and Shock, 2021, 40(23): 194-201

References

[1] Li H M, Huang J Y, Ji S W. Bearing fault diagnosis with a feature fusion method based on an ensemble convolutional neural network and deep neural network[J]. Sensors, 2019, 19(9) : 1-18.
[2] Nandi S, Toliyat H A, Li, X. Condition monitoring and fault diagnosis of electrical motors – a review[J]. IEEE Transactions on Energy Conversion, 2005, 20(4): 719–729.
[3] 代俊习，郑近德，潘海洋，等. 基于复合多尺度熵与拉普拉斯支持向量机的滚动轴承故障诊断方法[J]. 中国机械工程, 2017, 28(11) : 1339-1345.
Dai J X, Zheng J D, Pan H Y, el at. Rolling bearing fault diagnosis method based on composite multiscale entropy and Laplacian SVM[J]. China Mechanical Engineering, 2017, 28(11) : 1339-1345.
[4] Zheng X X, Zhou G W, Li D D. Application of variational mode decomposition and permutation entropy for rolling bearing fault diagnosis [J]. International Journal of Acoustics and Vibration, 2019, 24(2) : 303-311.
[5] Yan X A, Jia M P. A novel optimized SVM classification algorithm with multi-domain feature and its application to fault diagnosis of rolling bearing[J]. Neurocomputing, 2018, 313 : 47-64.
[6] Liu H M, Li L F, Ma J. Rolling bearing fault diagnosis based on STFT-deep learning and sound signals[J]. Shock and Vibration, 2016, 2016 : 1-12.
[7] Chen Z Q, Deng S C, Chen X D, et al. Deep neural networks-based rolling bearing fault diagnosis[J]. Microelectronics Reliability, 2017, 75(75) : 327-333.
[8] Shao H D, Jiang H K, Li X Q. Intelligent fault diagnosis of rolling bearing using deep wavelet auto-encoder with extreme learning machine[J]. Knowledge Based Systems, 2018, 140 : 1-14.
[9] 陈伟，陈锦雄，江永全，等. 基于RS-LSTM的滚动轴承故障识别[J]. 中国科技论文, 2018, 13(10) : 1134-1141.
Chen W, Chen J X, Jiang Y Q, et al. Fault identification of rolling bearing based on RS-LSTM[J]. China Science Paper, 2018, 13(10) : 1134-1141.
[10] 李彦冬，郝宗波，雷航. 卷积神经网络研究综述[J]. 计算机应用, 2016, 36(9) : 2508-2515, 2565.
Li Y D, Hao Z B, Lei H. Survey of convolutional neural network[J]. Journal of Computer Applications, 2016, 36(9) : 2508-2515, 2565.
[11] Zhang W, Peng G L, Li C H, et al. A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals[J]. Sensors, 2017, 17(2): 425-446.
[12] 曲建玲，余路，袁涛，等. 基于卷积神经网络的层级化智能故障诊断算法[J]. 控制与决策, 2019, 34(12) : 2619-2626.
Qu J L, Yu L, Yuan T, et al. A hierarchical intelligent fault diagnosis algorithm based on convolutional neural network[J]. Control and Decision, 2019, 34(12) : 2619-2626.
[13] Wang F, Jiang H K, Shao H D, et al. An adaptive deep convolutional neural network for rolling bearing fault diagnosis[J]. Measurement Science and Technology, 2017, 28(9): 1-25.
[14] Ding X X, He Q B. Energy-fluctuated multiscale feature learning with deep convnet for intelligent spindle bearing fault diagnosis[J]. IEEE Transactions on Instrumentation and Measurement, 2017, 66(8): 1926-1935.
[15] A. Krizhevsky, I. Sutskever, G. E. Hinton. ImageNet classification with deep convolutional neural networks[C]. Advances in neural information processing systems.London:MIT Press, 2012.
[16] Zhang W, Peng G L, Li C H. Bearings Fault Diagnosis Based on Convolutional Neural Networks with 2-D Representation of Vibration Signals as Input[C]. MATEC Web of Conferences, 2017, 95: 13001.
[17] 张弘斌，袁奇，赵柄锡, 等. 采用多通道样本和深度卷积神经网络的轴承故障诊断方法[J]. 西安交通大学学报, 2020, 54(8) : 1-10.
Zhang H B, Yuan Q, Zhao B X, et al. A Bearing Fault Diagnosis Method Based on Multi-Channel Sample and Deep Convolutional Neural Network[J]. Journal of Xi'an Jiaotong University, 2020, 54(8): 1-10.
[18] Hoang D T, Kang H J. Rolling element bearing fault diagnosis using convolutional neural network and vibration image[J]. Cognitive Systems Research, 2019, 53: 42-50.
[19] 张安安，黄晋英，冀书伟，等. 基于卷积神经网络图像分类的轴承故障模式识别[J]. 振动与冲击, 2020, 39(4) : 165-171.
Zhang A A, Huang J Y, Ji S W, et al. Bearing fault pattern recognition based on image classification with CNN [J]. Journal of Vibration and shock, 2020, 39(4): 165-171.
[20] Z. Wang, T. Oates. Imaging time-series to improve classification and imputation[J]. Computer Science, 2015, 3(27): 3939-3945.
[21] 陈敏，王娆芳. 基于二维图像与迁移卷积神经网络的心律失常分类[J]. 计算机工程, 2020, 46(10)：315-320.
CHEN Min，WANG Raofen. Arrhythmia classification based on two-dimensional image and tranfer convolutional neural network [J]. Computer Engineering, , 2020, 46(10)：315-320.
[22] Liu X Z, Cai H X, Zhong R X, et al. Learning traffic as images for incident detection using convolutional neural networks[J]. IEEE Access, 2020, 8: 7916–7924.
[23] Z. Wang, T. Oates. Encoding time series as images for visual inspection and classification using tiled convolutional neural networks[C]. Workshops at the Twenty-Ninth AAAI Conference on Artificial Intelligence, 2015: 40-46.
[24] Chung J, Gulcehre C, Cho K, et al. Empirical evaluation of gated recurrent neural networks on sequence modeling[C]. Annual Conference on Neural Information Processing Systems 2014. Montrea:CNIPS,2014.
[25] Tao Q, Liu F, Li Y, et al. Air pollution forecasting using a deep learning model based on 1D convnets and bidirectional GRU[J]. IEEE Access, 2019, 7: 76690–76698.
[26] W.A. Smith, R.B. Randall. Rolling element bearing diagnostics using the Case Western Reserve University data: a benchmark study[J]. Mechanical Systems and Signal Processing, 2015, 64–65: 100–131.
[27] 张伟. 基于卷积神经网络的轴承故障诊断算法研究[D]. 哈尔滨: 哈尔滨工业大学, 2017.
Zhang W. Study on bearing fault diagnosis algorithm based on convolutional neural network[D]. Harbin: Harbin Institute of Technology, 2017.
[28] 徐林,郑晓彤,付博,等.基于改进GAN算法的电机轴承故障诊断方法[J].东北大学学报(自然科学版),2019,40(12):1679-1684.
Xu L, Zheng X T, Fu B, et al. Fault Diagnosis Method of Motor Bearing Based on Improved GAN Algorithm [J]. Journal of Northeastern University (Natural Science), 2019, 40(12): 165-171. 1679-1684.
[29] MAATEN L, HINTON G. Visualizing data using t-SNE[J]. Journal of Machine Learning Research, 2008, 9: 2579