A Multiscale Pooling Attention-Based Graph Attention Network for Remaining Useful Life Prediction

Owing to the intricate spatial and temporal relationships inherent in data collected from multiple sensors, achieving precise predictions of Remaining Useful Life (RUL) becomes a challenging task. Recently, deep learning-based approaches have made substantial advancements in RUL prediction. However, the traditional neural network-based methods have encountered some trouble in extracting spatial features. Graph Neural Network (GNN) has demonstrated the ability to effectively capture the spatial dependencies between multi-sensor data, but current GNN-based approaches cannot achieve much in terms of the spatial-temporal dependencies at various scales. Motivated by this, a multi-scale pooling attention-based graph attention network (MSPA-GAT) is proposed. Firstly, a multi-GATv2 network is designed for the spatial dependencies modeling, and a bi-directional long short-term memory (BiLSTM) network is used for modeling the temporal dependencies. Secondly, a multi-scale pooling attention mechanism (MSPA) is constructed to highlight the local details of different scales and capture multi-level information. Finally, the effectiveness of the proposed MSPA-GAT to consider spatial and temporal dependencies is validated using two datasets. Moreover, the experimental results have shown that MSPA-GAT outperforms current state-of-the-art methods in RUL prediction.