Frontiers in Neurorobotics (Nov 2022)

Bearing fault diagnosis based on particle swarm optimization fusion convolutional neural network

  • Xian Liu,
  • Ruiqi Wu,
  • Rugang Wang,
  • Feng Zhou,
  • Zhaofeng Chen,
  • Naihong Guo

DOI
https://doi.org/10.3389/fnbot.2022.1044965
Journal volume & issue
Vol. 16

Abstract

Read online

Bearings are the most basic and important mechanical parts. The stable and safe operation of the equipment requires bearing fault diagnosis in advance. So, bearing fault diagnosis is an important technology. However, the feature extraction quality of the traditional convolutional neural network bearing fault diagnosis is not high and the recognition accuracy will decline under different working conditions. In response to these questions, a bearing fault model based on particle swarm optimization (PSO) fusion convolution neural network is proposed in this paper. The model first adaptively adjusts the hyperparameters of the model through PSO, then introduces residual connections to prevent the gradient from disappearing, uses global average pooling to replace the fully connected layer to reduce the training parameters of the model, and finally adds a dropout layer to prevent network overfitting. The experimental results show that the model is under four conditions, two of which can achieve 100% recognition, and the other two can also achieve more than 98% accuracy. And compared with the traditional diagnosis method, the model has higher accuracy under variable working conditions. This research has important research significance and economic value in the field of the intelligent machinery industry.

Keywords