Prediction of high-speed train wheel-rail relationship based on maximum information coefficient and LSTM

Yulong Cui; Wei Dong; Yanhong Shi; Yanghui Niu; Shuwei Shen

doi:10.1117/12.2674513

23 May 2023 Prediction of high-speed train wheel-rail relationship based on maximum information coefficient and LSTM

Yulong Cui, Wei Dong, Yanhong Shi, Yanghui Niu, Shuwei Shen

Proceedings Volume 12604, International Conference on Computer Graphics, Artificial Intelligence, and Data Processing (ICCAID 2022); 126041B (2023) https://doi.org/10.1117/12.2674513
Event: 2nd International Conference on Computer Graphics, Artificial Intelligence, and Data Processing (ICCAID 2022), 2022, Guangzhou, China

Abstract

Aiming at the prediction of wheel-rail relationship of high-speed trains, a prediction method combining maximum information coefficient (MIC) and LSTM was proposed. Firstly, the dynamic model of high-speed train was established by SIMPACK and the original dataset was obtained. Secondly, the maximum information coefficient is used to preprocess the data. Then, a prediction model of wheel-rail relationship based on long short-term memory neural network was built. Adam optimizer was used to optimize the learning rate and network structure. Finally, the optimized long short-term memory neural network is used to predict the wheel-rail relationship. The prediction results of wheel-rail relationship show that when the number of model iterations is 45, and the number of hidden layers is 110, the average absolute error percentage of model prediction is the smallest, and the value is 0.0247. Under these conditions, the predicted result is very close to the real value, that is, the data pretreated by the maximum information coefficient can make the model accurately predict the change trend of wheel-rail relationship, which can provide support for further research.

Citation Download Citation

Yulong Cui, Wei Dong, Yanhong Shi, Yanghui Niu, and Shuwei Shen "Prediction of high-speed train wheel-rail relationship based on maximum information coefficient and LSTM", Proc. SPIE 12604, International Conference on Computer Graphics, Artificial Intelligence, and Data Processing (ICCAID 2022), 126041B (23 May 2023); https://doi.org/10.1117/12.2674513

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available