Cross-domain Few-Shot Fault Diagnosis Based on Meta-Learning and Domain Adversarial Graph Convolutional Network

Junwei Hu,Weigang Li,Yong Zhang,Zhiqiang Tian
DOI: https://doi.org/10.1016/j.engappai.2024.108970
IF: 8
2024-01-01
Engineering Applications of Artificial Intelligence
Abstract:Problems such as small samples and variable working conditions arise in complex practical mechanical fault diagnosis scenarios. Although the domain-adaptive method has an excellent effect in the task of domain deviation diagnosis, it still needs a large number of source domain marker samples. Therefore, a new meta-learning domain-adversarial graph convolutional network (MDGCN) is proposed to solve the above problem of practical fault diagnosis for cross-domain few-shot (CDFS). Firstly, the graph convolutional network is used to learn the relationship between the structural features of the data from the input signal to construct the instance graph. Then, domain-adversarial training is used to minimize the domain discrimination error of source and target data for domain adaptation. Simultaneously, a scaled distance metric function is used to compute the similarity between the faulty sample and the sample prototype for fault diagnosis. Finally, the MDGCN model is evaluated using CDFS cases from three mechanical vibration datasets. Experimental results show that the proposed method not only achieves the best performance in comparison, but also extracts the features of CDFS faults for domain-adaptive diagnosis.
What problem does this paper attempt to address?