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.

Rolling Bearing Fault Diagnosis based on 2D Time-Frequency Images and Data Augmentation Technique

Rolling bearing fault diagnosis method by using feature extraction of convolutional time-frequency image

Rolling Bearing Fault Diagnosis Based on Convolutional Neural Network and Support Vector Machine

Rolling Bearing Fault Diagnosis Based on CWT and Two-Stream Convolutional Neural Network

Rolling Bearing Fault Monitoring for Sparse Time-Frequency Representation and Feature Detection Strategy

A Deep Learning Method for Bearing Fault Diagnosis Based on Time-Frequency Image

An Intelligent Fault Diagnosis Method of Rolling Bearings Based on Short-Time Fourier Transform and Convolutional Neural Network

Fault diagnosis of rolling bearings based on dynamic convolution and dual-channel feature fusion under variable working conditions

Rolling bearing fault diagnosis method using time-frequency information integration and multi-scale TransFusion network

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

Rolling bearing fault diagnosis based on feature fusion with parallel convolutional neural network

Optimal Time Frequency Fusion Symmetric Dot Pattern Bearing Fault Feature Enhancement and Diagnosis

Deep Learning Enabled Fault Diagnosis Using Time-Frequency Image Analysis of Rolling Element Bearings

A New Fault Diagnosis of Rolling Bearing on FFT Image Coding and L-CNN

A Rolling Bearing Fault Diagnosis Method via 2D Feature Map of CSCoh After Denoising and MSCNN Under Different Conditions

A rolling bearing fault diagnosis method based on a convolutional neural network with frequency attention mechanism

A Bearing Fault Diagnosis Method Based on Spectrum Map Information Fusion and Convolutional Neural Network

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

Deep transfer learning rolling bearing fault diagnosis method based on convolutional neural network feature fusion

Rolling Bearing Fault Diagnosis Model Based on External Attention Integrated Convolutional Neural Network under Imbalanced Data Conditions

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