Few-shot fault diagnosis (i.e., fault diagnosis with few samples) is a challenging issue in practice because fault samples are scarce and difficult to obtain. Data augmentation based on generative adversarial networks (GANs) has proven to be an effective solution. However, it often encounters problems such as difficult model training and low quality of generated samples. In this article, an improved GAN with filtering mechanism is developed for fault data augmentation, which introduces the self-attention mechanism and instance normalization (IN) into the GAN structure and utilizes a filtering mechanism. The implementation process comprises two parts, sample generation and abnormal sample filtering. The self-attention mechanism and IN adopted in sample generation can make the generative model easy to train and have better quality of the generated samples. For abnormal sample filtering, the isolated forest (IF) algorithm is used for detecting low-quality generated samples. The effectiveness of the proposed fault data augmentation method is verified using two public datasets for fault diagnosis purposes, and the results show that the proposed method can have better performance over the state-of-art GAN-based data augmentation methods.

Improved Generative Adversarial Networks with Filtering Mechanism for Fault Data Augmentation

Simani S.
Writing – Review & Editing
;
2023

Abstract

Few-shot fault diagnosis (i.e., fault diagnosis with few samples) is a challenging issue in practice because fault samples are scarce and difficult to obtain. Data augmentation based on generative adversarial networks (GANs) has proven to be an effective solution. However, it often encounters problems such as difficult model training and low quality of generated samples. In this article, an improved GAN with filtering mechanism is developed for fault data augmentation, which introduces the self-attention mechanism and instance normalization (IN) into the GAN structure and utilizes a filtering mechanism. The implementation process comprises two parts, sample generation and abnormal sample filtering. The self-attention mechanism and IN adopted in sample generation can make the generative model easy to train and have better quality of the generated samples. For abnormal sample filtering, the isolated forest (IF) algorithm is used for detecting low-quality generated samples. The effectiveness of the proposed fault data augmentation method is verified using two public datasets for fault diagnosis purposes, and the results show that the proposed method can have better performance over the state-of-art GAN-based data augmentation methods.
2023
Shao, L.; Lu, N.; Jiang, B.; Simani, S.; Le, Song; Zhengyuan, Liu
File in questo prodotto:
File Dimensione Formato  
Improved_Generative_Adversarial_Networks2023.pdf

solo gestori archivio

Descrizione: Full text editorial version
Tipologia: Full text (versione editoriale)
Licenza: NON PUBBLICO - Accesso privato/ristretto
Dimensione 3.17 MB
Formato Adobe PDF
3.17 MB Adobe PDF   Visualizza/Apri   Richiedi una copia

I documenti in SFERA sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11392/2529427
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 7
  • ???jsp.display-item.citation.isi??? 7
social impact