A Novel Method of Bearing Fault Diagnosis in Time-Frequency Graphs Using InceptionResnet and Deformable Convolution Networks
Shaobo Li, Wanli Yang, Ansi Zhang, Huibin Liu, Jinyuan Huang, Chuanjiang Li, Jianjun Hu
Abstract
Bearing fault diagnosis has attracted increasing attention due to its importance in the health status of rotating machinery. The data-driven models based on deep learning (DL) have become more and more intelligent in the field of fault diagnosis, and among them convolutional neural network (CNN) has been widely used in recent researches. However, traditional CNN is not easy to capture right fault features due to their fixed geometric structures, especially under complex working conditions in fault diagnosis. To address these challenges, we propose a novel model by combining InceptionResnetV2 with Deformable Convolution Networks, named DeIN. We replace the basic form of convolution with deformable convolution in specific layers, and a main classifier and an auxiliary classifier are designed to output the classification result of our proposed model, to adapt to the non-rigid characters and larger receptive field in time-frequency graph (TFG). Experimentally, the one-dimensional signals are transformed into TFGs and as input of the proposed model, and this aims to find useful features during the training process. To verify the generalization ability of the proposed model, we apply a set of cross-over tests based on two popular datasets, and our model achieved 99.87% and 94.52% highest-precision fault classification results comparing with other state-of-the-art CNN models.