Bearing fault diagnosis based on a domain adaptation model of convolutional neural network under multiple working conditions

PDF(2079 KB)

Journal of Vibration and Shock ›› 2022, Vol. 41 ›› Issue (24) : 192-200.

QIAN Siyu,QIN Dongchen,CHEN Jiangyi,YUAN Feng

Author information +

History +

Abstract

In view of the fact that the deep learning model trained under single working condition data can't realize effective fault diagnosis under complex working conditions, a convolutional neural network-domain adaptation model (CNN-DA) is proposed. Convolution network is used to extract high-level features of fault vibration signals. Channel Attention Mechanism (CAM) is added to the network at the beginning and end to dynamically allocate the weights of feature channels and reduce the interference of invalid information. Combining with the domain adaptive method, the high-level fault features obtained by the feature extraction layer are adapted in the Source and Target domains. The domain adaptation module integrates the Whole Domain Adaptation and the Category Domain Adaptation, so that the data distribution of the features of the same fault tag in the two domains tends to coincide gradually. Finally, the deep learning model is applied to a variety of different situations for training, and the training results and test results are obtained. Through experiments on data sets from different sources, the model is tested under various working conditions, and the results show that the proposed model can cope with the fault detection of rolling bearings under complex working conditions.

Key words

Fault Diagnosis / Deep Learning / convolutional neural network—domain adaptation（CNN-DA） / Domain Adaptation /

Cite this article

EndNote

Ris (Procite)

Bibtex

Download Citations

QIAN Siyu,QIN Dongchen,CHEN Jiangyi,YUAN Feng. Bearing fault diagnosis based on a domain adaptation model of convolutional neural network under multiple working conditions[J]. Journal of Vibration and Shock, 2022, 41(24): 192-200

References

[1] Lu W, Liang B, Yu C, et al. Deep model based domain adaptation for fault diagnosis[J]. IEEE Transactions on Industrial Electronics, 2017, 64(3):2296-2305.
[2] Pan, S. J. Tsang, I. W. Kwok, J. T. Yang. Domain adaptation via Transfer Component Analysis[J]. IEEE Transactions on Neural Networks, 2011, 22(2):199-210.
[3] Long M, Wang J, Ding G, et al. Adaptation regularization: A general framework for transfer learning[J]. IEEE Transactions on Knowledge & Data Engineering, 2014, 26(5):1076-1089.
[4] Wang Q, Michau G, Fink O. Domian adaptive transfer learning for fault diagnosis[C]//2019 Prognostics and System Health Management Conference (PHM-Paris), IEEE. 2019.
[5] Wen Long, Gao Liang, Li Xinyu.[J]. IEEE Transactions on Systems, 2019, 49 (1) :136-144.
[6] Mingsheng Long, Yue Cao, Jianmin Wang, Michael Jordan. Learning transferable features with deep adaptation networks[C]//Proceedings of the 32nd International Conference on Machine Learning. 2015.
[7] Gretton A, Borgwardt K M, Rasch M J, et al. A kernel two-sample test[J]. Journal of Machine Learning Research, 2012, 13:723-773.
[8] Li Y, Wang N, Shi J, et al. Revisiting batch normalization for practical domain adaptation[J]. Pattern Recognition, 2016, 80: 109-117.
[9] WOO S, PARK J, LEE J Y, et al. CBAM: Convolutional block attention module[C]//15th European Conference on Computer Vision. Munich: Springer Verlag. 2018.
[10] Lou X, Loparo K A. Bearing fault diagnosis based on wavelet transform and fuzzy inference[J]. Mechanical Systems and Signal Processing (MSSP), 2004, 18(5):1077-1095.
[11] Bilbao. A benchmark data set for data-driven classification[C]//European Conference of the Prognostics and Health Management Society (PHM Society). 2016.
[12] He K, Zhang X, Ren S. Delving deep into Rectifiers: Surpassing human-level performance on ImageNet classification[C]//2015 IEEE International Conference on Computer Vision (ICCV). 2015.
[13] 朱浩, 宁芊. 基于注意力机制-Inception-CNN 模型的滚动轴承故障分类[J]. 振动与冲击，2020, 39(19):84-93.
Zhu Hao, Ning Qian. Rolling bearing fault classification based on attention mechanism-initiation-CNN model [J]. Vibration and Shock, 2020,39 (19): 84-93.
[14] Guo L, Lei Y, Xing S, et al. Deep convolutional transfer learning network: A new method for intelligent fault diagnosis of machines with unlabeled data[J]. IEEE Transactions on Industrial Electronics, 2018, 66(9):7316-7325.
[15] Y Ganin, E Ustinova, H Ajakan, P Germain, et al. Domain-adversarial training of neural networks[J]. Journal of Machine Learning Research, 2016, 17(59):1-35.
[16] Maaten L V D, Hinton G. Visualizing data using t-SNE[J]. Journal of machine learning research, 2008, 9:2579-2605.