A life prediction method based on MDFF and DITCN-ABiGRU mixed network model

Heliyon. 2024 Jan 9;10(2):e24299. doi: 10.1016/j.heliyon.2024.e24299. eCollection 2024 Jan 30.

Abstract

A single network model exhibits limitations in the life prediction of rotating machinery for the various fault types and uncertain fault occurrence. Therefore, a network prediction model combining multi-domain feature fusion (MDFF) and distributed TCN-Attention-BiGRU (DITCN-ABiGRU) is proposed to enable a more accurate life prediction of rotating machinery. Firstly, the features of vibration signals collected from multiple sensors are extracted in the time, frequency, and time-frequency domains. Subsequently, dimensionality reduction optimization is conducted on these multi-domain features to eliminate useless information features. The temporal convolutional network (TCN) model is constructed to capture the critical information reflecting the fault characteristics of rotating machinery through the attention mechanism, and the dependencies of the whole training process are captured by the BiGRU network. Finally, precise prediction of the lifespan of rotating machinery is achieved by constructing a health indicator curve (HI). The proposed methods are verified through the life prediction of rolling bearings from the IEEE PHM Challenge 2012 dataset and ball screw pairs from a designed experiment. The experimental results show that the proposed MDFF and DITCN-ABiGRU model achieves a better score and lower error than the convolutional neural network (CNN) and GRU models.

Keywords: Life prediction; Multi-domain feature fusion; Rotating machinery; Temporal convolutional network.