CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism.
In industrial production, obtaining sufficient bearing fault signals is often extremely difficult, leading to a significant degradation in the performance of traditional deep learning-based fault diagnosis models. Many recent studies have shown that data augmentation using generative adversarial net...
Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2025-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://doi.org/10.1371/journal.pone.0319202 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850201885092347904 |
|---|---|
| author | Shun Yu Zi Li Jialin Gu Runpu Wang Xiaoyu Liu Lin Li Fusen Guo Yuheng Ren |
| author_facet | Shun Yu Zi Li Jialin Gu Runpu Wang Xiaoyu Liu Lin Li Fusen Guo Yuheng Ren |
| author_sort | Shun Yu |
| collection | DOAJ |
| description | In industrial production, obtaining sufficient bearing fault signals is often extremely difficult, leading to a significant degradation in the performance of traditional deep learning-based fault diagnosis models. Many recent studies have shown that data augmentation using generative adversarial networks (GAN) can effectively alleviate this problem. However, the quality of generated samples is closely related to the performance of fault diagnosis models. For this reason, this paper proposes a new GAN-based small-sample bearing fault diagnosis method. Specifically, this study proposes a continuous wavelet convolution strategy (CWCL) instead of the traditional convolution operation in GAN, which can additionally capture the signal's frequency domain features. Meanwhile, this study designed a new multi-size kernel attention mechanism (MSKAM), which can extract the features of bearing vibration signals from different scales and adaptively select the features that are more important for the generation task to improve the accuracy and authenticity of the generated signals. In addition, the structural similarity index (SSIM) is adopted to quantitatively evaluate the quality of the generated signal by calculating the similarity between the generated signal and the real signal in both the time and frequency domains. Finally, we conducted extensive experiments on the CWRU and MFPT datasets and made a comprehensive comparison with existing small-sample bearing fault diagnosis methods, which verified the effectiveness of the proposed approach. |
| format | Article |
| id | doaj-art-69910b671eef4c3685f43ec5f689a7c4 |
| institution | OA Journals |
| issn | 1932-6203 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-69910b671eef4c3685f43ec5f689a7c42025-08-20T02:11:54ZengPublic Library of Science (PLoS)PLoS ONE1932-62032025-01-01204e031920210.1371/journal.pone.0319202CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism.Shun YuZi LiJialin GuRunpu WangXiaoyu LiuLin LiFusen GuoYuheng RenIn industrial production, obtaining sufficient bearing fault signals is often extremely difficult, leading to a significant degradation in the performance of traditional deep learning-based fault diagnosis models. Many recent studies have shown that data augmentation using generative adversarial networks (GAN) can effectively alleviate this problem. However, the quality of generated samples is closely related to the performance of fault diagnosis models. For this reason, this paper proposes a new GAN-based small-sample bearing fault diagnosis method. Specifically, this study proposes a continuous wavelet convolution strategy (CWCL) instead of the traditional convolution operation in GAN, which can additionally capture the signal's frequency domain features. Meanwhile, this study designed a new multi-size kernel attention mechanism (MSKAM), which can extract the features of bearing vibration signals from different scales and adaptively select the features that are more important for the generation task to improve the accuracy and authenticity of the generated signals. In addition, the structural similarity index (SSIM) is adopted to quantitatively evaluate the quality of the generated signal by calculating the similarity between the generated signal and the real signal in both the time and frequency domains. Finally, we conducted extensive experiments on the CWRU and MFPT datasets and made a comprehensive comparison with existing small-sample bearing fault diagnosis methods, which verified the effectiveness of the proposed approach.https://doi.org/10.1371/journal.pone.0319202 |
| spellingShingle | Shun Yu Zi Li Jialin Gu Runpu Wang Xiaoyu Liu Lin Li Fusen Guo Yuheng Ren CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. PLoS ONE |
| title | CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. |
| title_full | CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. |
| title_fullStr | CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. |
| title_full_unstemmed | CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. |
| title_short | CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism. |
| title_sort | cwms gan a small sample bearing fault diagnosis method based on continuous wavelet transform and multi size kernel attention mechanism |
| url | https://doi.org/10.1371/journal.pone.0319202 |
| work_keys_str_mv | AT shunyu cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT zili cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT jialingu cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT runpuwang cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT xiaoyuliu cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT linli cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT fusenguo cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism AT yuhengren cwmsganasmallsamplebearingfaultdiagnosismethodbasedoncontinuouswavelettransformandmultisizekernelattentionmechanism |