Gaussian Mixture Variational-Based Transformer Domain Adaptation Fault Diagnosis Method and Its Application in Bearing Fault Diagnosis

Abstract

Unsupervised domain adaptation is widely used for fault diagnosis under variable working conditions. However, loss oscillation and slow convergence, which are caused by the dynamically varying alignment of targets during domain adaptation, are ignored. Therefore, a Gaussian mixture variational based transformer domain adaptation (GMVTDA) fault diagnosis method is proposed. A feature extractor based on transformer layers is designed to capture long-term dependency information and local features. Subsequently, a domain alignment term is proposed to project the features learned from both working conditions into the common assistance distribution and make them follow the same distribution after the alignment process. Additionally, considering that fault diagnosis is a multiclassification process, a Gaussian mixture is utilized to build the common assistance distribution. Ultimately, the proposed GMVTDA is applied to bearing fault diagnosis under variable working conditions, and the experimental results prove its effectiveness.