Time–Frequency Residual Self-Supervised Network for Bearing Fault Diagnosis with Limited Labeled Data
摘要
Conventional supervised learning methods for fault diagnosis require extensive labeled data, which is often unavailable due to the sparsity and high cost of labeling fault signals. This study aims to develop a self-supervised learning approach that can effectively extract discriminative features from unlabeled vibration data, enabling accurate fault diagnosis with minimal labeled samples.
MethodsA time–frequency residual self-supervised network (TFRSSN) is proposed, based on an improved MoBY framework. The Transformer backbone in the original MoBY architecture is replaced with a 1D ResNet-18 to accommodate one-dimensional vibration signals and reduce model complexity. Paired time–frequency representations are generated as input for contrastive pretraining. A dynamic queue is employed to store negative samples for enhancing contrastive learning. Several data augmentation techniques specifically designed for fault signals are developed, and the optimal combination is determined through ablation experiments. After self-supervised pretraining, the network can be adapted to specific diagnostic tasks by freezing or fine-tuning selected layers.
ResultsThe proposed TFRSSN is evaluated on two benchmark bearing fault datasets: SDUST and PU. Experimental results show that the model achieves competitive diagnostic accuracy even when fine-tuned with only 1% of labeled data. The integration of contrastive learning, time–frequency representations, and tailored augmentation strategies significantly improves feature extraction performance.
ConclusionThe TFRSSN model demonstrates strong potential for real-world fault diagnosis under limited labeled data conditions. Its effectiveness in learning robust representations from unlabeled signals makes it a promising solution for industrial applications where labeled data is scarce.