Changsheng Xi, Jie Yang, Xiaoxia Liang, Rahizar Bin Ramli, Shaoning Tian, Guojin Feng, Dong Zhen
{"title":"基于改进门控卷积神经网络的不平衡滚动轴承故障诊断","authors":"Changsheng Xi, Jie Yang, Xiaoxia Liang, Rahizar Bin Ramli, Shaoning Tian, Guojin Feng, Dong Zhen","doi":"10.1504/ijhm.2023.130520","DOIUrl":null,"url":null,"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.","PeriodicalId":29937,"journal":{"name":"International Journal of Hydromechatronics","volume":null,"pages":null},"PeriodicalIF":5.3000,"publicationDate":"2023-01-01","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"4","resultStr":"{\"title\":\"An improved gated convolutional neural network for rolling bearing fault diagnosis with imbalanced data\",\"authors\":\"Changsheng Xi, Jie Yang, Xiaoxia Liang, Rahizar Bin Ramli, Shaoning Tian, Guojin Feng, Dong Zhen\",\"doi\":\"10.1504/ijhm.2023.130520\",\"DOIUrl\":null,\"url\":null,\"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.\",\"PeriodicalId\":29937,\"journal\":{\"name\":\"International Journal of Hydromechatronics\",\"volume\":null,\"pages\":null},\"PeriodicalIF\":5.3000,\"publicationDate\":\"2023-01-01\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"4\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"International Journal of Hydromechatronics\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1504/ijhm.2023.130520\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q1\",\"JCRName\":\"ENGINEERING, MECHANICAL\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"International Journal of Hydromechatronics","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1504/ijhm.2023.130520","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"ENGINEERING, MECHANICAL","Score":null,"Total":0}
An improved gated convolutional neural network for rolling bearing fault diagnosis with imbalanced data
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.