Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model
Abstract This paper proposes a fault diagnosis method for rotating machinery that integrates transfer learning with the ConvNeXt model (TL-CoCNN), addressing challenges such as small sample sizes and varying operating conditions. To meet the input requirements of the model while minimizing feature l...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Nature Portfolio
2025-01-01
|
| Series: | Scientific Reports |
| Subjects: | |
| Online Access: | https://doi.org/10.1038/s41598-024-84783-5 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841559647249498112 |
|---|---|
| author | Zhikai Xing Yongbao Liu Qiang Wang Junqiang Fu |
| author_facet | Zhikai Xing Yongbao Liu Qiang Wang Junqiang Fu |
| author_sort | Zhikai Xing |
| collection | DOAJ |
| description | Abstract This paper proposes a fault diagnosis method for rotating machinery that integrates transfer learning with the ConvNeXt model (TL-CoCNN), addressing challenges such as small sample sizes and varying operating conditions. To meet the input requirements of the model while minimizing feature loss, an alternative approach to visualizing vibration data is introduced. Specifically, RGB images are synthesized from time-domain, frequency-domain, and time-frequency domain representations of the original signal, which are subsequently used as the input dataset. The fault diagnosis process leverages a pre-trained ConvNeXt model, initially trained on the ImageNet dataset, and fine-tunes its parameters using the synthesized RGB images to perform the fault classification task. Experimental results demonstrate that this data visualization method extracts more fault-related information compared to traditional time-domain and frequency-domain techniques, without the need to augment the sample size. The TL-CoCNN model achieves superior recognition accuracy when evaluated in terms of training time and model size across multiple test datasets. As an end-to-end fault diagnosis system, TL-CoCNN significantly enhances the feature representation capability of complex signals, showing promising potential for practical applications in fault detection and diagnosis. |
| format | Article |
| id | doaj-art-2c5c77502fc64f7c94282e09f7bfe57f |
| institution | Kabale University |
| issn | 2045-2322 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Nature Portfolio |
| record_format | Article |
| series | Scientific Reports |
| spelling | doaj-art-2c5c77502fc64f7c94282e09f7bfe57f2025-01-05T12:16:34ZengNature PortfolioScientific Reports2045-23222025-01-0115111810.1038/s41598-024-84783-5Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt modelZhikai Xing0Yongbao Liu1Qiang Wang2Junqiang Fu3Department of Power Engineering, Naval University of EngineeringDepartment of Power Engineering, Naval University of EngineeringDepartment of Power Engineering, Naval University of EngineeringDepartment of Power Engineering, Naval University of EngineeringAbstract This paper proposes a fault diagnosis method for rotating machinery that integrates transfer learning with the ConvNeXt model (TL-CoCNN), addressing challenges such as small sample sizes and varying operating conditions. To meet the input requirements of the model while minimizing feature loss, an alternative approach to visualizing vibration data is introduced. Specifically, RGB images are synthesized from time-domain, frequency-domain, and time-frequency domain representations of the original signal, which are subsequently used as the input dataset. The fault diagnosis process leverages a pre-trained ConvNeXt model, initially trained on the ImageNet dataset, and fine-tunes its parameters using the synthesized RGB images to perform the fault classification task. Experimental results demonstrate that this data visualization method extracts more fault-related information compared to traditional time-domain and frequency-domain techniques, without the need to augment the sample size. The TL-CoCNN model achieves superior recognition accuracy when evaluated in terms of training time and model size across multiple test datasets. As an end-to-end fault diagnosis system, TL-CoCNN significantly enhances the feature representation capability of complex signals, showing promising potential for practical applications in fault detection and diagnosis.https://doi.org/10.1038/s41598-024-84783-5Fault diagnosisTransfer learningVarying operating conditionsPre-trained ConvNeXt model |
| spellingShingle | Zhikai Xing Yongbao Liu Qiang Wang Junqiang Fu Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model Scientific Reports Fault diagnosis Transfer learning Varying operating conditions Pre-trained ConvNeXt model |
| title | Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model |
| title_full | Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model |
| title_fullStr | Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model |
| title_full_unstemmed | Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model |
| title_short | Fault diagnosis of rotating parts integrating transfer learning and ConvNeXt model |
| title_sort | fault diagnosis of rotating parts integrating transfer learning and convnext model |
| topic | Fault diagnosis Transfer learning Varying operating conditions Pre-trained ConvNeXt model |
| url | https://doi.org/10.1038/s41598-024-84783-5 |
| work_keys_str_mv | AT zhikaixing faultdiagnosisofrotatingpartsintegratingtransferlearningandconvnextmodel AT yongbaoliu faultdiagnosisofrotatingpartsintegratingtransferlearningandconvnextmodel AT qiangwang faultdiagnosisofrotatingpartsintegratingtransferlearningandconvnextmodel AT junqiangfu faultdiagnosisofrotatingpartsintegratingtransferlearningandconvnextmodel |