Similarity-based meta-learning network with adversarial domain adaptation for cross-domain fault identification

摘要

With wide applications of intelligent methods in mechanical fault diagnosis, satisfactory results have been achieved. However, complicated and diverse practical working conditions would significantly reduce the performance of the diagnostic model that works well in the laboratory, i.e. domain shift occurs. To address the problem, this paper proposed a novel similarity-based meta-learning network with adversarial domain adaptation for cross-domain fault identification. The proposed domain-adversarial similarity-based meta-learning network (DASMN) consists of three modules: a feature encoder, a classifier and a domain discriminator. First, the encoder and the classifier implement the similarity-based meta-learning algorithm, in while the good generalization ability for unseen tasks is obtained. Then, adversarial domain adaptation is conducted by minimizing and maximizing the domain-discriminative error adversarially, which takes unlabeled source data and target data as inputs. The effectiveness of DASMN is evaluated by multiple cross-domain cases using three bearing vibration datasets and is compared with five well-established methods. Experimental results demonstrate the availability and outstanding generalization ability of the proposed method for cross-domain fault identification.