An improved gated convolutional neural network for rolling bearing fault diagnosis with imbalanced data
by Changsheng Xi; Jie Yang; Xiaoxia Liang; Rahizar Bin Ramli; Shaoning Tian; Guojin Feng; Dong Zhen
International Journal of Hydromechatronics (IJHM), Vol. 6, No. 2, 2023

Abstract: To improve the ability of the deep learning model to handle imbalanced data, 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 normalisation (BN) layer applied to adjust the data distribution and enhance the generalisation performance of the model. Then, the feature learned by multiple gated convolution layers and pooling layers is fed to the fully connected layer for fault type identification. Finally, the label-distribution-aware margin (LDAM) loss function is employed to adjust the model being more sensitive to the minority class and mitigate the influence of imbalanced data on the model. Experimental validation is conducted using two bearing datasets. Results show that the proposed method is more robust than other fault diagnosis methods, with higher recognition accuracy in severely imbalanced dataset.

Online publication date: Tue, 25-Apr-2023

The full text of this article is only available to individual subscribers or to users at subscribing institutions.

 
Existing subscribers:
Go to Inderscience Online Journals to access the Full Text of this article.

Pay per view:
If you are not a subscriber and you just want to read the full contents of this article, buy online access here.

Complimentary Subscribers, Editors or Members of the Editorial Board of the International Journal of Hydromechatronics (IJHM):
Login with your Inderscience username and password:

Username:        Password:          Forgotten your password?

Want to subscribe?
A subscription gives you complete access to all articles in the current issue, as well as to all articles in the previous three years (where applicable). See our Orders page to subscribe.

If you still need assistance, please email subs@inderscience.com