A dual attention module and convolutional neural network based bearing fault diagnosis
DOI: 10.23977/jeis.2022.070306 | Downloads: 25 | Views: 593
Yazhou Zhang 1
1 College of Electrical and Information Engineering, Lanzhou University of Technology, Lanzhou 730050, People's Republic of China
Corresponding AuthorYazhou Zhang
Vibration signals of rolling bearings are affected by changing operating conditions and environmental noise, so they are characterized by a high degree of complexity. Although deep learning fault diagnosis methods have achieved considerable success in practical applications, the high complexity characteristics are ignored. To address this issue, we propose a dual attention module and convolutional neural network (DAM-CNN) for rolling bearing fault diagnosis. In this method, we designed a dual-attention module (DAM) by using a channel-attention module and a spatial-attention module. DAM can recode feature information in channel and spatial dimensions, so as to achieve adaptive enhancement of effective network information and suppression of interference information. In addition, to enhance the extraction of long-range features of the convolutional network, we introduce the non-local feature extraction module. This module can significantly expand the perceptual field of convolutional operations and enhance the generalization ability of the network. By verifying the effectiveness of the method in CWRU datasets, the results show that the method in this paper not only has good noise immunity in strong noise environment, but also has high diagnostic accuracy and good generalization performance in different load condition domains.
KEYWORDSFault diagnosis, Convolutional neural network, Dual attention module, Non-local feature extraction module
CITE THIS PAPER
Yazhou Zhang, A dual attention module and convolutional neural network based bearing fault diagnosis. Journal of Electronics and Information Science (2022) Vol. 7: 35-43. DOI: http://dx.doi.org/10.23977/jeis.2022.070306.
 D.-T. Hoang and H.-J. Kang, "A survey on deep learning based bearing fault diagnosis," Neurocomputing, vol. 335, pp. 327-335, 2019.
 H. Liang, J. Cao, and X. Zhao, "Multi-scale dynamic adaptive residual network for fault diagnosis," Measurement, vol. 188, p. 110397, 2022.
 R. Jegadeeshwaran and V. Sugumaran, "Fault diagnosis of automobile hydraulic brake system using statistical features and support vector machines," Mechanical Systems and Signal Processing, vol. 52, pp. 436-446, 2015.
 Y.-K. Gu, X.-Q. Zhou, D.-P. Yu, and Y.-J. Shen, "Fault diagnosis method of rolling bearing using principal component analysis and support vector machine," Journal of Mechanical Science and Technology, vol. 32, no. 11, pp. 5079-5088, 2018.
 G. E. Hinton and R. R. Salakhutdinov, "Reducing the dimensionality of data with neural networks," science, vol. 313, no. 5786, pp. 504-507, 2006.
 W. Zhang, G. Peng, C. Li, Y. Chen, and Z. Zhang, "A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals," Sensors, vol. 17, no. 2, p. 425, 2017.
 J. Lei, C. Liu, and D. Jiang, "Fault diagnosis of wind turbine based on Long Short-term memory networks," Renewable energy, vol. 133, pp. 422-432, 2019.
 J. Zhao, S. Yang, Q. Li, Y. Liu, X. Gu, and W. Liu, "A new bearing fault diagnosis method based on signal-to-image mapping and convolutional neural network," Measurement, vol. 176, p. 109088, 2021.
 X. Wang, R. Girshick, A. Gupta, and K. He, "Non-local neural networks," in Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, pp. 7794-7803.
 W. Zhang, X. Li, and Q. Ding, "Deep residual learning-based fault diagnosis method for rotating machinery," ISA transactions, vol. 95, pp. 295-305, 2019.
 X. Zhao and Y. Zhang, "An intelligent diagnosis method of rolling bearing based on multi-scale residual shrinkage convolutional neural network," Measurement Science and Technology, vol. 33, no. 8, p. 085103, 2022.
 H. Liang and X. Zhao, "Rolling bearing fault diagnosis based on one-dimensional dilated convolution network with residual connection," IEEE Access, vol. 9, pp. 31078-31091, 2021.