Prediction of Key Parameters of Wheelset Based on LSTM Neural Network

As a key component of rail vehicle operation, the running condition of the wheelset significantly affects the operational safety of track vehicles. The wheel diameter, flange thickness, and flange height are key dimensional parameters of the wheelset, which directly influence the correct position of wheelsets on the track, and the train needs to be continuously monitored during the passenger operation. A prediction model for the key parameters of the wheelset is established based on LSTM (long short-term memory) neural network, and real measured data of wheelsets from the Shanghai Metro vehicles are selected. The predicted results of the model are compared and analyzed, and the results show that the LSTM-based prediction model for key parameters of wheelsets performs well, with the mean absolute percentage errors (MAPEs) for wheel diameter, flange thickness, and flange height being 0.08%, 0.42%, and 0.44%, respectively, for the left wheel and 0.07%, 0.35%, and 0.44%, respectively, for the right wheel. The prediction model for the train wheelset parameters established in this paper has a good prediction accuracy. By predicting the key parameters of the wheelset, the faults and causes of the wheelset can be found and determined, which is helpful for engineers to overhaul the wheelset faults, make maintenance plans, and perform preventive maintenance.