Multi-source Domain Adaptation Intelligent Fault Diagnosis Method Based on Asymmetric Adversarial Training

doi:10.3901/JME.2024.18.076

Abstract

Abstract: When using the traditional domain adaptation method for cross-condition fault diagnosis of axial flow fan, the source domain and target domain features will move closer to each other, thus changing the trained source domain feature distribution. And when the source domain fault features are gathered at the decision boundary, the target domain fault features are also gathered at the decision boundary after domain adaptation, which is easy to cause misclassification of some target samples. In addition, single source domain adaptation will affect the generalization ability of the model. For the above problems, a multi-source domain adaptation intelligent fault diagnosis method based on asymmetric adversarial training (TC-MAADA) is proposed. The method first uses triplet-center loss to improve the discrimination of target samples by reducing the intra-class distance and increasing the inter-class distance of fault features in the source domain. Then adopts the asymmetric adversarial training to realize the one-way movement of the target domain fault features to the source domain. Finally, the domain-invariant features of different source and target domains are extracted and input to their respective fault classifiers, using the cosine similarity to align the outputs of each classifier while applying alignment weights to improve the cross-domain diagnostic ability of the model. Experiments show that the method is effective in solving relevant practical industrial problems.

Key words: fault diagnosis, domain adaptation, asymmetric adversarial training, triplet-center loss, alignment weights

CLC Number:

TH17
TP183

LI Zhipeng, MA Tianyu, LIU Jinping, XIANG Qingsong, TANG Junjie. Multi-source Domain Adaptation Intelligent Fault Diagnosis Method Based on Asymmetric Adversarial Training[J]. Journal of Mechanical Engineering, 2024, 60(18): 76-88.

References

[1] XING Saibo，LEI Yaguo，WANG Shuhui，et al. Distribution invariant deep belief network for intelligent fault diagnosis of machines under new working conditions[J]. IEEE Transactions on Industrial Electronics，2020，68(3)：2617-2625.
[2] 詹瀛鱼，程良伦，王涛. 解相关多频率经验模态分解的故障诊断性能优化方法[J]. 振动与冲击，2020，39(1)：115-122. ZHAN Yingyu，CHENG Lianglun，WANG Tao. Fault diagnosis performance optimization method based on decorrelation multi-frequency EMD[J]. Journal of Vibration and Shock，2020，39(1)：115-122.
[3] GUO Yanping，XIONG Yu，SONG Guocui. Rolling bearing fault diagnosis with EMD-based fault characteristic frequency difference analysis[J]. Mechanics and Materials，2014，596(9)：437-441.
[4] CHEN Gang，LIU Mei，CHEN Jin. Frequency-temporal-logic-based bearing fault diagnosis and fault interpretation using bayesian optimization with bayesian neural network[J]. Mechanical Systems and Signal Processing，2020，145：106951.
[5] 王贡献，张淼，胡志辉，等. 基于多尺度均值排列熵和参数优化支持向量机的轴承故障诊断[J]. 振动与冲击，2022，41(1)：221-228. WANG Gongxian，ZHANG Miao，HU Zhihui，et al. Bearing fault diagnosis based on multi-scale mean permutation entropy and parametric optimization SVM[J]. Journal of Vibration and Shock，2022，41(1)： 221-228.
[6] ROY S S，DEY S，CHATTERJEE S. Autocorrelation aided random forest classifier-based bearing fault detection framework[J]. IEEE Sensors Journal，2020，20(18)：10792-10800.
[7] HAN Dongying，GUO Xiaoci，SHI Peiming. An intelligent fault diagnosis method of variable condition gearbox based on improved DBN combined with WPEE and MPE[J]. IEEE Access，2020，8：131299-131309.
[8] LI Yingbin，ZOU Li，JIANG Li，et al. Fault diagnosis of rotating machinery based on combination of deep belief network and one-dimensional convolutional neural network[J]. IEEE Access，2019，7：165710-165723.
[9] SAUFI S R，AHMAD Z A B，LEONG M S，et al. Low-speed bearing fault diagnosis based on ArSSAE model using acoustic emission and vibration signals[J]. IEEE Access，2019，7：46885-46897.
[10] WANG Hui，XU Jiawen，YAN Ruqiang，et al. A new intelligent bearing fault diagnosis method using SDP representation and SE-CNN[J]. IEEE Transactions on Instrumentation and Measurement，2020，69(5)：2377-2389.
[11] PENG Dandan，WANG Huan，LIU Zhiliang，et al. Multibranch and multiscale CNN for fault diagnosis of wheelset bearings under strong noise and variable load condition[J]. IEEE Transactions on Industrial Informatics，2020，16(7)：4949-4960.
[12] 张俊红，孙诗跃，朱小龙，等. 基于改进卷积神经网络的柴油机故障诊断方法研究[J]. 振动与冲击，2022，41(6)：139-146. ZHANG Junhong，SUN Shiyue，ZHU Xiaolong，et al. Diesel engine fault diagnosis based on an improved convolutional neural network[J]. Journal of Vibration and Shock，2022，41(6)：139-146.
[13] DAS D，LEE C S. Sample-to-sample correspondence for unsupervised domain adaptation[J]. Engineering Applications of Artificial Intelligence，2018，73：80-91.
[14] LI Rui，JIAO Qianfen，CAO Wenming，et al. Model adaptation：Unsupervised domain adaptation without source data[C]// IEEE/CVF Conference on Computer Vision and Pattern Recognition，Seattle：IEEE，2020.
[15] LI Yibin，SONG Yan，JIA Lei，et al. Intelligent fault diagnosis by fusing domain adversarial training and maximum mean discrepancy via ensemble learning[J]. IEEE Transactions on Industrial Informatics，2021,17(4)：2833-2841.
[16] LIU Zhaohua，LU Biliang，WEI Hualiang，et al. Deep adversarial domain adaptation model for bearing fault diagnosis[J]. IEEE Transactions on Systems，Man，and Cybernetics：Systems，2021，51(7)：4217-4226.
[17] HE Xinwei，ZHOU Yang，ZHOU Zhichao. Triplet-center loss for multi-view3D object retrieval[C]// IEEE/CVF Conference on Computer Vision and Pattern Recognition. SaltLake：IEEE，2018：1945-1954.
[18] HAN Zou，YANG Jianfeng，WU Xiaojian. Unsupervised energy-based adversarial domaina daptation for cross-domain text classification[C/CD]//The Joint Conference of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (ACL-IJCNLP).Online：ACL，2021.
[19] LECUN Y，BOTTOU L，BENGIO Y，et al. Gradient- based learning applied to document recognition[J]. Proceedings of the IEEE，1998，86(11)：2278-2324.
[20] TZENG E，HOFFMAN J，ZHANG Ning，et al. Deep domain confusion：maximizing for domain invariance[J]. Arxiv,2014.https://doi.org/10.48550/arXiv.1412.3474
[21] LONG Mingsheng，CAO Yue，WANG Jianmin，et al. Learning transferable features with deep adaptation networks[C]// Proceedings of the 32nd International Conference on Machine Learning. Lille：PMLR，2015，37：95-105.
[22] GANIN Y，EVGENIYA U，HANA A，et al. Domain adversarial training of neural networks[J]. The Journal of Machine Learning Research，2016，17(1)：2096-2030.
[23] PENG Xingchao，BAI Qinxun，XIA Xide，et al. Moment matching for multi-source domain adaption[C]// Proceedings of the IEEE/CVF International Conference on Computer Vision(ICCV).Seoul：IEEE，2019：1406-1415.
[24] ZHU Yongchun，ZHUANG Fuzhen，WANG Deqing. Aligning domain-specific distribution and classifier for cross-domain classification from multiple sources[C]//The Thirty Third AAAI Conference on Artificial Intelligence. Hawaii：AAAI，2019：5989-5996.