Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2
A rolling bearing fault diagnosis model based on two-step transfer learning and EfficientNetV2 (TSTE) is proposed for the real fault diagnosis environment in engineering, where the scarcity of available data leads to the low accuracy of the intelligent diagnosis model in bearing health status diagno...
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
Editorial Office of Journal of Mechanical Transmission
2023-01-01
|
| Series: | Jixie chuandong |
| Subjects: | |
| Online Access: | http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.07.024 |
| _version_ | 1797743686881116160 |
|---|---|
| author | Du Kangning Ning Shaohui |
| author_facet | Du Kangning Ning Shaohui |
| author_sort | Du Kangning |
| collection | DOAJ |
| description | A rolling bearing fault diagnosis model based on two-step transfer learning and EfficientNetV2 (TSTE) is proposed for the real fault diagnosis environment in engineering, where the scarcity of available data leads to the low accuracy of the intelligent diagnosis model in bearing health status diagnosis. Firstly, the model is trained on the full life time bearing data set and then the model shallow weights are freezed to train it on the multi-condition bearing data set for the first transfer learning. Secondly, by constructing a class-imbalance dataset, the impact of scarcity of available data on fault diagnosis performance is studied in actual fault environments is studied. Then, the synthetic minority oversampling technique (SMOTE) oversampling method and the edited nearest neighbors (ENN) under the sampling method are proposed to expand the fault data, reconstructing the class-imbalanced dataset into a class-balanced dataset. Finally, the model is trained on the class-balanced dataset, freezing the model's bottom weights and training the model deeper for a second transfer learning, enabling the model to take control of the failure characteristics of the balanced dataset. The experiments are evaluated by a variety of metrics, while comparing it with other methods, and using the Grad-CAM method for feature visualization. The results show that the proposed method is able to transfer the fault diagnosis knowledge accumulated by the model in a laboratory environment to actual engineering equipment. It is suitable for the diagnosis of rolling bearing faults in situations where test data is scarce. |
| first_indexed | 2024-03-12T14:59:00Z |
| format | Article |
| id | doaj.art-eac266e1e84f460984ceb2c922a18219 |
| institution | Directory Open Access Journal |
| issn | 1004-2539 |
| language | zho |
| last_indexed | 2024-03-12T14:59:00Z |
| publishDate | 2023-01-01 |
| publisher | Editorial Office of Journal of Mechanical Transmission |
| record_format | Article |
| series | Jixie chuandong |
| spelling | doaj.art-eac266e1e84f460984ceb2c922a182192023-08-14T10:02:24ZzhoEditorial Office of Journal of Mechanical TransmissionJixie chuandong1004-25392023-01-014716817640048487Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2Du KangningNing ShaohuiA rolling bearing fault diagnosis model based on two-step transfer learning and EfficientNetV2 (TSTE) is proposed for the real fault diagnosis environment in engineering, where the scarcity of available data leads to the low accuracy of the intelligent diagnosis model in bearing health status diagnosis. Firstly, the model is trained on the full life time bearing data set and then the model shallow weights are freezed to train it on the multi-condition bearing data set for the first transfer learning. Secondly, by constructing a class-imbalance dataset, the impact of scarcity of available data on fault diagnosis performance is studied in actual fault environments is studied. Then, the synthetic minority oversampling technique (SMOTE) oversampling method and the edited nearest neighbors (ENN) under the sampling method are proposed to expand the fault data, reconstructing the class-imbalanced dataset into a class-balanced dataset. Finally, the model is trained on the class-balanced dataset, freezing the model's bottom weights and training the model deeper for a second transfer learning, enabling the model to take control of the failure characteristics of the balanced dataset. The experiments are evaluated by a variety of metrics, while comparing it with other methods, and using the Grad-CAM method for feature visualization. The results show that the proposed method is able to transfer the fault diagnosis knowledge accumulated by the model in a laboratory environment to actual engineering equipment. It is suitable for the diagnosis of rolling bearing faults in situations where test data is scarce.http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.07.024Rolling bearing;Two-step transfer learning;EfficientNetV2;Class-imbalance;Resample |
| spellingShingle | Du Kangning Ning Shaohui Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 Jixie chuandong Rolling bearing;Two-step transfer learning;EfficientNetV2;Class-imbalance;Resample |
| title | Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 |
| title_full | Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 |
| title_fullStr | Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 |
| title_full_unstemmed | Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 |
| title_short | Fault Diagnosis of Rolling Bearings Based on Two-step Transfer Learning and EfficientNetV2 |
| title_sort | fault diagnosis of rolling bearings based on two step transfer learning and efficientnetv2 |
| topic | Rolling bearing;Two-step transfer learning;EfficientNetV2;Class-imbalance;Resample |
| url | http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.07.024 |
| work_keys_str_mv | AT dukangning faultdiagnosisofrollingbearingsbasedontwosteptransferlearningandefficientnetv2 AT ningshaohui faultdiagnosisofrollingbearingsbasedontwosteptransferlearningandefficientnetv2 |