Bearing fault diagnosis network based on adaptive dimension-increasing and convolutional self-attention

GUAN Le, WANG Xinyang, YANG Duo, ZHANG Tianqi, ZHU Li, CHEN Jianguo, WANG Zhen

Journal of Vibration and Shock ›› 2024, Vol. 43 ›› Issue (17) : 289-299.

PDF(4079 KB)
PDF(4079 KB)
Journal of Vibration and Shock ›› 2024, Vol. 43 ›› Issue (17) : 289-299.

Bearing fault diagnosis network based on adaptive dimension-increasing and convolutional self-attention

  • GUAN Le, WANG Xinyang, YANG Duo, ZHANG Tianqi, ZHU Li, CHEN Jianguo, WANG Zhen
Author information +
History +

Abstract

Deep learning networks are considered as end-to-end black-box models. Performing interpretable analysis of these networks can facilitate a deeper understanding of their internal operations, enabling rational optimization of network architecture and parameter tuning. Currently, bearing fault diagnosis models based on Transformer networks require the use of methods such as time-frequency analysis to elevate the dimensionality of time-domain signals, transforming them into two-dimensional images like time-frequency graphs for interpretable analysis. However, this approach has drawbacks, including fixed parameters during the dimensionality expansion process, excessive network parameterization, and limited interpretability. To address these issues, we propose a Convolutional Self-Attention Adaptive Dimension-Increasing Network (CSADI-Net) that combines the integration of an in-network adaptive dimensionality expansion method, a Convolutional Self-Attention Module, and Mid Layer Class Activation Maps (ML-CAM). The Convolutional Self-Attention Module efficiently computes query (Q), key (K), and value (V) matrices from feature maps using convolutional layers, significantly reducing the number of trainable parameters. The in-network adaptive dimensionality expansion method seamlessly incorporates the dimensionality expansion process with network training, endowing it with adaptive parameter tuning capabilities. ML-CAM visualizes the network's focus on various regions of the two-dimensional feature maps obtained through the in-network dimensionality expansion method, offering an intuitive and visually interpretable analytical approach. Furthermore, we conducted tests on CSADI-Net and ML-CAM. CSADI-Net achieved an accuracy rate of 97.32±0.12% on the Case Western Reserve University bearing dataset and achieved complete classification on the experimentally measured bearing fault dataset from Dalian University. Additionally, ML-CAM was employed to generate class activation heatmaps for each sample in both datasets, elucidating the network's operational mechanisms. These results affirm that CSADI-Net possesses merits such as high accuracy, robust noise resistance, and strong interpretability.

Key words

deep learning / self-attention / Explainable AI / bearing / fault diagnosis

Cite this article

Download Citations
GUAN Le, WANG Xinyang, YANG Duo, ZHANG Tianqi, ZHU Li, CHEN Jianguo, WANG Zhen. Bearing fault diagnosis network based on adaptive dimension-increasing and convolutional self-attention[J]. Journal of Vibration and Shock, 2024, 43(17): 289-299

References

[1] ZHANG J Q, SUN Y, GUO L, et al. A new bearing fault diagnosis method based on modified convolutional neural networks [J]. Chinese Journal of Aeronautics, 2020, 33(2): 439-447.
[2] LECUN Y, BOTTOU L, BENGIO Y, et al. Gradient-based learning applied to document recognition [J]. Proc IEEE, 1998, 86: 2278-2324.
[3] KRIZHEVSKY A, SUTSKEVER I, HINTON G E. ImageNet Classification with Deep Convolutional Neural Networks [J]. Communications of the Acm, 2017, 60(6): 84-90.
[4] HE K, ZHANG X, REN S, et al. Deep Residual Learning for Image Recognition [J/OL] 2015, arXiv:1512.03385[https://ui.adsabs.harvard.edu/abs/2015arXiv151203385H]. 
[5] WEI Z, GAOLIANG P, CHUANHAO L, 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(3): 425.
[6] XING Z Y, ZHAO R Z, WU Y C, et al. Intelligent fault diagnosis of rolling bearing based on novel CNN model considering data imbalance [J]. Applied Intelligence, 2022, (14): 16281-16293.
[7] ZHOU B, KHOSLA A, LAPEDRIZA A, et al. Learning Deep Features for Discriminative Localization [J/OL] 2015,arXiv:1512.04150[https://ui.adsabs.harvard.edu/abs/2015arXiv151204150Z].10.48550/arXiv.1512.04150
[8] SELVARAJU R R, COGSWELL M, DAS A, et al. Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization [J]. International Journal of Computer Vision, 2020, 128(2): 336-359.
[9] CHATTOPADHYAY A, SARKAR A, HOWLADER P, et al. Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networks [J/OL] 2017, arXiv:1710.11063[https://ui.adsabs.harvard.edu/abs/2017arXiv171011063C].10.48550/arXiv.1710.11063
[10] JIANG P-T, ZHANG C-B, HOU Q, et al. LayerCAM: Exploring hierarchical class activation maps for localization [J]. IEEE Transactions on Image Processing, 2021, 30: 5875-5888.
[11] Wang H , Wang Z , Du M ,et al.Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networks[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).IEEE, 2020.DOI:10.1109/CVPRW50498.2020.00020.1465-1476
[12] Desai S , Ramaswamy H G .Ablation-CAM: Visual Explanations for Deep Convolutional Network via Gradient-free Localization[C]//2020 IEEE Winter Conference on Applications of Computer Vision (WACV).IEEE,2020.DOI:10.1109/WACV45572.2020.9093360.972-980
[13] 王琦, 邓林峰, 赵荣珍. 基于一维CNN迁移学习的滚动轴承故障诊断 [J]. 振动测试与诊断, 2023, 43(01): 24-30+195.
[14] ZHU J, CHEN N, SHEN C Q. A New Deep Transfer Learning Method for Bearing Fault Diagnosis Under Different Working Conditions [J]. Ieee Sensors Journal, 2020, 20(15): 8394-8402.
[15] XU Y D, YAN X A, SUN B B, et al. Multireceptive Field Denoising Residual Convolutional Networks for Fault Diagnosis [J]. Ieee Transactions on Industrial Electronics, 2022, 69(11): 11686-11696.
[16] QIAO M Y, YAN S H, TANG X X, et al. Deep Convolutional and LSTM Recurrent Neural Networks for Rolling Bearing Fault Diagnosis Under Strong Noises and Variable Loads [J]. IEEE Access, 2020, 8: 66257-66269.
[17] 李旭东. 基于深度学习的旋转部件故障诊断研究 [D]; 中国科学院大学(中国科学院国家空间科学中心), 2021.
[18] CHEN H Y, LEE C H. Vibration Signals Analysis by Explainable Artificial Intelligence (XAI) Approach: Application on Bearing Faults Diagnosis [J]. IEEE Access, 2020, 8: 134246-134256.
[19] HE Q C, LI S B, BAI Q, et al. A Siamese Vision Transformer for Bearings Fault Diagnosis [J]. Micromachines, 2022, 13(10): 599-613.
[20] GUO S, ZHANG B, YANG T, et al. Multitask Convolutional Neural Network With Information Fusion for Bearing Fault Diagnosis and Localization [J]. Ieee Transactions on Industrial Electronics, 2020, 67(9): 8005-8015.
[21] DING Y F, JIA M P, MIAO Q H, et al. A novel time-frequency Transformer based on self-attention mechanism and its application in fault diagnosis of rolling bearings [J]. Mechanical Systems and Signal Processing, 2022, 168: 19.
[22] HOU Y D, WANG J J, CHEN Z Q, et al. Diagnosisformer: An efficient rolling bearing fault diagnosis method based on improved Transformer [J]. Engineering Applications of Artificial Intelligence, 2023, 124: 14.
[23] LI G, DENG C, WU J, et al. Rolling Bearing Fault Diagnosis Based on Wavelet Packet Transform and Convolutional Neural Network [J]. Applied Sciences, 2020, 10(3): 770.
[24] LIN Z, FENG M, NOGUEIRA DOS SANTOS C, et al. A Structured Self-attentive Sentence Embedding [J/OL] 2017,arXiv:1703.03130[https://ui.adsabs.harvard.edu/abs/2017arXiv170303130L].10.48550/arXiv.1703.03130
[25] VASWANI A, SHAZEER N, PARMAR N, et al. Attention Is All You Need [J/OL] 2017, arXiv:1706.03762[https://ui.adsabs.harvard.edu/abs/2017arXiv170603762V].010.48550/arXiv.1706.03762
[26] DENG J, DONG W, SOCHER R, et al. ImageNet : A Large-Scale Hierarchical Image Database [J]. Proc CVPR, 2009, 2009: 248-255.
[27] PALANGI H, DENG L, SHEN Y L, et al. Deep Sentence Embedding Using Long Short-Term Memory Networks: Analysis and Application to Information Retrieval [J]. Ieee-Acm Transactions on Audio Speech and Language Processing, 2016, 24(4): 694-707.

PDF(4079 KB)

Accesses

Citation

Detail

Sections
Recommended

/