Abstract:Applying semi-supervised learning (SSL) methods, such as pseudo-labeling and consistency regularization, to rotating machinery fault diagnosis alleviates the difficulty of obtaining a large amount of labeled data in industrial practice. However, the low accuracy of teacher models initially trained with limited labeled samples often affects the performance of pseudo-labeling methods. Besides, when the existing consistency regularization methods are applied to vibration signals, there is also the problem that the label information may be damaged due to the difficulty in interpreting the one-dimensional (1D) data augmentation (DA) tool. To address these obstacles, a novel two-stage hybrid SSL method based on grouped pseudo-labeling and consistency regularization is developed to generalize mechanical fault diagnosis ability under limited labeled samples. In the first stage, inspired by the label coherence among samples at similar acquisition times, the proposed method fed signal samples with similar acquisition times into the teacher network in groups. Then, the obtained predicted labels are averaged, and a threshold evaluates the validity of the obtained averaged pseudo-labels. In the second stage, mixed labeled and remaining grouped unlabeled samples are fed into the student model for supervised and consistency regularization training. In addition, compared with traditional consistency regularization algorithms using mean square error (MSE) metric to estimate label information consistency between the original and related augmented samples, the trace of the covariance matrix is introduced to realize estimating the label information consistency among multiple unlabeled samples in a group. Experiments verify that the proposed hybrid SSL method steadily improves the fault diagnosis accuracy of two gearboxes with less labeled samples to nearly 100%.

A fault diagnosis framework for rotating machinery of marine equipment: A semi-supervised learning framework based on contractive stacked autoencoder

A hybrid classification autoencoder for semi-supervised fault diagnosis in rotating machinery

Semisupervised Deep Sparse Auto-Encoder With Local and Nonlocal Information for Intelligent Fault Diagnosis of Rotating Machinery

Semi-Supervised Fault Diagnosis Method Based on Dual-Stream Attention Cyclic Auto-Encoder for Rotating Components With Low Label Rate

Modified Stacked Autoencoder Using Adaptive Morlet Wavelet for Intelligent Fault Diagnosis of Rotating Machinery

Fault Diagnosis of Rotary Machinery Components Using a Stacked Denoising Autoencoder-Based Health State Identification

Nonlinear Dynamic System Based on SAE-LDS Model for Fault Diagnosis

Stacked Multilevel-Denoising Autoencoders: A New Representation Learning Approach for Wind Turbine Gearbox Fault Diagnosis

Deep multiple auto-encoder with attention mechanism network: A dynamic domain adaptation method for rotary machine fault diagnosis under different working conditions

Automatic fault diagnosis of rolling bearings under multiple working conditions based on unsupervised stack denoising autoencoder

Fault Diagnosis of Multichannel Bearing-Rotor System via Multistructure Collaborative Discriminative Embedding

Rolling Element Bearing Fault Diagnosis Based on Multi-objective Optimized Deep Auto-encoder

Fault Diagnosis for Rotor Based on Multi-Sensor Information and Progressive Strategies

Hybrid Semi-Supervised Learning for Rotating Machinery Fault Diagnosis Based on Grouped Pseudo Labeling and Consistency Regularization.

A Novel Rotating Machinery Fault Diagnosis System Using Ensemble learning Capsule Autoencoder

A self-supervised contrastive learning framework with the nearest neighbors matching for the fault diagnosis of marine machinery

Machinery fault diagnosis based on a modified hybrid deep sparse autoencoder using a raw vibration time-series signal

A Stacked Auto-Encoder Based Partial Adversarial Domain Adaptation Model for Intelligent Fault Diagnosis of Rotating Machines

An autoencoder with adaptive transfer learning for intelligent fault diagnosis of rotating machinery

Deep representation clustering-based fault diagnosis method with unsupervised data applied to rotating machinery