Deep Feature Generating Network: A New Method for Intelligent Fault Detection of Mechanical Systems Under Class Imbalance

Abstract

Class imbalance issue has been a major problem in mechanical fault detection, which exists when the number of instances presenting in a class is significantly fewer than that in another class. This article focuses on the problem of zero-shot fault detection of rolling bearing, which is the extreme case of class imbalance. Aiming at this problem, a two-stage zero-shot fault recognition method is proposed. First, inspired by the conditional generative adversarial network, a novel feature generating network which is composed of a feature extractor, a discriminator, and a generator is designed to capture the potential distribution of normal samples. Then, the generator will generate abundant pseudofault features by adding an additional sequence to the condition. Second, an improved deep neural network is trained with these synthetic pseudofault features as the classifier. Specially, a condition index is designed to represent different fault classes so that it can recognize the unseen fault samples. Finally, the effectiveness of the proposed method is verified by three datasets and a comparison method is also given to show the superiority. Results show that the feature generation network can effectively detect the typical faults even though the fault data are unavailable during training, which is practical for industrial application.