基于原型对齐与条件对抗的轴承跨域故障诊断

Cross-domain bearing fault diagnosis based on prototype alignment and conditional adversarial learning

  • 摘要: 针对现有无监督域自适应故障诊断方法主要关注全局分布对齐、忽视类条件分布差异从而导致细粒度故障特征混淆的问题,提出一种基于原型对齐与条件对抗的轴承故障诊断方法。该方法构建时频双流特征提取网络,以充分捕获非平稳信号中的多尺度故障模式;设计一种分布形态感知的类原型对齐机制,通过同步匹配特征分布的位置与形态散度,有效抑制了强域偏移下的特征坍塌;同时,将指数滑动平均动态更新的类别原型作为先验条件嵌入域判别器,迫使模型在各自的类别子空间内进行对抗博弈,实现精细化的类别级域对齐。所提方法在CWRU数据集和江南大学数据集的18个不同跨工况任务中平均准确率分别达到了99.71%和99.12%,相较于经典对抗网络及近期同类方法在强域偏移场景下提升显著,在无标注工业应用场景下具有较好的鲁棒性与泛化能力。

     

    Abstract: Rolling bearings are critical load-bearing components in rotating machinery, and their reliable operation is paramount for industrial safety. In practical applications, bearing fault diagnosis often faces challenges such as scarce labeled data and variable working conditions, which lead to significant distribution discrepancies between the source and target data. While Unsupervised Domain Adaptation (UDA) methods based on deep learning have been introduced to address these domain shifts, existing approaches primarily focus on global marginal distribution alignment. They frequently neglect the fine-grained, class-conditional distribution discrepancies, which inevitably leads to the confusion of fault features at the decision boundaries and negative transfer, especially under severe domain shift scenarios. To overcome these critical limitations and ensure highly accurate cross-domain knowledge transfer, a novel unsupervised domain adaptation framework termed Prototype-Aligned Conditional Domain Adversarial Network (PACDAN) is proposed for cross-domain bearing fault diagnosis. The proposed method operates through three core innovations. First, the framework constructs a time-frequency dual-stream feature extraction network. It processes raw one-dimensional temporal vibration signals alongside their corresponding frequency-domain amplitude spectra in parallel branches. This dual-stream architecture deeply and comprehensively captures the multi-scale fault patterns embedded in non-stationary signals, providing a robust and semantically rich feature representation for subsequent adaptation. Second, a distribution-morphology-aware class prototype alignment mechanism is designed to explicitly model and align the intra-class structures. Instead of merely computing the distance between geometric centers, this strategy synchronously matches the first-order position (expectation) and the second-order morphological divergence (central moment) of the feature distributions across domains. By enforcing the statistical isomorphism of the cross-domain distributions, the method compels samples of the same category to tightly cluster around their respective prototypes in the shared feature space. Consequently, it effectively suppresses feature dispersion and structural collapse under severe domain shifts. Third, the framework integrates a conditional domain adversarial learning strategy to achieve precise, class-level domain alignment. The class prototypes, which are dynamically updated using an Exponential Moving Average (EMA) mechanism to guarantee stability during training, are explicitly embedded into the domain discriminator as prior conditional information. This strategic embedding forces the generator and discriminator to perform their adversarial minimax game within specific, fine-grained class subspaces. Therefore, the discriminator focuses on aligning the conditional distributions rather than the global distribution, sharply defining the class boundaries and avoiding cross-class misclassification. Extensive validation experiments are conducted on two widely recognized public platforms: the CWRU dataset and the JNU rotor test rig dataset. The method is rigorously evaluated across 18 distinct cross-condition transfer tasks involving various motor speeds and defect sizes. The experimental results demonstrate that the proposed PACDAN framework achieves remarkable average classification accuracies of 99.71% and 99.12% on the CWRU and JNU datasets, respectively. Furthermore, comprehensive ablation studies, confusion matrices, and t-SNE feature visualizations confirm the necessity of the core components and the superiority of the learned discriminative features. Compared to classical adversarial networks and recent state-of-the-art domain adaptation methods, the proposed approach demonstrates significant performance improvements and exceptional stability under severe domain-shift scenarios, exhibiting superior robustness and immense potential for generalization in unlabeled industrial applications.

     

/

返回文章
返回