VGAN: generalizing MSE GAN and WGAN-GP for robot fault diagnosis

Pu, Ziqiang; Cabrera, Diego; Li, Chuan; Valente de Oliveira, JOSÉ

http://hdl.handle.net/10400.1/18600

Use this identifier to reference this record.

Name:	Description:	Size:	Format:
VGAN_Generalizing_MSE_GAN_and_WGAN-GP_for_Robot_Fault_Diagnosis.pdf		2.74 MB	Adobe PDF	Download

Send Feedback

Authors

Pu, Ziqiang

Cabrera, Diego

Li, Chuan

Valente de Oliveira, JOSÉ

Abstract(s)

Generative adversarial networks (GANs) have shown their potential for data generation. However, this type of generative model often suffers from oscillating training processes and mode collapse, among other issues. To mitigate these, this work proposes a generalization of both mean square error (mse) GAN and Wasserstein GAN (WGAN) with gradient penalty, referred to as VGAN. Within the framework of conditional WGAN with gradient penalty, VGAN resorts to the Vapnik V-matrix-based criterion that generalizes mse. Also, a novel early stopping-like strategy is proposed that keeps track during training of the most suitable model. A comprehensive set of experiments on a fault-diagnosis task for an industrial robot where the generative model is used as a data augmentation tool for dealing with imbalance datasets is presented. The statistical analysis of the results shows that the proposed model outperforms nine other models, including vanilla GAN, conditional WGAN with and without conventional regularization, and synthetic minority oversampling technique, a classic data augmentation technique.

Keywords

Generative adversarial networks Generators Fault diagnosisIntelligent systems RobotsData models Data generation Training conditional Wasserstein generative adversarial networkregularizationV-matrixfault diagnosisrobotics