结合加权对抗学习的跨域自适应融合诊断方法
A cross domain adaptive fusion diagnosis method based on weighted adversarial learning
-
摘要: 针对目标域与源域标签空间交叉的跨域诊断,即目标域和源域均存在对方领域没有的样本类型这一典型开放域诊断问题,提出一种结合加权对抗学习的跨域自适应融合诊断方法。利用熵可以表征样本已知类型和未知类型的特性,引入两个结构相同的卷积神经网络进行基于熵的加权对抗性训练,以提取域不变特征增强辨识已知类型的能力,另构建源域和目标域样本输出的二元交叉方案用以隔离未知类型,此外,将两个卷积神经网络的全连接层隐藏特征作为两个标签传递模型的输入,采用投票法则融合三个诊断模型的概率输出。采用变工况的机械传动部件失效实验台数据和自吸式离心泵损伤数据进行分析验证,实验结果表明:所提跨域自适应融合诊断方法能更准确地辨识出目标域数据中已知的故障类型和未知的故障类型。Abstract: For cross domain diagnosis of the label spaces of source domain and target domain are partially overlapped, that is to say, both the target domain and the source domain contain the classes that the other does not have, a cross domain adaptive fusion diagnosis method based on weighted adversarial learning is proposed. As entropy can be used to reflect the characteristics of the shared known classes and unknown classes, two convolutional neural networks with the same structure are introduced to carry out entropy-based weighted adversarial training, which is aim to enhance the ability to identify the shared known classes by extracting the domain-invariant features, as well as the binary cross schemes of the source domain and target domain sample outputs are used to isolate the unknown classes. In addition, the fully connected layer hidden features of these two convolutional neural networks are taken as the input of two label transfer models, and the probability outputs of these three diagnostic models are fused by voting rule. The failure test bench data of mechanical transmission components under variable working conditions and the damage data of self-priming centrifugal pump are used for analysis and verification, the experimental results show that the proposed cross domain adaptive fusion diagnosis method can distinguish the shared known classes and unknown classes in the target domain more accurately.