Abstract
With the rapid development of urban rail transit in recent years, it becomes necessary to ensure the operation safety of train wheelset axle boxes. Aiming at the problems of large model size and long diagnosis time in traditional fault diagnosis methods, this paper proposed a novel model to identify concurrent faults in wheelset axle boxes based on an efficient neural network and the attention mechanism. The model was developed based on an improved Ghost bottleneck module sequence to achieve an efficient model. Furthermore, the use of the convolutional block attention module adaptively refined the feature map to improve the generalization of the model. Feature pyramid network was used to fuse the shallow and deep features in the network to improve the extraction ability of various size features. Experiments were carried out on wheelset axle box concurrent fault datasets. Model size, diagnosis speed and accuracy were used as evaluation indexes to compare the efficiencies of existing fault diagnosis methods and our proposed model. Experimental results revealed that the proposed algorithm effectively diagnosed wheelset axle box concurrent faults.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.References
Bu, X., Wu, Y., Gao, Z., & Jia, Y. (2019). Deep convolutional network with locality and sparsity constraints for texture classification. Pattern Recognition, 91, 34–46. https://doi.org/10.1016/j.patcog.2019.02.003.
Chollet, F. (2017). Xception: Deep learning with depthwise separable convolutions. In Proceedings of IEEE conference on computer vision and pattern recognition (pp. 1800–1807). https://doi.org/10.1109/cvpr.2017.195.
Cummins, N., Baird, A., & Schuller, B. W. (2018). Speech analysis for health: Current state-of-the-art and the increasing impact of deep learning. Methods, 151, 41–54. https://doi.org/10.1016/j.ymeth.2018.07.007.
Ding, J. (2018). Fault detection of a wheelset bearing in a high-speed train using the shock-response convolutional sparse-coding technique. Measurement, 117, 108–124. https://doi.org/10.1016/j.measurement.2017.12.010.
Goyal, D., Choudhary, A., Pabla, B. S., & Dhami, S. S. (2020). Support vector machines based non-contact fault diagnosis system for bearings. Journal of Intelligent Manufacturing, 31(5), 1275–1289. https://doi.org/10.1007/s10845-019-01511-x.
Han, K., Wang, Y., Tian, Q., Guo, J., Xu, C., & Xu, C. (2020). GhostNet more features from cheap operations. arXiv:1911.11907.
Hannemann, R., Köster, P., & Sander, M. (2019). Fatigue crack growth in wheelset axles under bending and torsional loading. International Journal of Fatigue, 118, 262–270. https://doi.org/10.1016/j.ijfatigue.2018.07.038.
He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. In Proceedings of IEEE conference on computer vision and pattern recognition (pp. 770–778). https://doi.org/10.1109/cvpr.2016.90.
Hoang, D., & Kang, H. (2019). A survey on deep learning based bearing fault diagnosis. Neurocomputing, 335, 327–335. https://doi.org/10.1016/j.neucom.2018.06.078.
Howard, A. G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., et al. (2017). MobileNets: Efficient convolutional neural networks for mobile vision applications. arXiv:1704.04861.
Iandola, F. N., Han, S., Moskewicz, M. W., Ashraf, K., Dally, W. J., & Keutzer, K. (2016). SqueezeNet: AlexNet-Level accuracy with 50X fewer parameters and < 0.5 MB model size. arXiv:1602.07360.
Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2012). ImageNet classification with deep convolutional neural networks. In Proceedings of 26th annual conference on neural information processing systems (Vol. 2, pp. 1097–1105). https://doi.org/10.1145/3065386.
Lecun, Y., Bottou, L., Bengio, Y., & Haffner, P. (1998). Gradient-based learning applied to document recognition. In Proceedings of the IEEE (Vol. 86, No. 11, pp. 2278–2324). https://doi.org/10.1109/5.726791.
Li, J., Li, X., He, D., & Qu, Y. (2020a). Unsupervised rotating machinery fault diagnosis method based on integrated SAE-DBN and a binary processor. Journal of Intelligent Manufacturing. https://doi.org/10.1007/s10845-020-01543-8.
Li, X., Li, J., Zhao, C., Qu, Y., & He, D. (2020b). Gear pitting fault diagnosis with mixed operating conditions based on adaptive 1D separable convolution with residual connection. Mechanical Systems and Signal Processing. https://doi.org/10.1016/j.ymssp.2020.106740.
Li, X., Zhang, W., & Ding, Q. (2019). Understanding and improving deep learning-based rolling bearing fault diagnosis with attention mechanism. Signal Processing, 161, 136–154. https://doi.org/10.1016/j.sigpro.2019.03.019.
Li, X., Zhang, W., Ding, Q., & Sun, J. (2018). Intelligent rotating machinery fault diagnosis based on deep learning using data augmentation. Journal of Intelligent Manufacturing, 31(2), 433–452. https://doi.org/10.1007/s10845-018-1456-1.
Liang, P., Deng, C., Wu, J., Yang, Z., Zhu, J., & Zhang, Z. (2019). Compound fault diagnosis of gearboxes via multi-label convolutional neural network and wavelet transform. Computers in Industry. https://doi.org/10.1016/j.compind.2019.103132.
Lin, T. -Y., Piotr, D., Girshick, R., He, K., Hariharan, B., & Belongie, S. (2017). Feature pyramid networks for object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 2117–2125).
Liu, R., Wang, F., Yang, B., & Qin, S. J. (2020). Multiscale kernel based residual convolutional neural network for motor fault diagnosis under nonstationary conditions. IEEE Transactions on Industrial Informatics, 16(6), 3797–3806. https://doi.org/10.1109/tii.2019.2941868.
Liu, H., Yao, D., Yang, J., & Li, X. (2019). Lightweight convolutional neural network and its application in rolling bearing fault diagnosis under variable working conditions. Sensors (Basel). https://doi.org/10.3390/s19143109.
Lu, C., Wang, Z., & Zhou, B. (2017). Intelligent fault diagnosis of rolling bearing using hierarchical convolutional network based health state classification. Advanced Engineering Informatics, 32, 139–151. https://doi.org/10.1016/j.aei.2017.02.005.
Maaten, L., & Hinton, G. (2008). Visualizing data using t-SNE. Journal of Intelligent and Robotic Systems, 9, 2579–2605. https://doi.org/10.1007/s10846-008-9235-4.
Pan, Z., Meng, Z., Chen, Z., Gao, W., & Shi, Y. (2020). A two-stage method based on extreme learning machine for predicting the remaining useful life of rolling-element bearings. Mechanical Systems and Signal Processing. https://doi.org/10.1016/j.ymssp.2020.106899.
Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv:1409.1556.
Wang, H., Liu, Z., Peng, D., & Qin, Y. (2020). Understanding and learning discriminant features based on multi-attention 1DCNN for wheelset bearing fault diagnosis. IEEE Transactions on Industrial Informatics, 16(9), 5735–5745. https://doi.org/10.1109/tii.2019.2955540.
Woo, S., Park, J., Lee, J. -Y., & Kweon, I. S. (2018). CBAM: Convolutional block attention module. In Proceedings of the European conference on computer vision (pp. 3–19).
Wu, S. C., Liu, Y. X., Li, C. H., Kang, G. Z., & Liang, S. L. (2018). On the fatigue performance and residual life of intercity railway axles with inside axle boxes. Engineering Fracture Mechanics, 197, 176–191. https://doi.org/10.1016/j.engfracmech.2018.04.046.
Yang, S., Gu, X., Liu, Y., Hao, R., & Li, S. (2020a). A general multi-objective optimized wavelet filter and its applications in fault diagnosis of wheelset bearings. Mechanical Systems and Signal Processing. https://doi.org/10.1016/j.ymssp.2020.106914.
Yang, J. W., Wang, J. H., Huang, Q., & Zhou, M. (2018). Reliability assessment for the solenoid valve of a high-speed train braking system under small sample size. Chinese Journal of Mechanical Engineering. https://doi.org/10.1186/s10033-018-0248-z.
Yang, J. W., Zhao, Y., Wang, J. H., Bai, Y. L., & Liu, C. (2020b). Investigation on impact response feature of railway vehicles with wheel flat fault under variable speed conditions. Journal of Vibration and Acoustics. https://doi.org/10.1115/1.4046126.
Yao, D., Liu, H., Yang, J., & Li, X. (2020). A lightweight neural network with strong robustness for bearing fault diagnosis. Measurement. https://doi.org/10.1016/j.measurement.2020.107756.
Zhang, X., Liu, Z., Wang, J., & Wang, J. (2019). Time-frequency analysis for bearing fault diagnosis using multiple Q-factor Gabor wavelets. ISA Transactions, 87, 225–234. https://doi.org/10.1016/j.isatra.2018.11.033.
Zhang, Y., Xing, K., Bai, R., Sun, D., & Meng, Z. (2020a). An enhanced convolutional neural network for bearing fault diagnosis based on time–frequency image. Measurement. https://doi.org/10.1016/j.measurement.2020.107667.
Zhang, X., Zhang, J., Wang, F., Liu, X., Wu, J., & Li, T. (2020b). Multimodal polarization image simulated crater detection. Journal of Electronic Imaging. https://doi.org/10.1117/1.jei.29.2.023027.
Zhang, X., Zhou, X., Lin, M., & Sun, J. (2018). ShuffleNet: An extremely efficient convolutional neural network for mobiledevices. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 6848–6856).
Zhao, R., Yan, R., Chen, Z., Mao, K., Wang, P., & Gao, R. X. (2019). Deep learning and its applications to machine health monitoring. Mechanical Systems and Signal Processing, 115, 213–237. https://doi.org/10.1016/j.ymssp.2018.05.050.
Zheng, J., Pan, H., Yang, S., & Cheng, J. (2017). Adaptive parameterless empirical wavelet transform based time-frequency analysis method and its application to rotor rubbing fault diagnosis. Signal Processing, 130, 305–314. https://doi.org/10.1016/j.sigpro.2016.07.023.
Acknowledgements
The present work was funded by the National Natural Science Foundation of China (Grant No. 51975038), the Nature Science Foundation of Beijing, China (Grant No. L191005), the Support plan for the development of high-level teachers in Beijing municipal universities (Grant Nos. CIT&TCD201904062 and CIT&TCD201704052), the General Project of Scientific Research Program of Beijing Education Commission Grant No. (SQKM201810016015), the Scientific Research Fund of Beijing University of Civil Engineering Architecture (Grant No. ZF15068), the BUCEA Post Graduate Innovation Project (Grant No. PG2020088) and the Fundamental Research Funds for Beijing University of Civil Engineering and Architecture (Grant No. X18133).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no conflict of interest.
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
Yao, D., Liu, H., Yang, J. et al. Implementation of a novel algorithm of wheelset and axle box concurrent fault identification based on an efficient neural network with the attention mechanism. J Intell Manuf 32, 729–743 (2021). https://doi.org/10.1007/s10845-020-01701-y
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10845-020-01701-y