Graph Mutual Mapping Transfer Networks for machine fault diagnosis

Zhaoqian Wu,Haolun Chen,Jie Han,Shaohui Zhang,Jinglun Liang,Fei Jiang
DOI: https://doi.org/10.1109/tii.2023.3271734
IF: 12.3
2023-01-01
IEEE Transactions on Industrial Informatics
Abstract:Graph neural network (GNN) as the mainstream method of graph processing in deep learning plays an important role in processing non-Euclidean spatial data. However, insufficient training samples weaken the model's transfer performance and feature generalization ability. What's more, the complicated calculations of the convolutional neural network (CNN) used in GNN lead to low convergence speed. Therefore, this paper proposes a novel neural network, named graph mutual mapping transfer network (GMMTN), to improve the transfer effect and increase the convergence rate. The proposed method first maps the source and target domain samples to make the samples evenly distributed in different domains, which improves the overall transfer effect. After that, the graph generation and multi-receptive field map convolutional network modeling are carried out. Then, the classifier and domain discriminator are used for unsupervised adaptation to improve model generalization. Experimental results show that the proposed GMMTN method has significant fault diagnosis accuracy.
automation & control systems,computer science, interdisciplinary applications,engineering, industrial
What problem does this paper attempt to address?