Semi-supervised joint adaptation transfer network with conditional adversarial learning for rotary machine fault diagnosis
Chun Liu, Shaojie Li, Hongtian Chen, Xianchao Xiu, Peng Chen
Abstract
At present, artificial intelligence is booming and has made major breakthroughs in fault diagnosis scenarios. However, the high diagnostic accuracy of most mainstream fault diagnosis methods must rely on sufficient data to train the diagnostic models. In addition, there is another assumption that needs to be satisfied: the consistency of training and test data distribution. When these prerequisites are not available, the effectiveness of the diagnosis model declines dramatically. To address this problem, we propose a semi-supervised joint adaptation transfer network with conditional adversarial learning for rotary machine fault diagnosis. To fully utilize the fault features implied in unlabeled data, pseudo-labels are generated through threshold filtering to obtain an initial pre-trained model. Then, a joint domain adaptation transfer network module based on conditional adversarial learning and distance metric is introduced to ensure the consistency of the distribution in two different domains. Lastly, in three groups of experiments with different settings: a single fault with variable load, a single fault with variable speed, and a mixed fault with variable speed and load, it was confirmed that our method can obtain competitive diagnostic performance.