Data augmentation and intelligent fault diagnosis of planetary gearbox using ILoFGAN under extremely limited samples

Mingzhi Chen, Haidong Shao, Haoxuan Dou, Wei Li, Bin Liu

Research output: Contribution to journalArticlepeer-review

5 Downloads (Pure)

Abstract

Although the existing generative adversarial networks (GAN) have the potential for data augmentation and intelligent fault diagnosis of planetary gearbox, it remains difficult to deal with extremely limited training samples and effectively fuse the representative and diverse information. To tackle the abovementioned challenges, an improved local fusion generative adversarial network is proposed. Time-domain waveforms are first transformed into the time–frequency diagrams to highlight the fault characteristics. Subsequently, a local fusion module is used to fully utilize extremely limited samples and fuse the local features. Finally, a new generator embedded with multi-head attention modules is constructed to effectively improve the accuracy and flexibility of the feature fusion process. The proposed method is applied to the analysis of planetary gearbox vibration signals. The results show that the proposed method can generate a large number of samples with higher similarity and better diversity compared with the existing mainstream GANs using six training samples in each type. The generated samples are used to augment the limited dataset, prominently improving the accuracy of the fault diagnosis task.
Original languageEnglish
Pages (from-to)1-9
Number of pages9
JournalIEEE Transactions on Reliability
Early online date27 Oct 2022
DOIs
Publication statusPublished - 27 Oct 2022

Keywords

  • intelligent fault diagnosis
  • extremely few samples
  • multi-head attention mechanism
  • ILoFGAN
  • planetary gearbox

Fingerprint

Dive into the research topics of 'Data augmentation and intelligent fault diagnosis of planetary gearbox using ILoFGAN under extremely limited samples'. Together they form a unique fingerprint.

Cite this