Machine fault diagnosis with small sample based on variational information constrained generative adversarial network

Shaowei Liu, Hongkai Jiang, Zhenghong Wu, Yunpeng Liu, Ke Zhu

Research output: Contribution to journalArticlepeer-review

52 Scopus citations

Abstract

In actual engineering scenarios, limited fault data leads to insufficient model training and over-fitting, which negatively affects the diagnostic performance of intelligent diagnostic models. To solve the problem, this paper proposes a variational information constrained generative adversarial network (VICGAN) for effective machine fault diagnosis. Firstly, by incorporating the encoder into the discriminator to map the deep features, an improved generative adversarial network with stronger data synthesis capability is established. Secondly, to promote the stable training of the model and guarantee better convergence, a variational information constraint technique is utilized, which constrains the input signals and deep features of the discriminator using the information bottleneck method. In addition, a representation matching module is added to impose restrictions on the generator, avoiding the mode collapse problem and boosting the sample diversity. Two rolling bearing datasets are utilized to verify the effectiveness and stability of the presented network, which demonstrates that the presented network has an admirable ability in processing fault diagnosis with few samples, and performs better than state-of-the-art approaches.

Original languageEnglish
Article number101762
JournalAdvanced Engineering Informatics
Volume54
DOIs
StatePublished - Oct 2022

Keywords

  • Fault diagnosis
  • Generative adversarial network
  • Rolling bearing
  • Small sample
  • Variational information constraint

Fingerprint

Dive into the research topics of 'Machine fault diagnosis with small sample based on variational information constrained generative adversarial network'. Together they form a unique fingerprint.

Cite this