Saturday , May 18 2024

A Mining Algorithm to Improve LSTM for Predicting Customer Churn in Railway Freight Traffic

Fangcan ZHAO1, Baotian DONG1*, Hongqin PAN2, Anqi SHI3
1 School of Traffic and Transportation, Beijing Jiaotong University,
No. 3 Shangyuancun, Haidian District, Beijing, 100044, China, btdong@ (*Corresponding author)
2 China Railway Information Technology Center, No. 6 Fuxing Road, Haidian District,
Beijing, 100844, China
3 Beijing Didi Infinity Technology and Development Co., Ltd., Building B1&B2, Digital Valley,
Zhongguancun Software Park Compound 8, Dongbeiwang Road, Haidian District, Beijing, 100000, China

Abstract: Railway freight is at risk of losing customers due to intense competition in the market. Effectively managing potential customer loss is a long-term problem for railway freight. Big data technology has been widely researched and applied in recent years, and using data mining techniques to fully extract information from railway freight ticket data and discover potential customer loss is a research topic. In this study, a mining algorithm is proposed to improve the accuracy of predicting customer churn in railway freight using Long Short-Term Memory (LSTM) network. The algorithm involves three steps: constructing the time series of railway freight volume, predicting customer churn trends using a modified LSTM model, and analyzing the characteristics of customers with different degrees of loss and the characteristics of transportation demand of lost and potential lost customers. Experimental verification was conducted using customer freight ticket data from the Shanghai Railway Bureau in China and compared the proposed modified LSTM model was compared with other commonly used machine learning algorithms for churn prediction. The results showed that the present algorithm demonstrated good accuracy and adaptability. The study proposed in this paper enriches the theoretical basis of railway freight customer management, provides an effective method for predicting customer loss in railway freight, and offers technical support for railway freight management practices.

Keywords: Rail freight, Customer churn prediction, Big Data, LSTM, Data mining, Transportation demand, Customer management.


Fangcan ZHAO, Baotian DONG, Hongqin PAN, Anqi SHI, A Mining Algorithm to Improve LSTM for Predicting Customer Churn in Railway Freight Traffic, Studies in Informatics and Control, ISSN 1220-1766, vol. 32(2), pp. 25-38, 2023.