Bearing fault diagnosis based on partial domain adaptation adversarial network
Huafeng Zhou,Peiyuan Cheng,Siyu Shao,Yuwei Zhao,Xinyu Yang
DOI: https://doi.org/10.1088/1361-6501/ac85d4
IF: 2.398
2022-08-03
Measurement Science and Technology
Abstract:The existing fault diagnosis algorithm based on domain adaptation solves the problem of degradation of model diagnosis performance due to different data distribution under variable working conditions and cross-machine conditions, and its excellent fault diagnosis performance relies on the assumption that the fault category space of source and target domains is the same, however, it is difficult to meet the above assumption in practical application scenarios. For this reason, aiming at the matter of imbalance within the fault category, this paper proposes a novel unsupervised partial domain adaptational fault diagnosis method—Partial Domain Adaptation Adversarial Network (PDAAN). On the one hand, it uses the source domain fault samples to expand the target domain and promotes the effective alignment of the fault feature area of the source domain and the target domain, in order that the model will effectively extract domain invariant features; on the other hand, class-level weights and weighted entropy weights are introduced into the loss function to suppress the uncertainty within the transfer process and avoid negative transfer of the model. Finally, experiments are conducted in the case of variable working conditions and cross-mechanical devices, and it is confirmed that the partial domain adaptation adversarial network model has high recognition accuracy in the case of class space asymmetry.
engineering, multidisciplinary,instruments & instrumentation