Temperature Prediction Model of High-Speed Train Traction Motor Based on SE-BILSTM-CNN

With the widespread operation of high-speed trains, traction motors are susceptible to failures under high-temperature conditions and complex working scenarios. This leads to abnormal temperature increases, which pose significant threats to train safety. Therefore, researching the temperature prediction of traction motors is of great significance for ensuring the safe operation of trains. This paper proposes a temperature prediction method based on an improved bidirectional long short-term memory (Bi-LSTM) network. Initially, convolutional neural networks (CNN) are employed to extract temporal structures from the data. Simultaneously, the squeeze-and-excitation (SE) attention mechanism is integrated with the key-query-value (KQV) mechanism. Finally, Bi-LSTM is utilized for temperature prediction. The proposed SE-BiLSTM-CNN model exhibits significantly enhanced prediction performance compared with recurrent neural networks (RNN), gated recurrent units (GRU), LSTM, and Bi-LSTM models, with improvements of 35.51, 56.07, and 33.75% in the evaluation metrics of MAE, MSE, and RMSE, respectively.