RESEARCH PAPER
A Semi-supervised Fault Diagnosis Method for Rolling Bearing Using Time-Frequency Transform Enhanced Contrastive Learning
More details
Hide details
Submission date: 2025-03-09
Final revision date: 2025-05-03
Acceptance date: 2025-06-11
Online publication date: 2025-06-19
Publication date: 2025-06-19
Corresponding author
Chen Gao
School of Electromechanical and Transportation, Jiaxing Nanyang Polytechnic Institute, China
HIGHLIGHTS
- A semi-supervised learning method for bearing fault diagnosis was proposed.
- A STFT enhanced Contrastive Learning is employed to utilize large unlabeled samples.
- The diagnostic accuracy exceeded 99% when using 50 labeled samples per fault type.
KEYWORDS
TOPICS
ABSTRACT
This paper proposes a novel semi-supervised framework, time-frequency Contrastive Learning (CL), to address the challenge of accurate rolling bearing fault diagnosis under industrial small-sample conditions. Raw vibration signals are transformed into discriminative time-frequency images using short-time Fourier transform (STFT). A CL network with a ResNet18 model is pre-trained on a lot of unlabeled samples to learn generalized feature, and the ResNet18 model is fine-tuned using small labeled samples for fault classification. Experimental validation on bearing fault datasets demonstrates that the proposed STFT-CL method achieves above 99% diagnosis accuracy with only 50 labeled samples per fault type, outperforming conventional semi-supervised methods by 6-12%. The proposed method provides a potential solution to the "small sample dilemma" in industrial applications through the synergistic effect of physically driven signal processing and self supervised representation learning.
ACKNOWLEDGEMENTS
We gratefully thank Kumar et al. for their publicly available dataset of bearing faults.
FUNDING
This research was funded by the Science and Technology Plan Project of Jiaxing in China (Grant. 2024AY10010), and the Research Projects of Zhejiang Provincial Department of Education in China (Grant. Y202354102).