Adversarial flow-based model for unsupervised fault diagnosis of rolling element bearings

Abstract

Nowadays, numerous supervised deep learning models have been applied to bearing fault diagnosis. However, labelling the health states of the bearing vibration data is a time-consuming work and dependent on expert experience. In order to tackle this problem, a novel unsupervised bearing fault diagnosis method named adversarial flow-based model is explored in this paper. Flow-based model is a type of generative models that is proved to be better than other types in many aspects. This paper introduces the flow-based model into the field of machinery fault diagnosis, and designs an appropriate model architecture so as to train the model in unsupervised and adversarial ways. The proposed model contains an autoencoder (AE), a flow-based model, and a classifier. Firstly, the AE maps the vibration data from signal space to latent vector space. Then, the flow-based model aligns the distributions of the latent vectors of different bearing states with specific prior distributions. Finally, the classifier tries to discriminate the aligned latent vectors from the vectors sampled from the prior distributions. With the help of distinguishable prior distributions and the adversarial training mechanism between the classifier and the flow-based model together with the AE, the bearing data with the same health states are clustered into the same areas. The good clustering performance of the adversarial flow-based model is verified by a dataset with 10 health states from a bearing test rig.