Generative adversarial network and transfer-learning-based fault detection for rotating machinery with imbalanced data condition. (11th January 2022)

• Record Type: Journal Article
• Authors: Li, Jun; Liu, Yongbao; Li, Qijie
• Is Part Of: Measurement science & technology. Volume 33:Number 4(2022)
• Publication Date: 2022-01-11
• Subjects: fault diagnosis -- rotating machinery -- imbalanced data -- generative adversarial networks -- transfer learning; Physical measurements -- Periodicals; Scientific apparatus and instruments -- Periodicals; Equipment and Supplies -- Periodicals; Science -- instrumentation -- Periodicals; Technology -- instrumentation -- Periodicals; Mesures physiques -- Périodiques; Physical measurements; Scientific apparatus and instruments; Periodicals; 502.87
• DOI: 10.1088/1361-6501/ac3945
• Languages: English
• ISSNs: 0957-0233

Abstract

Abstract: Intelligent fault diagnosis achieves tremendous success in machine fault diagnosis because of its outstanding data-driven capability. However, the severely imbalanced dataset in practical scenarios of industrial rotating machinery is still a big challenge for the development of intelligent fault diagnosis methods. In this paper, we solve this issue by constructing a novel deep learning model incorporated with a transfer learning (TL) method based on the time-generative adversarial network (Time-GAN) and efficient-net models. Firstly, the proposed model, called Time-GAN-TL, extends the imbalanced fault diagnosis of rolling bearings using time-series GAN. Secondly, balanced vibration signals are converted into two-dimensional images for training and classification by implementing the efficient-net into the transfer learning method. Finally, the proposed method is validated using two types of rolling bearing experimental data. The high-precision diagnosis results of the transfer learning experiments and the comparison with other representative fault diagnosis classification methods reveal the efficiency, reliability, and generalization performance of the presented model.