Intelligent Fault Diagnosis for Machinery Based on Enhanced Transfer Convolutional Neural Network

CHEN Zhuyun,ZHONG Qi,HUANG Ruyi,LIAO Yixiao,LI Jipu,LI Weihua
DOI: https://doi.org/10.3901/jme.2021.21.096
2021-01-01
Journal of Mechanical Engineering
Abstract:: Existing intelligent fault diagnosis methods based on deep transfer learning techniques usually align the features of source domain and target domain to reduce distribution discrepancy, ignoring the influences of category decision boundary on target domain samples. Therefore, an enhanced transfer convolutional neural network (ETCNN) is proposed to improve the diagnostic accuracy and generalization ability of mechanical equipment under variable operation conditions. To make full use of the label information and extract high-dimensional features, convolutional neural network with two independent classifiers is separately trained on the source domain data to detect target domain samples away from the decision boundary. To effectively reduce the mismatch of samples at different decision boundaries, a classification loss function and a classifier discriminative loss function are constructed. Furthermore, an adversarial training strategy is introduced to adaptively match samples across domains and improve the classification performance by maximizing the classification differences between the two classifiers while minimizing the differences in feature distributions. The proposed method is fully evaluated on the rolling bearing data set and compared with the other three diagnosis methods including domain adaptive network (DAN), multi-layer DAN (MLDAN) and deep adversarial convolutional neural network (DACNN). The results show that the proposed method not only presents good classification and generalization performance, but also significantly outperforms other methods.
What problem does this paper attempt to address?