TR2021-002

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


    •  Zhang, S., Ye, F., Wang, B., Habetler, T.G., "Semi-Supervised Bearing Fault Diagnosis and Classification using Variational Autoencoder-Based Deep Generative Models", IEEE Sensors Journal, DOI: 10.1109/JSEN.2020.3040696, Vol. 21, No. 5, pp. 6476-6486, January 2021.
      BibTeX TR2021-002 PDF
      • @article{Zhang2021jan,
      • author = {Zhang, Shen and Ye, Fei and Wang, Bingnan and Habetler, Thomas G},
      • title = {Semi-Supervised Bearing Fault Diagnosis and Classification using Variational Autoencoder-Based Deep Generative Models},
      • journal = {IEEE Sensors Journal},
      • year = 2021,
      • volume = 21,
      • number = 5,
      • pages = {6476--6486},
      • month = jan,
      • doi = {10.1109/JSEN.2020.3040696},
      • url = {https://www.merl.com/publications/TR2021-002}
      • }
  • MERL Contact:
  • Research Areas:

    Electric Systems, Machine Learning

Many industries are evaluating the use of the Internet of Things (IoT) technology with big data 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. However, in practical applications, obtaining accurate labels based on real-time bearing conditions can be more challenging than simply collecting large amounts of unlabeled data using multiple sensors. 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 utilize 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 show 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 issues when identifying naturally-evolved bearing faults.

 

  • Related Publication

  •  Zhang, S., Ye, F., Wang, B., Habetler, T.G., "Semi-Supervised Learning of Bearing Anomaly Detection via Deep Variational Autoencoders", arXiv, November 2019.
    BibTeX arXiv
    • @article{Zhang2019nov,
    • author = {Zhang, Shen and Ye, Fei and Wang, Bingnan and Habetler, Thomas G},
    • title = {Semi-Supervised Learning of Bearing Anomaly Detection via Deep Variational Autoencoders},
    • journal = {arXiv},
    • year = 2019,
    • month = nov,
    • url = {https://arxiv.org/abs/1912.01096}
    • }