An evaluation method of conditional deep convolutional generative adversarial networks for mechanical fault diagnosis

Abstract

Generative models have been applied in many fields and can be evaluated with many methods. In the evaluation of generative models, the proper evaluation metric varies with the application field. Therefore, the evaluation of generative adversarial networks is inherently challenging. In this study, conditional deep convolutional generative adversarial networks were applied in mechanical fault diagnosis and then evaluated. We proposed three evaluation metrics of conditional deep convolutional generative adversarial networks: Jensen-Shannon divergence, kernel maximum mean discrepancy, and the 1-nearest neighbor classifier which were used to distinguish generated samples from real samples, test mode collapsing and detect overfitting based on the dataset of Electronic Engineering Laboratory of Case Western Reserve University and the planetary gearbox dataset measured in the laboratory. The Jensen-Shannon divergence could not well distinguish generated samples from real samples. However, the two metrics (maximum mean discrepancy and 1-nearest neighbor classifier) well-distinguished generated samples from real samples, thus verifying the applicability of conditional deep convolutional generative adversarial networks in the field of mechanical diagnosis.