MSTAN: multi-scale spatiotemporal attention network with adaptive relationship mining for remaining useful life prediction in complex systems

In the era of smart manufacturing and advanced industrial systems, the high degree of integration and intelligence of equipment demands higher reliability and safety from systems. Existing methods often rely on historical data for Remaining Useful Life (RUL) prediction to achieve Prognostic and Health Management (PHM). However, the internal units of complex equipment exhibit significant spatial correlation and temporal diversity, making PHM for complex equipment a multidimensional challenge involving both temporal and spatial information, thereby severely limits the effectiveness of RUL prediction for complex systems. Addressing these challenges, this study introduces a multi-scale spatiotemporal attention network with adaptive relationship mining, specifically designed for the remaining useful life (RUL) prediction of such equipment. The core of the proposed method lies in the multi-scale feature perception module, which adeptly extracts varied scale features from multidimensional sensor data. Following this, an innovative adaptive relationship mining module is integrated to uncover multi-order coupling relationships between diverse sensors, enhancing the model’s predictive accuracy. Furthermore, a spatiotemporal attention module is employed to discern and emphasize crucial spatiotemporal correlations. To validate the effectiveness and superiority of the proposed method, the Commercial Modular Aero-propulsion System Simulation (C-MAPSS) dataset is employed for comprehensive performance evaluation, the IEEE 2012 PHM bearing dataset is also adopted to demonstrate the generalization and robustness of the proposed method. The results not only show a notable improvement over existing methods but also offer a more intuitive understanding through visual representations, marking a significant stride in enhancing the safety and efficiency of complex systems.