Abstract:Currently, many fault diagnosis methods for rolling bearings based on deep learning are facing two main challenges. Firstly, the deep learning model exhibits poor diagnostic performance and limited generalization ability in the presence of noise signals and varying loads. Secondly, there is incomplete utilization of fault information and inadequate extraction of fault features, leading to the low diagnostic accuracy of the model. To address these problems, this paper proposes an improved dual-branch convolutional capsule neural network for rolling bearing fault diagnosis. This method converts the collected bearing vibration signals into grayscale images to construct a grayscale image dataset. By fully considering the types of bearing faults and damage diameters, the data are labeled using a dual-label format. A multi-scale convolution module is introduced to extract features from the data and maximize feature information extraction. Additionally, a coordinate attention mechanism is incorporated into this module to better extract useful channel features and enhance feature extraction capability. Based on adaptive fusion between fault type (damage diameter) features and labels, a dual-branch convolutional capsule neural network model for rolling bearing fault diagnosis is established. The model was experimentally validated using both Case Western Reserve University's bearing dataset and self-made datasets. The experimental results demonstrate that the fault type branch of the model achieves an accuracy rate of 99.88%, while the damage diameter branch attains an accuracy rate of 99.72%. Both branches exhibit excellent classification performance and display robustness against noise interference and variable working conditions. In comparison with other algorithm models cited in the reference literature, the diagnostic capability of the model proposed in this study surpasses them. Furthermore, the generalization ability of the model is validated using a self-constructed laboratory dataset, yielding an average accuracy rate of 94.25% for both branches.

Conditional GAN and 2-D CNN for Bearing Fault Diagnosis With Small Samples

Combining Kernel Principal Component Analysis and Spatial Group-Wise Enhance Convolutional Neural Network for Fault Recognition of Rolling Element Bearings

An Efficient Method Based on Conditional Generative Adversarial Networks for Imbalanced Fault Diagnosis of Rolling Bearing

Rolling bearing fault diagnosis in electric motors based on IDIG-GAN under small sample condition

Rolling Bearing Fault Diagnosis Based on Improved GAN and 2-D Representation of Acoustic Emission Signals

Synthesizing Rolling Bearing Fault Samples in New Conditions: A Framework Based on a Modified CGAN

Intelligent fault diagnosis of bearings under small samples: A mechanism-data fusion approach

Rolling bearing fault diagnosis method based on generative adversarial enhanced multi-scale convolutional neural network model

Wasserstein Generative Adversarial Network and Convolutional Neural Network (WG-CNN) for Bearing Fault Diagnosis

Bearing Anomaly Detection Based on Generative Adversarial Network

Bearing Fault Diagnosis Using ACWGAN-GP Enhanced by Principal Component Analysis

Image Enhancement Based on Dual-Branch Generative Adversarial Network Combining Spatial and Frequency Domain Information for Imbalanced Fault Diagnosis of Rolling Bearing

The Fault Diagnosis of Rolling Bearings Is Conducted by Employing a Dual-Branch Convolutional Capsule Neural Network

Fault Diagnosis for Rolling Bearings of a Freight Train under Limited Fault Data: Few-Shot Learning Method

Compound Fault Diagnosis of Rolling Bearings with Few-shot based on DCGAN-RepLKNet

A Lightweight Model for Bearing Fault Diagnosis Based on Gramian Angular Field and Coordinate Attention

Imbalanced Sample Fault Diagnosis of Rolling Bearing Using Deep Condition Multidomain Generative Adversarial Network

Bearing fault diagnosis method based on CNN-LightGBM

Limited fault data augmentation with compressed sensing for bearing fault diagnosis

Application of C-InGAN Model in Interpretable Feature of Bearing Fault Diagnosis

Data-augmented wavelet capsule generative adversarial network for rolling bearing fault diagnosis