Gear Fault Diagnosis Method Based on Deep Transfer Learning
Aiming at the problem of insufficient gear fault samples, a fault diagnosis method of transfer learning based on Hilbert-Huang spectrum and pre-trained VGG16 model is proposed. Firstly, the intrinsic mode function (IMF) is obtained by Empirical Mode Decomposition (EMD) of vibration signals, and the...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
Editorial Office of Journal of Mechanical Transmission
2023-05-01
|
| Series: | Jixie chuandong |
| Subjects: | |
| Online Access: | http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.05.021 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841547023751315456 |
|---|---|
| author | Liu Shihao Wang Xiyang Gong Tingkai |
| author_facet | Liu Shihao Wang Xiyang Gong Tingkai |
| author_sort | Liu Shihao |
| collection | DOAJ |
| description | Aiming at the problem of insufficient gear fault samples, a fault diagnosis method of transfer learning based on Hilbert-Huang spectrum and pre-trained VGG16 model is proposed. Firstly, the intrinsic mode function (IMF) is obtained by Empirical Mode Decomposition (EMD) of vibration signals, and the time spectrum is obtained by Hilbert transform of IMF with the largest correlation coefficient. Secondly, pre-trained VGG16 is used to extract Hilbert-Huang spectrum image features of gears under varying loads and under various health conditions. Finally, the global average pooling layer is used to replace partial full connection layer of VGG16 model for classification output. Experimental results show that with a small amount of sample data, the accuracy of gear fault diagnosis reaches 98.86%, which is better than the transfer learning methods such as TLCNN and Tran VGG-19. It is proved that the method presented in this paper has some research value in gear fault diagnosis. |
| format | Article |
| id | doaj-art-4b23e8951fd1481c989a57dc3ac84f01 |
| institution | Kabale University |
| issn | 1004-2539 |
| language | zho |
| publishDate | 2023-05-01 |
| publisher | Editorial Office of Journal of Mechanical Transmission |
| record_format | Article |
| series | Jixie chuandong |
| spelling | doaj-art-4b23e8951fd1481c989a57dc3ac84f012025-01-10T14:57:36ZzhoEditorial Office of Journal of Mechanical TransmissionJixie chuandong1004-25392023-05-014713414233200903Gear Fault Diagnosis Method Based on Deep Transfer LearningLiu ShihaoWang XiyangGong TingkaiAiming at the problem of insufficient gear fault samples, a fault diagnosis method of transfer learning based on Hilbert-Huang spectrum and pre-trained VGG16 model is proposed. Firstly, the intrinsic mode function (IMF) is obtained by Empirical Mode Decomposition (EMD) of vibration signals, and the time spectrum is obtained by Hilbert transform of IMF with the largest correlation coefficient. Secondly, pre-trained VGG16 is used to extract Hilbert-Huang spectrum image features of gears under varying loads and under various health conditions. Finally, the global average pooling layer is used to replace partial full connection layer of VGG16 model for classification output. Experimental results show that with a small amount of sample data, the accuracy of gear fault diagnosis reaches 98.86%, which is better than the transfer learning methods such as TLCNN and Tran VGG-19. It is proved that the method presented in this paper has some research value in gear fault diagnosis.http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.05.021Transfer learningVGG16Hilbert-Huang spectrumGear fault diagnosisGlobal average pooling |
| spellingShingle | Liu Shihao Wang Xiyang Gong Tingkai Gear Fault Diagnosis Method Based on Deep Transfer Learning Jixie chuandong Transfer learning VGG16 Hilbert-Huang spectrum Gear fault diagnosis Global average pooling |
| title | Gear Fault Diagnosis Method Based on Deep Transfer Learning |
| title_full | Gear Fault Diagnosis Method Based on Deep Transfer Learning |
| title_fullStr | Gear Fault Diagnosis Method Based on Deep Transfer Learning |
| title_full_unstemmed | Gear Fault Diagnosis Method Based on Deep Transfer Learning |
| title_short | Gear Fault Diagnosis Method Based on Deep Transfer Learning |
| title_sort | gear fault diagnosis method based on deep transfer learning |
| topic | Transfer learning VGG16 Hilbert-Huang spectrum Gear fault diagnosis Global average pooling |
| url | http://www.jxcd.net.cn/thesisDetails#10.16578/j.issn.1004.2539.2023.05.021 |
| work_keys_str_mv | AT liushihao gearfaultdiagnosismethodbasedondeeptransferlearning AT wangxiyang gearfaultdiagnosismethodbasedondeeptransferlearning AT gongtingkai gearfaultdiagnosismethodbasedondeeptransferlearning |