高级检索

原型对比与对抗熵解耦的开集跨域诊断

Open-Set Cross-Domain Diagnosis via Prototype Contrastive Learning and Adversarial Entropy Decoupling

  • 摘要: 针对变工况下新型故障频现、传统领域适应方法难以识别未知类别的问题,提出一种原型对比学习与对抗熵解耦驱动的开集跨域故障诊断方法(PC-AED)。该方法将多视图增强、原型对比与对抗熵解耦纳入统一诊断流程:引入时域、频域及时频域多视图增强,提升振动信号在工况变化和噪声干扰的鲁棒性;利用动态类别原型约束源域与目标域的特征分布,促进同类特征聚合和异类特征分离,增强已知故障特征的跨域一致;依据源域最大预测熵确定动态不确定性阈值,并结合梯度反转机制进行对抗熵解耦,在缺少未知类标签的条件下优化已知类与未知类之间的决策边界,降低固定经验阈值和未知样本错误迁移的影响。在CWRU和SEU轴承数据集上构建19个开集跨域诊断任务进行实验验证。结果表明:PC-AED在CWRU和SEU数据集上的平均H-score分别达到98.04%和93.83%,较最优对比方法分别提高6.22%和4.48%。该方法在工况变化条件下既能保持已知故障的识别性能,又能有效检测未知故障,为开放工业环境下的旋转机械故障诊断提供了可行方案。

     

    Abstract: New fault types may continually emerge under varying operating conditions, while conventional domain adaptation methods often fail to recognize them as unknown. To address this issue, this study proposes PC-AED, an open-set cross-domain fault diagnosis method driven by prototype-guided contrastive learning and adversarial entropy decoupling. The method uses multi-view augmentation in the time, frequency, and time–frequency domains to improve the stability of vibration features under operating-condition changes and noise. Dynamic class prototypes are used to constrain the feature distributions of the source and target domains. This promotes the aggregation of same-class features and the separation of different-class features. Source-domain predictive entropy is used to determine a dynamic uncertainty threshold. Adversarial learning then refines the decision boundary between known and unknown classes. Nineteen open-set cross-domain diagnosis tasks were constructed using the CWRU and SEU bearing datasets. PC-AED achieved mean H-scores of 98.04% and 93.83% on the CWRU and SEU datasets, respectively. These values exceeded those of the best-performing baseline method by 6.22 and 4.48 percentage points, respectively. The results show that PC-AED maintains its ability to recognize known faults under varying operating conditions while effectively detecting unknown faults. It provides a feasible approach to rotating machinery fault diagnosis in open industrial environments.

     

/

返回文章
返回