Abstract:To obtain excellent classification performance for fault diagnosis, most intelligent fault diagnosis methods based on deep learning require massive labeled samples for training. However, collecting sufficient labeled fault samples is very difficult in practice due to the time-consuming and laborious work, which means the actual available dataset is the unbalanced dataset, i.e., normal data is the vast majority, while the fault samples are very small. To address this problem, a modified active learning intelligent fault diagnosis method is proposed for rolling bearings with unbalanced samples. The proposed method can adeptly employ a limited number of labeled samples to intelligently label the unlabeled samples. Therefore, the proposed method can improve classification performance while simultaneously minimizing the requisite amount of labeled samples during training. First, time and time–frequency features of vibration signals are extracted to obtain their distribution in the feature space. Second, to solve the problem of sample class unbalance, a Gaussian mixture model is constructed to obtain the distribution representation of the samples. The random undersampling method was used in Gaussian sub-model, which can extract some samples from majority classes. These extracted samples have similar distribution to the original sample set, and hence can represent the original dataset and be used to establish balanced labeled sample set. Third, an initial active learning classifier based on density peak clustering is established, utilizing the representative examples to intelligently label the unlabeled samples. To optimize the utilization of unlabeled samples, batch process method is adopted to update the initial classifier. The effectiveness of the proposed method is verified by two rolling bearings fault simulation experiments. The results show that our method can effectively improve fault diagnosis accuracy with unbalanced samples, and the updated classifier needs fewer training data to achieve comparable diagnostic performance.

A limited labeled data bearing fault diagnosis method based on self-supervised learning

Self-supervised Knowledge Mining from Unlabeled Data for Bearing Fault Diagnosis under Limited Annotations

A Wavelet Transform and self-supervised learning-based framework for bearing fault diagnosis with limited labeled data

A Semi-Supervised Intelligent Fault Diagnosis Method for Bearings Under Low Labeled Rates

A multi-stage semi-supervised learning approach for intelligent fault diagnosis of rolling bearing using data augmentation and metric learning

A Universal Feature Extractor Based on Self-Supervised Pre-Training for Fault Diagnosis of Rotating Machinery under Limited Data

A Fault Diagnosis Method for Rolling Bearing Based on Deep Adversarial Transfer Learning with Transferability Measurement

A New Bearing Fault Diagnosis Method Based on Deep Transfer Network and Supervised Joint Matching

Unsupervised cross-domain rolling bearing fault diagnosis based on time-frequency information fusion

A self-attention based contrastive learning method for bearing fault diagnosis

A contrastive learning-based fault diagnosis method for rotating machinery with limited and imbalanced labels

Semi-Supervised Transfer Learning Method for Bearing Fault Diagnosis with Imbalanced Data

A Hybrid Supervised Approach for Fault Diagnosis Based on Temporal and Frequency Domains

A novel approach for bearings multiclass fault diagnosis fusing multiscale deep convolution and hybrid attention networks

Fault Diagnosis of Rotating Machinery Based on Improved Self-Supervised Learning Method and Very Few Labeled Samples

A modified active learning intelligent fault diagnosis method for rolling bearings with unbalanced samples

An Integrated Multitasking Intelligent Bearing Fault Diagnosis Scheme Based on Representation Learning Under Imbalanced Sample Condition

Self-attention Metric Learning Based on Multi-Scale Feature Fusion for Few-shot Fault Diagnosis

Rolling bearing intelligent fault diagnosis towards variable speed and imbalanced samples using multiscale dynamic supervised contrast learning

A novel simulation-assisted transfer method for bearing unknown fault diagnosis

A Novel Study on a Generalized Model Based on Self-Supervised Learning and Sparse Filtering for Intelligent Bearing Fault Diagnosis