Predicting remaining useful life of a machine based on embedded attention parallel networks

Xiong Zhang,Yunfei Guo,Hong Shangguan,Ranran Li,Xiaojia Wu,Anhong Wang
DOI: https://doi.org/10.1016/j.ymssp.2023.110221
IF: 8.4
2023-06-01
Mechanical Systems and Signal Processing
Abstract:Remaining useful life (RUL) prediction of mechanical equipment is the core task of the mechanical system of prediction and health management (PHM). To find data that is highly correlated with equipment degradation in enormous data, the mutual use of long short-term memory (LSTM) network and attention mechanism achieves good performance. However, the different combination of the two will greatly affect the efficiency of data feature extraction, and some disadvantages of the traditional sequential LSTM structure and the basic attention mechanism also limit the performance of the network. In this paper, we propose an embedded attention-based parallel network (EAPN) framework for RUL prediction on machine equipment. A new LSTM-based structure, ResLSTMa, is designed for feature extraction of data, and the Query branch of the attention mechanism is improved. Moreover, the ResLSTMa structure is embedded in the Query branch to improve the attention of the relevant features of the attention network, which is combined with the gate recurrent unit (GRU) basic feature extraction network of the parallel branch to achieve multi-level feature extraction and fusion to improve the performance of RUL prediction. The proposed EAPN is evaluated on the aircraft engine dataset and compared with some state-of-the-art prediction methods, and the experimental results demonstrate that our algorithm outperforms these state-of-the-art methods.
engineering, mechanical
What problem does this paper attempt to address?