Construction of a batch-normalized autoencoder network and its application in mechanical intelligent fault diagnosis

Abstract

Among various fault diagnosis methods, deep learning has shown state-of-the-art performance in processing mechanical big data. This paper investigates a reliable deep learning method known as autoencoder, which is most suitable for automatic feature extraction of fault signals. However, traditional autoencoders have two deficiencies: (1) the multi-layer structure of autoencoder has an internal covariate shift problem, which will cause great difficulty for the network training. (2) The application of autoencoder in the case of rotating speed fluctuation is not mature. To overcome the aforementioned deficiencies, batch normalization strategy is employed in every layer of the autoencoder network to obtain a steady distribution of activation values during training. It can regularize the network without parameter adjustment, and deal with the speed fluctuation problem perfectly. So, a new network named batch-normalized autoencoder is first proposed for intelligent fault diagnosis. The raw vibration signals are directly fed into the network and the extracted features are employed to train a softmax classifier for health state identification. A bearing and a gearbox data set are finally used to confirm the effectiveness of the proposed method. The results manifest that the proposed method can extract salient features from the raw signals and handle the fault diagnosis problem under the speed fluctuation problem.