Multi-input Parallel Graph Neural Network for Semi-supervised Rolling Bearing Fault Diagnosis
Shouyang Bao,Jing Feng,Xiaobin Xu,Pingzhi Hou,Zhenjie Zhang,Jianfang Meng,Felix Steyskal
DOI: https://doi.org/10.1088/1361-6501/acb5b7
IF: 2.398
2023-01-25
Measurement Science and Technology
Abstract:Rolling bearing fault diagnosis is the key technology to ensure the reliable, efficient and sustainable operation of the rotating machinery. Previously, many fault diagnosis methods are proposed based on vibration signal analysis from the perspective of data-driven. However, these methods normally take signals of multiple sensors as a whole for feature extraction without considering the relationship among the samples. These drawbacks lead to insufficient feature mining thereby affecting the accuracy of fault diagnosis. Moreover, these methods need large numbers of labelled sample to achieve high diagnosis accuracy, which need extensive human labour and is impractical in many real-world applications. To address these issues, a semi-supervised rolling bearing fault diagnosis method based on multi-input parallel graph neural network is proposed in this paper. In the proposed model, signals of multiple sensors are treated separately thus features will be extracted parallelly in a more sufficient way. Then signal of each sensor is constructed into a graph based on limited radius nearest neighbor which will add extra relation information to aid fault diagnosis. In addition, with the aid of graph convolutional neural network, the proposed method is able to achieve more accurate diagnosis in the case of few labelled data than the comparison methods. Finally, the proposed model is evaluated on rolling bearing dataset provided by Case Western Reserve University. Compared with some classical fault diagnosis methods, the proposed model can improve the diagnosis accuracy reaching to more than 99% of accuracy even when the proportion of training samples is only 20%.
engineering, multidisciplinary,instruments & instrumentation