An intelligent fault diagnosis method for rolling bearings based on feature transfer with improved DenseNet and joint distribution adaptation
Abstract Mechanical intelligent fault diagnosis is an important method to accurately identify the health status of mechanical equipment. Traditional fault diagnosis methods perform poorly in the diagnosis of rolling bearings under complex conditions. In this paper, a feature transfer learning model based on improved DenseNet and joint distribution adaptation (FT-IDJ) is proposed. With this model, we apply it to implement rolling bearing fault diagnosis. A lightweight DenseNet model is firstly proposed to extract the transferable features of the raw vibration signal. Furthermore, the parameters in the DenseNet are constrained by the domain adaptive regularization term and pseudo label learning. The marginal distribution discrepancy and the conditional distribution discrepancy of the learned transferable features are reduced by this way. The proposed method is validated by the diagnosis experiments with CWRU and Jiangnan University rolling bearing datasets. The experimental results showed that the proposed FT-IDJ has higher classification accuracy than DAN and other eight methods, which demonstrated its effectively learning transferable features from auxiliary data.