Bearing Fault Diagnosis Based on Improved Gated Convolutional Network with Imbalanced Data
Intelligent methods based on deep learning have been widely used in the fault diagnosis of rolling bearings.However,the actual data from industrial plant is severely imbalanced,which will greatly affect the diagnostic performance of the model.To solve this problem,a fault diagnosis method based on improved gated convolutional neural network(IGCNN)is proposed.Firstly,an improved gated convolution layer is proposed for feature extraction,with the batch normalization layer applied in this layer to enhance the generalization performance of the model.Then,the label-distribution-aware margin(LDAM)loss function is employed to raise the sensitivity of the model to the minority class and mitigate the influence of imbalanced data on the model.The proposed method is applied to two sets of faulty bearing data.It is found that when the imbalance ratio of the data sets is 20:1,the recognition accuracy of the proposed method can still achieve 92.71%and 94.47%.While the other compared mainstream deep learning models only achieve 60%-72%accuracy.It shows that the proposed method has strong diagnostic capability and robustness for the datasets with severe imbalance.