Fusion with Joint Distribution and Adversarial Networks: A New Transfer Learning Approach for Intelligent Fault Diagnosis



Published Sep 4, 2023
Xueyi Li Tianyu Yu David He Zhijie Xie Xiangwei Kong


Bearings and gears are important components in rotating machinery, and the diagnosis of faults in bearings and gears has always been an important topic. Currently, data-driven fault diagnosis is a better method. However, under actual working conditions, domain shift can easily occur due to different operating conditions, leading to difficulties in transfer learning and significantly reducing the diagnostic performance of the model. Re-labeling the fault types of the model is time-consuming and costly. To overcome these difficulties, a new unsupervised transfer learning framework based on the fusion of joint distribution and adversarial networks has been introduced for the fault diagnosis of bearings and gears in rotating machinery. The joint adaptation network learns the transfer network by aligning the joint distribution of multiple specific domain layers across domains, based on Joint Maximum Mean Discrepancy (JMMD) to achieve domain alignment. At the same time, the domain classifier in the adversarial network is used to minimize the domain classification loss as domain distribution difference to minimize domain shift. The fusion of these two methods achieves domain alignment, reduces model training time, and improves the accuracy and stability of the model. The experimental results demonstrate that the proposed model framework exhibits excellent performance in detecting and classifying different types of faults. The new model framework also demonstrates outstanding performance across various fault detection and classification tasks.

Abstract 139 | PDF Downloads 141



Joint Maximum Mean Discrepancy, Conditional adversarial network, Convolutional neural network

Z. Wu and N. E. Huang, "Ensemble empirical mode decomposition: a noise-assisted data analysis method," Advances in adaptive data analysis, vol. 1, no. 01, pp. 1-41, 2009.

O. Janssens, V. Slavkovikj, B. Vervisch et al., "Convolutional neural network based fault detection for rotating machinery," Journal of Sound and Vibration, vol. 377, pp. 331-345, 2016.

Z. Zhao, Q. Zhang, X. Yu et al., "Applications of unsupervised deep transfer learning to intelligent fault diagnosis: A survey and comparative study," IEEE Transactions on Instrumentation and Measurement, vol. 70, pp. 1-28, 2021.

J. Zhou, X. Yang, L. Zhang et al., "Multisignal VGG19 network with transposed convolution for rotating machinery fault diagnosis based on deep transfer learning," Shock and Vibration, vol. 2020, pp. 1-12, 2020.

Y. Li, N. Wang, J. Shi et al., "Adaptive batch normalization for practical domain adaptation," Pattern Recognition, vol. 80, pp. 109-117, 2018.

W. Qian, S. Li, and X. Jiang, "Deep transfer network for rotating machine fault analysis," Pattern recognition, vol. 96, p. 106993, 2019.

J. Zhu, N. Chen, and C. Shen, "A new deep transfer learning method for bearing fault diagnosis under different working conditions," IEEE Sensors Journal, vol. 20, no. 15, pp. 8394-8402, 2019.

Y. Ganin and V. Lempitsky, "Unsupervised domain adaptation by backpropagation," in International conference on machine learning, 2015: PMLR, pp. 1180-1189.

M. Long, H. Zhu, J. Wang et al., "Deep transfer learning with joint adaptation networks," in International conference on machine learning, 2017: PMLR, pp. 2208-2217.

M. Long, Z. Cao, J. Wang et al., "Conditional adversarial domain adaptation," Advances in neural information processing systems, vol. 31, 2018.

X. Li, K. Su, Q. He, et al., "Research on Fault Diagnosis of Highway Bi-LSTM Based on Attention Mechanism," Eksplo atacja i Niezawodność – Maintenance and Reliability, vol. 25, no. 2, 2023.
Regular Session Papers