Data augmentation methods have recently been widely used in fault classification under limited data. However, the traditional data augmentation model often suffers from the problems of gradient explosion and gradient disappearance in training, which limits its application to rolling bearing fault classification to some extent. A new modeling framework is proposed in this paper to address the aforementioned issues. The model first converts the original one-dimensional vibration data of rolling bearings into frequency-domain signals by Fourier transform and then does sample augmentation of the small amount of fault data by using conditional variational auto-encoding generative adversarial network (CVAE-GAN), and then finally trains convolutional neural network with a first wide convolutional layer by using the generative fault data and the original fault data together. The proposed method is validated using a publicly available dataset from the University of Paderborn Laboratory. The experimental results show that the CVAE-GAN method significantly improves fault diagnosis accuracy for rolling bearings under limited data.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Fault Diagnosis Method for Rolling Bearings Based on CVAE-GAN Under Limited Data

  • Wei Hao,
  • Chao Chen,
  • Fengfei Huang,
  • Longqing Fan,
  • Kai Zhang

摘要

Data augmentation methods have recently been widely used in fault classification under limited data. However, the traditional data augmentation model often suffers from the problems of gradient explosion and gradient disappearance in training, which limits its application to rolling bearing fault classification to some extent. A new modeling framework is proposed in this paper to address the aforementioned issues. The model first converts the original one-dimensional vibration data of rolling bearings into frequency-domain signals by Fourier transform and then does sample augmentation of the small amount of fault data by using conditional variational auto-encoding generative adversarial network (CVAE-GAN), and then finally trains convolutional neural network with a first wide convolutional layer by using the generative fault data and the original fault data together. The proposed method is validated using a publicly available dataset from the University of Paderborn Laboratory. The experimental results show that the CVAE-GAN method significantly improves fault diagnosis accuracy for rolling bearings under limited data.