1. Wasserstein Generative Adversarial Network and Convolutional Neural Network (WG-CNN) for Bearing Fault Diagnosis
- Author
-
Fei Li, Jiankai Zuo, Kang Yang, Hedan Liu, Zhongzhi Li, and Hang Yin
- Subjects
Article Subject ,Computer science ,Test data generation ,General Mathematics ,Sample (statistics) ,02 engineering and technology ,Machine learning ,computer.software_genre ,Fault (power engineering) ,Convolutional neural network ,Set (abstract data type) ,QA1-939 ,0202 electrical engineering, electronic engineering, information engineering ,Training set ,business.industry ,Deep learning ,020208 electrical & electronic engineering ,General Engineering ,Engineering (General). Civil engineering (General) ,Identification (information) ,Benchmark (computing) ,020201 artificial intelligence & image processing ,Artificial intelligence ,TA1-2040 ,business ,computer ,Mathematics - Abstract
In recent years, intelligent fault diagnosis technology with deep learning algorithms has been widely used in industry, and they have achieved gratifying results. Most of these methods require large amount of training data. However, in actual industrial systems, it is difficult to obtain enough and balanced sample data, which pose challenges in fault identification and classification. In order to solve the problems, this paper proposes a data generation strategy based on Wasserstein generative adversarial network and convolutional neural network (WG-CNN), which uses generator and discriminator to conduct confrontation training, expands a small sample set into a high-quality dataset, and uses one-dimensional convolutional neural network (1D-CNN) to learn sample characteristics and classify different fault types. Experimental results over the standard Case Western Reserve University (CWRU) bearing fault diagnosis benchmark dataset showed that the proposed method has obvious and satisfactory fault diagnosis effect with 100% classification accuracy for few-shot learning. In different noise environments, this method also has excellent performance.
- Published
- 2020
- Full Text
- View/download PDF