Deep Domain Adaptation Model for Bearing Fault Diagnosis with Riemann Metric Correlation Alignment
Jing An, Ping Ai
Abstract
In many real-world fault diagnosis applications, due to the frequent changes in working conditions, the distribution of labeled training data (source domain) is different from the distribution of the unlabeled test data (target domain), which leads to performance degradation. In order to solve this problem, an end-to-end unsupervised domain adaptation bear fault diagnosis model that combines Riemann metric correlation alignment and one-dimensional convolutional neural network (RMCA-1DCNN) is proposed in this study. Second-order statistic alignment of the specific activation layer in source and target domains is considered to be a regularization item and embedded in the deep convolutional neural network architecture to compensate for domain shift. Experimental results on the Case Western Reserve University motor bearing database demonstrate that the proposed method has strong fault-discriminative and domain-invariant capacity. Therefore, the proposed method can achieve higher diagnosis accuracy than that of other existing experimental methods.