Multi-Sensor Graph Transfer Network for Health Assessment of High-Speed Rail Suspension Systems

Abstract

Suspension systems are significant for safe and comfort operation of high-speed trains. Health assessment is a useful tool to schedule maintenance plans of suspension systems, and furthermore ensure safety operation of high-speed railway transportation. In real operating condition, two problems, i.e. data imbalance and shortage of labelled data, result in difficult for health assessment of the suspension system using deep learning. In this work, a multi-sensor information fusion method, called as multi-sensor graph transfer network (MSGTN), is proposed in basis of deep transfer learning and graph neural network. In the proposed method, a domain-share multi-sensor graph neural network (MSGNN) is firstly proposed to extract features from vibration signals collected from three different positions in train vehicles. A graph-based fusion layer in MSGNN is proposed to fuse multi-sensor information by combining frequency response curves of the suspension system. The MSGTN mainly includes two parts in source and target domains respectively. In the source domain, a simple physical dynamic model of high-speed rail suspension system is built to generate labelled simulation datasets to pre-train MSGNN. In the target domain, the initial hyper-parameters of MSGNN are that of the pre-train model in the source domain. The labelled data in the target domain is fed to fine-tune MSGNN and then the final model for health assessment can be obtained by minimizing the loss function. The effectiveness of the proposed method was verified using real-work operation data.