Abstract
The rolling bearing is the key component of rotating machinery, and fault diagnosis for rolling bearings can ensure the safe operation of rotating machinery. Fault diagnosis technology based on deep learning has been largely studied for bearing fault diagnosis. However, for the deep learning model based on convolutional neural network, there are some intrinsic problems of producing inconspicuous features and useful feature information loss in the process of feature extraction of the raw fault vibration signals. In this work, an intelligent fault diagnosis method of rolling bearings based on short-time Fourier transform and convolutional neural network (STFT-CNN) is proposed. The one-dimensional vibration signals are converted into time–frequency images by STFT. Then, time–frequency images are inputted into STFT-CNN model for fault feature learning and fault identification. For the STFT of the vibration signals, the window type, window width and translation overlap width of the five typical window functions are studied and optimal one is obtained. And in the STFT-CNN model, the stacked double convolutional layers are adopted to improve the nonlinear expression capability of the model. To verify the effectiveness of the proposed method, experiments are carried out on the Case Western Reserve University (CWRU) and the Machine Failure Prevention Technology (MFPT) Society bearing datasets. The results show that the proposed method outperforms other comparative methods and reaches the identification accuracy of 100% and 99.96% for CWRU and MFPT, respectively.
Similar content being viewed by others
References
T.F. Zhang, S.Y. Liu, S. Zhang et al., Improved sparse representation of rolling bearing fault feature based on nested dictionary. J. Fail. Anal. and Preven. 22, 815–828 (2022)
X.Q. Zhao, Y.Z. Zhang, An intelligent diagnosis method of rolling bearing based on multi-scale residual shrinkage convolutional neural network. Meas. Sci. Technol. 33, 085103 (2022)
I. González-Prieto, M.J. Duran, N. Rios-Garcia et al., Open-switch fault detection in five-phase induction motor drives using model predictive control. IEEE Trans. Ind. Electron. 65, 3045–3055 (2018)
D. Jung, C. Sundstrom, A combined data-driven and model-based residual selection algorithm for fault detection and isolation. IEEE Trans. Control Syst. Technol. 27, 616–630 (2017)
D.C. Zhu, Y.Y. Pan, W.P. Gao, Fault feature extraction of rolling element bearing under complex transmission path based on multiband signals cross-correlation spectrum. J. Fail. Anal. and Preven. 22, 1164–1179 (2022)
H.D. Shao, J.S. Cheng, H.K. Jiang et al., Enhanced deep gated recurrent unit and complex wavelet packet energy moment entropy for early fault prognosis of bearing. Knowl. Based Syst. 188, 105022 (2020)
Z.W. Gao, C. Cecati, S.X. Ding, A survey of fault diagnosis and fault-tolerant techniques-Part II: fault diagnosis with knowledge-based and hybrid/active approaches. IEEE Trans. Ind. Electron. 62, 3768–3774 (2015)
Z.W. Gao, C. Cecati, S.X. Ding, 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, 3757–3767 (2015)
T. Jin, C. Yan, C. Chen et al., Light neural network with fewer parameters based on CNN for fault diagnosis of rotating machinery. Measurement. 181, 109639 (2021)
J.Y. Jiao, M. Zhao, J. Lin et al., A comprehensive review on convolutional neural network in machine fault diagnosis. Neurocomputing. 417, 36–63 (2020)
X. Wang, D. Mao, X. Li, Bearing fault diagnosis based on vibro-acoustic data fusion and 1D-CNN network. Measurement. 173, 108518 (2021)
J.S.L. Senanayaka, H.V. Khang, K.G. Robbersmyr, Toward self-supervised feature learning for online diagnosis of multiple faults in electric powertrains. IEEE Trans. Ind. Inform. 17, 3772–3781 (2021)
W. Zhang, G.L. Peng, C.H. Li et al., A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals. Sensors. 17, 425 (2017)
O. Abdeljaber, O. Avci, S. Kiranyaz et al., Real-time vibration-based structural damage detection using one-dimensional convolutional neural networks. J. Sound Vib. 388, 154–170 (2017)
L.Y. Su, L. Ma, N. Qin et al., Fault diagnosis of high-speed train bogie by residual-squeeze net. IEEE Trans. Ind. Inform. 15, 3856–3863 (2019)
H. Wang, Z.L. Liu, D.D. Peng et al., Understanding and learning discriminant features based on multiattention 1DCNN for wheelset bearing fault diagnosis. IEEE Trans. Ind. Inform. 16, 5735–5745 (2020)
Z.B. Zhao, T.F. Li, J.Y. Wu et al., Deep learning algorithms for rotating machinery intelligent diagnosis: an open source benchmark study. ISA Trans. 107, 224–255 (2020)
O. Janssens, V. Slavkovikj, B. Vervisch et al., Convolutional neural network based fault detection for rotating machinery. J. Sound Vib. 377, 331–345 (2016)
S. Zhang, S.B. Zhang, B.N. Wang et al., Deep learning algorithms for bearing fault diagnostics-a comprehensive review. IEEE Access. 8, 29857–29881 (2020)
M. Bhadane, K.I. Ramachandran, Bearing fault identification and classification with convolutional neural network. International Conference on Circuit, Power and Computing Technologies (ICCPCT), (Kollam, India, 2017).
D.T. Hoang, H.J. Kang, Rolling element bearing fault diagnosis using convolutional neural network and vibration image. Cogn. Syst. Res. 53, 42–50 (2019)
Q.B. Wang, B. Zhao, H.B. Ma et al., A method for rapidly evaluating reliability and predicting remaining useful life using two-dimensional convolutional neural network with signal conversion. J. Mech. Sci. Technol. 33, 2561–2571 (2019)
L. Wen, X.Y. Li, L. Gao et al., A new convolutional neural network-based data-driven fault diagnosis method. IEEE Trans. Ind. Electron. 65, 5990–5998 (2018)
B.X. Zhao, Q. Yuan, Improved generative adversarial network for vibration-based fault diagnosis with imbalanced data. Measurement. 169, 108522 (2021)
D. Verstraete, A. Ferrada, D.E. López et al., Deep learning enabled fault diagnosis using time-frequency image analysis of rolling element bearings. Shock Vib. 2017, 1–17 (2017)
Z.Y. Zhu, G.L. Peng, Y.H. Chen et al., A convolutional neural network based on a capsule network with strong generalization for bearing fault diagnosis. Neurocomputing. 323, 62–75 (2019)
H.F. Tao, P. Wang, Y.Y. Chen et al., An unsupervised fault diagnosis method for rolling bearing using STFT and generative neural networks. J. Franklin Inst. 357, 7286–7307 (2020)
Y. Lecun, L. Bottou, Y. Bengio et al., Gradient-based learning applied to document recognition. P. IEEE. 86, 2278–2324 (1998)
A. Krizhevsky, I. Sutskever, G.E. Hinton, ImageNet classification with deep convolutional neural networks. Advances in Neural Information Processing Systems, (Lake Tahoe, USA, 2012)
C. Szegedy, W. Liu, Y.Q. Jia et al., A going deeper with convolutions. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), (Boston, USA, 2015)
K.M. He, X.Y. Zhang, S.Q. Ren et al., Deep residual learning for image recognition. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), (Las Vegas, USA, 2016)
M. Sandler, A. Howard, M.L. Zhu et al., MobileNetV2: inverted residuals and linear bottlenecks. In IEEE conference on computer vision and pattern recognition (CVPR). (Salt Lake City, USA, 2018)
X. Zhang, S. Liu, L. Li et al., Multiscale holospectrum convolutional neural network-based fault diagnosis of rolling bearings with variable operating conditions. Meas. Sci. Technol. 32, 105027 (2021)
J.L. Yang, T.Y. Gao, S.D. Jiang et al., Fault diagnosis of rotating machinery based on one-dimensional deep residual shrinkage network with a wide convolution layer. Shock Vib. 2020, 1–12 (2020)
W.A. Smith, R.B. Randall, Rolling element bearing diagnostics using the case western reserve university data: a benchmark study. Mech. Syst. Signal Process. 64–65, 100–131 (2015)
X.J. Guo, L. Chen, C.Q. Shen, Hierarchical adaptive deep convolution neural network and its application to bearing fault diagnosis. Measurement. 93, 490–502 (2016)
M. Gan, C. Wang, C.A. Zhu, Construction of hierarchical diagnosis network based on deep learning and its application in the fault pattern recognition of rolling element bearings. Mech. Syst. Signal Process. 72–73, 92–104 (2016)
Y.G. Lei, F. Jia, J. Lin et al., An intelligent fault diagnosis method using unsupervised feature learning towards mechanical big data. IEEE Trans. Ind. Electron. 63, 3137–3147 (2016)
X. Li, W. Zhang, Q. Ding et al., Intelligent rotating machinery fault diagnosis based on deep learning using data augmentation. J. Intell. Manuf. 31, 433–452 (2020)
H. Wang, Z.L. Liu, Y.P. Ge et al., Self-supervised signal representation learning for machinery fault diagnosis under limited annotation data. Knowl. Based Syst. 239, 107978 (2022)
Y. Xu, Z.X. Li, S.Q. Wang et al., A hybrid deep-learning model for fault diagnosis of rolling bearings. Measurement. 169, 108502 (2021)
S. Ayas, M.S. Ayas, A novel bearing fault diagnosis method using deep residual learning network. Multimed. Tools Appl. 81, 1–17 (2022)
E.A. Bechhoefer, Quick introduction to bearing envelope analysis MFPT Data (available at: www.mfpt.org/fault-data-sets)
Acknowledgments
The authors thank the supports from the National Natural Science Foundation of China (Grant No. 62241308) and Technological Innovation Guidance Plan of Gansu Province (Grant No. 22CX8GA130). The authors also sincerely thank the Case Western Reserve University Bearing Data Center and the Machine Failure Prevention Technology Society for supplying fault bearing datasets, and the anonymous reviewers for their constructive suggestions and comments on this paper.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflicts of interests
The authors declare no competing interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Zhang, Q., Deng, L. An Intelligent Fault Diagnosis Method of Rolling Bearings Based on Short-Time Fourier Transform and Convolutional Neural Network. J Fail. Anal. and Preven. 23, 795–811 (2023). https://doi.org/10.1007/s11668-023-01616-9
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11668-023-01616-9