Abstract:The realization of accurate fault diagnosis is crucial to ensure the normal operation of machines. At present, an intelligent fault diagnosis method based on deep learning has been widely applied in mechanical areas due to its strong ability of feature extraction and accurate identification. However, it often depends on enough training samples. Generally, the model performance depends on sufficient training samples. However, the fault data are always insufficient in practical engineering as the mechanical equipment often works under normal conditions, resulting in imbalanced data. Deep learning-based models trained directly with the imbalanced data will greatly reduce the diagnosis accuracy. In this paper, a diagnosis method is proposed to address the imbalanced data problem and enhance the diagnosis accuracy. Firstly, signals from multiple sensors are processed by the wavelet transform to enhance data features, which are then squeezed and fused through pooling and splicing operations. Subsequently, improved adversarial networks are constructed to generate new samples for data augmentation. Finally, an improved residual network is constructed by introducing the convolutional block attention module for enhancing the diagnosis performance. The experiments containing two different types of bearing datasets are adopted to validate the effectiveness and superiority of the proposed method in single-class and multi-class data imbalance cases. The results show that the proposed method can generate high-quality synthetic samples and improve the diagnosis accuracy presenting great potential in imbalanced fault diagnosis.

Imbalanced Sample Selection with Deep Reinforcement Learning for Fault Diagnosis

A Novel Imbalanced Data Classification Method Based on Weakly Supervised Learning for Fault Diagnosis

Imbalanced Chemical Process Fault Diagnosis Using Balancing GAN with Active Sample Selection

Imbalanced Fault Diagnosis by Supervised Contrastive Learning

Attention-Aware Meta-Reweighted Optimization for Enhanced Intelligent Fault Diagnosis

Deep reinforcement learning for class imbalance fault diagnosis of equipment in nuclear power plants

Residual Shrinkage ViT with Discriminative Rebalancing Strategy for Small and Imbalanced Fault Diagnosis

Contrastive Feature-Based Learning-Guided Elevated Deep Reinforcement Learning: Developing an Imbalanced Fault Quantitative Diagnosis under Variable Working Conditions

Research and Application of Deep Reinforcement Learning in Rotating Machinery Fault Diagnosis Under Unbalanced Samples Condition

An adaptive fault diagnosis framework under class-imbalanced conditions based on contrastive augmented deep reinforcement learning

Reinforcement Learning-Based Cost-Sensitive Classifier for Imbalanced Fault Classification

Imbalanced fault diagnosis based on semi-supervised ensemble learning

A novel imbalance fault diagnosis method based on data augmentation and hybrid deep learning models

Novel Sparse Auto-encoder Framework with Pseudo-labeled Reinforcement for Cross Domain Fault Diagnosis with Imbalanced Samples

An Adaptable Cost-sensitive Imbalanced Fault Diagnosis Method Considering Sample-significance Information

An Imbalanced Sample Intelligent Fault Diagnosis Method Using Data Enhancement and Improved Broad Learning System

Fault Diagnosis Method for Imbalanced Data Based on Multi-Signal Fusion and Improved Deep Convolution Generative Adversarial Network

Adaptive cost-sensitive learning: Improving the convergence of intelligent diagnosis models under imbalanced data

A Novel Adaptive Undersampling Framework for Class-Imbalance Fault Detection

Dual-Attention Generative Adversarial Networks for Fault Diagnosis under the Class-Imbalanced Conditions