As a useful data augmentation technique, generative adversarial networks have been successfully applied in fault diagnosis field. But traditional generative adversarial networks can only generate one category fault signals in one time, which is time-consuming and costly. To overcome this weakness, we develop a novel fault diagnosis method which combines conditional generative adversarial networks and stacked autoencoders, and both of them are built by stacking one-dimensional full connection layers. First, conditional generative adversarial networks is used to generate artificial samples based on the frequency samples, and category labels are adopted as the conditional information to simultaneously generate different category signals. Meanwhile, spectrum normalization is added to the discriminator of conditional generative adversarial networks to enhance the model training. Then, the augmented training samples are transferred to stacked autoencoders for feature extraction and fault classification. Finally, two datasets of bearing and gearbox are employed to investigate the effectiveness of the proposed conditional generative adversarial network–stacked autoencoder method.
The problem of insufficient datasets has long been a hot topic in the field of prognosis and health management of rotary machines. Generative adversarial network (GAN) and other data augmentation algorithms can solve the problem of insufficient samples. However, the premise of the above method is the signal collected at a constant speed rather than at large speed fluctuation. To deal with data augmentation under large speed fluctuation, this paper proposes an effective deep learning method, namely, domain adaptive efficient sub-pixel network (DAESPN). The core idea of DAESPN is to enhance the resolution of the original sample for data augmentation. The DAESPN framework is implemented as follows: after the data passes through the fully connected neural network, the multi-feature maps of the four channels are outputted. A group of high resolution (HR) features is obtained through the sub-pixel fully connected layer. In addition, maximum mean discrepancy (MMD) and mean square error (MSE) are used to construct the loss function of the model. Experimental results of gearbox and bearing datasets show that the DAESPN model has strong feasibility to carry out data augmentation for fault diagnosis of rotating machines under speed fluctuation condition. In addition, the feature learning process of DAESPN is visually displayed and analyzed.
Data augmentation has become a hot topic in the field of mechanical intelligent fault diagnosis. It can expand the limited training dataset by generating simulated samples, but there is still no effective method augmenting the resolution of low resolution sample. In this paper, a simple algorithm, namely, efficient subpixel convolutional neural network (ESPCN), is proposed to solve this deficiency. The ESPCN model performs the arrange operation on the raw low resolution data through the subpixel layer and outputs the result of four-channel multifeature maps. Then, the sample resolution is increased to four times compared with the raw low resolution sample. Finally, the generated high resolution dataset is employed to train the stacked autoencoders (SAE) for fault classification, and the raw high resolution dataset is used for testing. Two fault diagnosis cases with different sample dimensions and rotating speeds are set up to simulate the low resolution situation, and the experimental results verify the feasibility of the proposed algorithm.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.