TR2019-153

Semi-Supervised Learning of Bearing Anomaly Detection via Deep Variational Autoencoders


    •  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}
      • }
  • MERL Contact:
  • Research Areas:

    Electric Systems, Machine Learning

Most of the data-driven approaches applied to bearing fault diagnosis up to date are established in the supervised learning paradigm, which usually requires a large set of labeled data collected a priori. In practical applications, however, obtaining accurate labels based on real-time bearing conditions can be far more challenging than simply collecting a huge amount of unlabeled data using various sensors. In this paper, we thus propose a semi-supervised learning approach for bearing anomaly detection using variational autoencoder (VAE) based deep generative models, which allows for effective utilization of dataset when only a small subset of data have labels. Finally, a series of experiments is performed using both the Case Western Reserve University (CWRU) bearing dataset and the University of Cincinnati’s Center for Intelligent Maintenance Systems (IMS) dataset. The experimental results demonstrate that the proposed semi-supervised learning scheme greatly outperforms two mainstream semi-supervised learning approaches and a baseline supervised convolutional neural network approach, with the overall accuracy improvement ranging between 3% to 30% using different number of labeled samples.