Stability Training Method of C‑DCGAN in Mechanical Fault Diagnosis Based on TTUR

被引：0
|
作者：
机构：
来源：
Zhendong Ceshi Yu Zhenduan/Journal of Vibration, Measurement and Diagnosis | 2022年 / 42卷 / 04期
关键词：
D O I：
10.16450/j.cnki.issn.1004-6801.2022.04.016
中图分类号：
学科分类号：
摘要：
To solve the instability of conditional deep convolutional generative adversarial network （C-DCGAN） in training process， we propose a two time-scale update rule （TTUR） for C-DCGAN with stochastic gradient descent in model training for mechanical fault diagnosis. The model stability is improved when the discriminator and generator have learning rates for their own. Firstly， the convergence of TTUR in C-DCGAN model is proved. Secondly， the validity of the method is verified on the bearing data set of Case Western Reserve University （CWUR） and planetary gearbox data set of laboratories. Finally， the Jensen-Shannon divergence （JSD） is introduced to capture the similarity of generated data to real ones. Experiments suggest that TTUR improves the learning for C-DCGAN and outperforms conventional C-DCGAN. © 2022 Journal of Vibration,Measurement & Diagnosis. All rights reserved.
引用
收藏
页码：733 / 740
页数：7
相关论文
共 29 条
  • [1] GOODFELLOW I J, POUGET-ABADIE J, MIRZA M, Et al., Generative adversarial nets, International Conference on Neural Information Processing Systems, pp. 2672-2680, (2014)
  • [2] WANG Kunfeng, ZUO Wangmeng, TAN Ying, Et al., Generative adversarial networks: from generating data to creating intelligence, Acta Automatica Sinica, 44, 5, pp. 769-774, (2018)
  • [3] BROCK A, DONAHUE J, SIMONYAN K., Large scale GAN training for high fidelity natural image synthesis
  • [4] MA J, YU W, LIANG P, Et al., Fusion GAN: a generative adversarial network for infrared and visible image fusion, Information Fusion, 48, pp. 11-26, (2019)
  • [5] LI J, HUO H T, LIU K, Et al., Infrared and visible image fusion using dual discriminators generative adversarial networks with wasserstein distance, Information Sciences, 529, 8, pp. 28-41, (2020)
  • [6] ZHANG Y, GAN Z, CARIN L., Generating text via adversarial training, Conference on Neural Information Processing Systems, (2016)
  • [7] LEE Y O, JO J, HWANG J., Application of deep neural network and generative adversarial network to industrial maintenance: a case study of induction motor fault detection, IEEE International Conference on Big Data, pp. 3248-3253, (2018)
  • [8] WANG Z, WANG J, WANG Y., An intelligent diagnosis scheme based on generative adversarial learning deep neural networks and its application to planetary gearbox fault pattern recognition, Neurocomputing, 310, pp. 213-222, (2018)
  • [9] SUH S, LEE H, JO J, Et al., Generative oversampling method for imbalanced data on bearing fault detection and diagnosis, Applied Sciences - Basel, 9, 4, (2019)
  • [10] LUO J, HUANG J Y, LI H M., A case study of conditional deep convolutional generative adversarial networks in machine fault diagnosis, Journal of Intelligent Manufacturing, 32, 2, pp. 407-425, (2021)
123