A Multi-Scale Graph Convolutional Neural Network Framework for Fault Diagnosis of Rolling Bearing

• Published in: IEEE transactions on instrumentation and measurement p. 1
• Main Authors: Yin, Peizhe, Nie, Jie, Liang, Xinyue, Yu, Shusong, Wang, Chenglong, Nie, Weizhi, Ding, Xiangqian
• Format: Journal Article
• Language: English
• Published: IEEE 03-07-2023
• Subjects: Bayesian; Convolution; Convolutional neural networks; Deep learning; Fault diagnosis; Feature extraction; Graph convolutional neural network; horizontal visibility graph; multi-scale; Rolling bearings; Vibrations
• ISSN: 0018-9456
• DOI: 10.1109/TIM.2023.3291768

Fault diagnosis for rolling bearing has been an important engineering problem through decades. To detect the damaged bearing surface, engineers analyze the features from the extracted vibration signals of the machine. As artificial intelligence rapidly develops and provides favorable effects in data analytics, using deep learning technology to attack fault diagnosis problems has attracted increasing research interests in recent years. However most existing methods do not provide satisfactory performance in mining the relationship between the signals. Even graph convolutional networks that can perform well in non-Euclidean Spaces have limitations, since it only considers extracting features from a single scale, ignoring the potential relationship between signals at various scales. For salvation, we propose a multi-scale graph convolutional network(MS-GCN) for this specific problem. In MS-GCN, we put forward a multi-scale feature extraction module, which extracts the features of vibration signals two times under diverse receptive field ranges, ensuring that the regularities of signal features can be fully discovered. In addition, we devise a multi-scale graph iteration module, which incorporates two single-scale graph iteration modules and a cross-scale graph iteration module, which can fully retain local features based on extensive mining of global information. We also propose a mutual fusion module based on the Bayesian method, which forcibly manipulates various features as a prior and achieves an convincing result. The horizontal visibility graph (HVG) method is used to construct graph models at multiple scales, which can better capture the hidden information of signal vibrations. In experiments, we verify the proposed model on the CWRU dataset to evaluate the method performance. The results show that our model significantly improved accuracy compared with other state-of-the-art methods.