Multi-source domain adversarial graph convolutional networks for rolling mill health states diagnosis under variable working conditions

Abstract

As the rolling mill often encounters variable and complicated working conditions and shock loads, unsupervised domain adaptive (UDA) methods are imperative in its health monitoring. However, efforts of applying UDA methods on the rolling mill are negligible, and many existing approaches have constraints in domain adaptation, domain label, and data construction that prevent meaningful features from being extracted. Hence, a multi-source domain adversarial graph convolutional networks framework (MSDAGCNs) is presented to overcome these challenges and combine three essential elements to achieve cross-domain health states diagnosis under variable working conditions. First, a shared feature extract module is introduced to extract common features. Then, the features are input to a multi-source feature extract module to extract the data construction from the graphs generated by a graph construction module. Meanwhile, a multi-source domain adversarial classifier module is modeled to extract multi-source invariant features and classify them. After that, the local maximum mean discrepancy is employed to align the domain categories. Next, a task classifier module integrates the results of the multi-source classifier for reliable health state diagnosis. Results on the two cases can verify that the proposed MSDAGCNs can not only outperform other state-of-the-art methods, but also extract domain-invariant knowledge. Compared with the best-performing method, the proposed method can boost accuracy by 0.53% and 0.83% in the simplest task of the two case studies, respectively. Furthermore, the arrangement of sensors on the rolling mill is discussed to select the optimal location for collecting vibrations.