A weak prior embedding-based method for transfer fault diagnosis of rolling bearing

作者：Sun, Haoran; Wang, Yi^*; Ruan, Hulin; Qin, Yi; Tang, Baoping; Chen, Baojia

来源：MEASUREMENT, 2022, 199: 111519.

DOI：10.1016/j.measurement.2022.111519

• Fault diagnosis
• Cross domain
• Weak prior embedding
• Template samples

Transfer learning extends the application scope of deep intelligent fault diagnosis models. However, when operation conditions are significantly varied, existing transfer learning methods may encounter a bottleneck. The unsupervised methods will mismatch features, while supervised methods are constrained by insufficient target labeled instances. A weak prior embedding-based domain adaption network (WPEDAN) is proposed to resolve this problem. Specifically, the cluster ability of source data pre-trained network to target data is emphasized and enhanced directly and indirectly so as to obtain discriminative feature clusters. The target prior template samples are embedded and automatically delivered to the most relevant feature clusters, so the class information of target instances can be estimated based on similarity and supervised training on target is following. Moreover, an improved conditional maximum mean discrepancy (ICMMD) is developed to further align conditional distri-bution. Experiments on two challenging bearing datasets show that proposed method achieves more than 99% diagnosis accuracy in all tasks, and is significantly ahead of the comparison methods in complex transfer tasks.

• 单位
  重庆大学; 西安交通大学; y