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.

What problem does this paper attempt to address?

The problems that this paper attempts to solve mainly focus on two major challenges in rolling bearing fault diagnosis: 1. **Poor diagnostic performance and generalization ability under noisy signals and variable load conditions**: Existing deep - learning - based rolling bearing fault diagnosis methods show poor diagnostic performance and limited generalization ability when faced with noisy signals and different load conditions. 2. **Inadequate utilization of fault information and insufficient feature extraction**: Existing methods do not fully utilize fault information and have insufficient feature extraction, resulting in lower diagnostic accuracy of the model. To address these challenges, the paper proposes an improved Dual - Branch Convolutional Capsule Neural Network (DB - CapsNet) for rolling bearing fault diagnosis. This method improves diagnostic performance through the following steps: - **Data pre - processing**: Convert the collected bearing vibration signals into grayscale images, construct a grayscale image data set, and label it using a dual - label format. - **Multi - scale convolution module**: Introduce a multi - scale convolution module to extract data features and maximize the extraction of feature information. - **Coordinate attention mechanism**: Introduce a coordinate attention mechanism in the multi - scale convolution module to better extract useful channel features and enhance the feature extraction ability. - **Adaptive fusion**: Establish a dual - branch convolutional capsule neural network model based on the adaptive fusion between fault type (damage diameter) features and labels. Through these improvements, the paper aims to improve the accuracy, robustness, and generalization ability of rolling bearing fault diagnosis. Experimental results show that the accuracy of this model on the fault type branch reaches 99.88%, the accuracy on the damage diameter branch reaches 99.72%, and it performs well under noise interference and variable working conditions. Compared with other algorithm models in the references, the model proposed in this study has stronger diagnostic ability. In addition, the generalization ability of the model is verified using a self - built laboratory data set, with an average accuracy of 94.25%.

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

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

The Method of Rolling Bearing Fault Diagnosis Based on Multi-Domain Supervised Learning of Convolution Neural Network

Intelligent fault diagnosis of roller bearing based on dual-flow convolutional neural network

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

An improved multi-scale branching convolutional neural network for rolling bearing fault diagnosis

Intelligent Fault Diagnosis of Bearing Based on Convolutional Neural Network and Bidirectional Long Short-Term Memory

A new dual-channel convolutional neural network and its application in rolling bearing fault diagnosis

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

Fault migration diagnosis method for rolling bearing based on 2D convolutional neural network

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

Rolling bearing fault diagnosis algorithm using overlapping group sparse-deep complex convolutional neural network

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

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

Fault Diagnosis Method for Bearing Based on Attention Mechanism and Multi-Scale Convolutional Neural Network

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

Bearing fault diagnosis method based on the Gramian angular field and an SE-ResNeXt50 transfer learning model

Rolling Bearing Fault Diagnosis Based on Convolutional Neural Network and Multi-sensor Information Fusion

A Multi-scale Temporal Convolutional Capsule Network with Parameter-free Attention Module-Dynamic Routing for Intelligent Diagnosis of Rolling Bearing

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

Fault diagnosis method for rolling bearings based on BICNN under complex operating conditions