Progressive Attention-Guided Two-Stage Transfer Learning for Few-Shot Cross-Condition Bearing Fault Diagnosis

Abstract

Cross-condition bearing fault diagnosis suffers from severe performance degradation due to domain shift across different operating conditions, especially when only a few labeled target-domain samples are available. To address this challenge, this paper proposes a progressive attention-guided two-stage transfer learning framework for few-shot bearing fault diagnosis across different fixed operating points. First, the raw time-domain vibration signals are fused with frequency-domain representations extracted by short-time Fourier transform (STFT) to enhance fault feature representation. Then, a progressive attention-guided feature learning strategy is developed by integrating dual efficient channel attention (ECA) modules into a deep one-dimensional convolutional neural network (1D-CNN), enabling the network to adaptively emphasize fault-sensitive features while suppressing redundant information. Subsequently, a two-stage transfer learning strategy is designed, consisting of transferable feature learning from the source domain and few-shot adaptation to the target domain. During target-domain adaptation, key feature extraction layers are frozen, and a sample-balanced optimization mechanism is introduced to alleviate the dominance of source-domain samples during joint training. Experimental results on the Case Western Reserve University (CWRU) bearing dataset demonstrate that the proposed method achieves an average accuracy of 99.96% across three cross-condition transfer tasks. Furthermore, experiments conducted on a self-built shaft system dataset show that the proposed method achieves an average accuracy of 87.11% under three representative transfer scenarios. The results verify that the proposed framework effectively mitigates domain shift and enables accurate bearing fault diagnosis with limited labeled target-domain samples.