A Novel Spatiotemporal Enhanced Convolutional Autoencoder Network for Unsupervised Health Indicator Construction

Shuaiyong Li,Chao Zhang,Xuyuntao Zhang
DOI: https://doi.org/10.1109/tim.2024.3383052
IF: 5.6
2024-04-09
IEEE Transactions on Instrumentation and Measurement
Abstract:The health indicator (HI) of rotating machinery affects the reliability and accuracy of its remaining useful life (RUL) prediction. Convolutional autoencoder (CAE) is widely used for HI construction with the advantage of being unsupervised, but it still suffers from two problems: 1) the shallow learning of local spatial features and the neglect of global temporal features result in poor performance of the constructed HI trend and 2) a non-end-to-end form that requires human involvement, which is time-consuming and labor-intensive. Therefore, a novel unsupervised spatiotemporal enhanced convolutional autoencoder (STECAE) network is proposed in this article, which can directly enhance feature mining on raw data in both temporal and spatial dimensions without any prior knowledge. The STECAE network mainly consists of three components: spatial mining module (SMM), temporal learning module (TLM), and trend compensation mechanism (TCM). SMM enhances the deep mining of local spatial features via deep residual shrinkage module. TLM mines global temporal features through the gate recurrent unit (GRU). TCM acts as a bridge for spatiotemporal connectivity to improve the model's global sensing field and HI's trend performance. The effectiveness of the proposed method is validated through the dataset of rolling bearings and aero engines. The results show that compared to the existing state-of-the-art methods, the HI constructed by the proposed method is closer to 1 in terms of the evaluation metric values and has a lower RUL prediction error. The high-quality HI constructed by the proposed method can provide a reliable basis for RUL prediction.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?