A convolutional neural network based on a capsule network with strong generalization for bearing fault diagnosis

By Zhiyu Zhu, Gaoliang Peng, Yuanhang Chen, Huijun Gao. Neurocomputing, 2019

327 citations recorded by the source corpus.

Abstract Bearing fault diagnosis is a significant part of rotating machine health monitoring. In the era of big data, various data-driven methods, which are mainly based on deep learning (DL), have been applied in the field of fault diagnosis. As a common deep learning method, convolutional neural networks play an important role in the field of image classification. To take advantage of existing research results of convolutional neural networks that were originally designed for images, short-time Fourier transform is applied to convert signals to two-dimensional graphs. However, the traditional convolutional neural network with max-pooling layers may disregard the positional relationship between features. Inspired by dynamic routing capsule net, a novel capsule network with an inception block and a regression branch is proposed. First, during the preprocessing stage, the one-dimensional signals are transformed into a time-frequency graph. Second, the graph data are fed into the network, and two convolution layers are applied to extract higher information; following that, an inception block is applied to the output feature maps to improve the nonlinearity of the capsule. Then, after dynamic routing, the lengths of the capsules are used to classify the fault category. There are two other branches: one uses the longest capsule to regress the damage size of the capsule, and the other reconstructs the input graph. To verify the generalization ability of the proposed model, comparisons among the proposed model, conventional methods and state-of-the-art convolutional neural network models are carried out.

Subfields: Control and Systems Engineering, Mechanical Engineering

View Paper: https://www.semanticscholar.org/paper/f89a16fb5aac6f5ebf35e6805f2fb6e600b581ed

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Control and Systems Engineering

In Mechanical Engineering

Citations in the corpus

Machine Learning and Deep Learning Algorithms for Bearing Fault Diagnostics - A Comprehensive Review (2019, background): “Other examples include a mixture of CNN and DBN [163], a deep residual network (DRN) [164], [165], a deep stack network [166], a RNN based auto-encoders [167], sparse filtering [168], and a ew deep le rning architecture – the capsule network [169], proposed by Hinton et al.”
A comprehensive review on convolutional neural network in machine fault diagnosis (2020, background): “ gear pitting fault diagnosis. In [158], Chen et al. proposed a deep inception net with atrous convolution to bridge the gap between artificial and real damage for bearing fault diagnosis. Zhu et al. [159] employed the STFT to convert signals into 2-D graphs as the input and proposed a capsule netwo”
Deep Learning Algorithms for Bearing Fault Diagnosticsx—A Comprehensive Review (2019, background): “For instance, in [167], a capsule network with inception block is proposed aiming at improving the model generalization capability.”