Semi-Supervised Bearing Fault Diagnosis and Classification Using Variational Autoencoder-Based Deep Generative Models

被引:88
作者
Zhang, Shen [1 ]
Ye, Fei [2 ]
Wang, Bingnan [3 ]
Habetler, Thomas G. [1 ]
机构
[1] Georgia Inst Technol, Dept Elect & Comp Engn, Atlanta, GA 30332 USA
[2] Univ Calif Berkeley, Calif PATH, Berkeley, CA 94720 USA
[3] Mitsubishi Elect Res Labs, Cambridge, MA 02139 USA
关键词
Data models; Fault diagnosis; Sensors; Semisupervised learning; Decoding; Supervised learning; Training; Bearing fault; generative model; semi-supervised learning; variational autoencoders; VIBRATION; NETWORK;
D O I
10.1109/JSEN.2020.3040696
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Many industries are evaluating the use of the Internet of Things (IoT) technology to perform remote monitoring and predictive maintenance on their mission-critical assets and equipment, for which mechanical bearings are their indispensable components. Although many data-driven methods have been applied to bearing fault diagnosis, most of them belong to the supervised learning paradigm that requires a large amount of labeled training data to be collected in advance. In practical applications, however, obtaining labeled data that accurately reflect real-time bearing conditions can be more challenging than collecting large amounts of unlabeled data. In this paper, we thus propose a semi-supervised learning scheme for bearing fault diagnosis using variational autoencoder (VAE)-based deep generative models, which can effectively leverage a dataset when only a small subset of data have labels. Finally, a series of experiments were conducted using the University of Cincinnati Intelligent Maintenance System (IMS) Center dataset and the Case Western Reserve University (CWRU) bearing dataset. The experimental results demonstrate that the proposed semi-supervised learning schemes outperformed some mainstream supervised and semi-supervised benchmarks with the same percentage of labeled data samples. Additionally, the proposed methods can mitigate the label inaccuracy issue when identifying naturally-evolved bearing defects.
引用
收藏
页码:6476 / 6486
页数:11
相关论文
共 50 条
  • [31] A Semi-supervised Learning Based on Variational Autoencoder for Visual-Based Robot Localization
    Liang, Kaiyun
    He, Fazhi
    Zhu, Yuanyuan
    Gao, Xiaoxin
    COMPUTER SUPPORTED COOPERATIVE WORK AND SOCIAL COMPUTING, CHINESECSCW 2021, PT I, 2022, 1491 : 615 - 627
  • [32] A fault diagnosis framework for rotating machinery of marine equipment: A semi-supervised learning framework based on contractive stacked autoencoder
    Pan, Penghao
    Zhao, Dong
    Li, Yueyang
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART M-JOURNAL OF ENGINEERING FOR THE MARITIME ENVIRONMENT, 2023, 237 (03) : 625 - 636
  • [33] Semi-Supervised Recurrent Variational Autoencoder Approach for Visual Diagnosis of Atrial Fibrillation
    Costa, Nahuel
    Sanchez, Luciano
    Couso, Ines
    IEEE ACCESS, 2021, 9 : 40227 - 40239
  • [34] Semi-Supervised Encrypted Traffic Classification With Deep Convolutional Generative Adversarial Networks
    Iliyasu, Auwal Sani
    Deng, Huifang
    IEEE ACCESS, 2020, 8 : 118 - 126
  • [35] Semi-supervised COVID-19 CT image segmentation using deep generative models
    Judah Zammit
    Daryl L. X. Fung
    Qian Liu
    Carson Kai-Sang Leung
    Pingzhao Hu
    BMC Bioinformatics, 23
  • [36] Temporal Variational Auto-Encoders for Semi-Supervised Remaining Useful Life and Fault Diagnosis
    San Martin, Gabriel
    Droguett, Enrique Lopez
    IEEE ACCESS, 2022, 10 : 55112 - 55125
  • [37] Semi-supervised COVID-19 CT image segmentation using deep generative models
    Zammit, Judah
    Fung, Daryl L. X.
    Liu, Qian
    Leung, Carson Kai-Sang
    Hu, Pingzhao
    BMC BIOINFORMATICS, 2022, 23 (SUPPL 7)
  • [38] Improving Colonoscopy Lesion Classification Using Semi-Supervised Deep Learning
    Golhar, Mayank
    Bobrow, Taylor L.
    Khoshknab, Mirmilad Pourmousavi
    Jit, Simran
    Ngamruengphong, Saowanee
    Durr, Nicholas J.
    IEEE ACCESS, 2021, 9 : 631 - 640
  • [39] Multi-Scale Recursive Semi-Supervised Deep Learning Fault Diagnosis Method with Attention Gate
    Tang, Shanjie
    Wang, Chaoge
    Zhou, Funa
    Hu, Xiong
    Wang, Tianzhen
    MACHINES, 2023, 11 (02)
  • [40] Multimodal deep generative adversarial models for scalable doubly semi-supervised learning
    Du, Changde
    Du, Changying
    He, Huiguang
    INFORMATION FUSION, 2021, 68 : 118 - 130