Bearing fault diagnosis based on transfer learning with dual-flow manifold ResNet and improved CapsNet

Abstract

In intelligent fault diagnosis for rolling bearings, the variable operating conditions of rotating machinery lead to inconsistent distributions between training and testing data, inevitably decreasing fault diagnosis accuracy. To overcome this issue, we propose a bearing fault diagnosis method based on transfer learning with a dual-flow manifold residual network and improved capsule network (DMRCN). Firstly, we introduce a dual-flow manifold residual network (DMResNet) structure consisting of continuous wide-kernel convolution, continuous narrow-kernel convolution, and manifold learning spatial information description blocks. This architecture is utilized to extract low-level features from the data. Then, we present a novel architecture of the capsule network with self-attention output (SAO-CN). We incorporate a self-attention mechanism into the dynamic routing algorithm (SA-DR) to transfer richer high-level features from the low-level capsule layer to the higher-level capsule layer. During training, we propose a loss function known as global-local distribution discrepancy loss (G-L-DDL) to enhance the model's generalization ability and accelerate model convergence. Finally, we validate the performance of DMRCN in cross-condition transfer fault diagnosis on both the CWRU public dataset and a laboratory-built bearing experimental rig dataset. Experimental results demonstrate that DMRCN outperforms other commonly used deep transfer learning methods regarding fault diagnosis across different operating conditions.