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.

Prior Knowledge-Based Self-Supervised Learning for Intelligent Bearing Fault Diagnosis with Few Fault Samples

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

Sparse Sample Train Axle Bearing Fault Diagnosis: A Semi-Supervised Model Based on Prior Knowledge Embedding

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

Few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization

Prior knowledge-embedded meta-transfer learning for few-shot fault diagnosis under variable operating conditions

Prior knowledge-based residuals shrinkage prototype networks for cross-domain fault diagnosis

A Meta-learning Method for Few-shot Bearing Fault Diagnosis under Variable Working Conditions

A novel method based on meta-learning for bearing fault diagnosis with small sample learning under different working conditions

Online Sequential Prediction of Bearings Imbalanced Fault Diagnosis by Extreme Learning Machine

Rolling Bearing Fault Diagnosis with Variable Load and Few Samples Based on Multifeature Fusion Meta-Learning

A Physical Knowledge-Based Extreme Learning Machine Approach to Fault Diagnosis of Rolling Element Bearing from Small Datasets

Prior Knowledge-Augmented Self-Supervised Feature Learning for Few-Shot Intelligent Fault Diagnosis of Machines

Knowledge-informed deep networks for robust fault diagnosis of rolling bearings

Meta-learning for few-shot bearing fault diagnosis under complex working conditions

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

Category knowledge-guided few-shot bearing fault diagnosis

Few-Shot Bearing Fault Diagnosis Based on Model-Agnostic Meta-Learning

Limited Data Rolling Bearing Fault Diagnosis with Few-Shot Learning

A semi-supervised learning method based on pseudo-label iterative purification for intelligent fault diagnosis of rolling bearing

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