Taxi demand prediction based on LSTM with residuals and multi-head attention

Chih Jung Hsu, Hung Hsuan Chen

研究成果: 書貢獻/報告類型會議論文篇章同行評審

摘要

This paper presents a simple yet effective framework to accurately predict the taxi demands of different regions in a city in the near future. This framework is based on a deep-learning structure with residual connections in the LSTM layers and the attention mechanism. We found that adding residuals accelerates optimization and that adding the attention mechanism makes the model better predict the taxi demands, especially when the demand fluctuates greatly in the peak hours and off-peak hours. We conducted extensive experiments by comparing the proposed models to the time-series model (ARIMA), traditional supervised learning model (ridge regression), strong machine learning model that won many Kaggle competitions (Gradient Boosted Decision Tree implemented in the XGBoost library), and deep learning models (LSTM and DMVST-Net) on two real and open-source datasets. Experimental results show that the proposed models outperform the baselines for most cases. We believe the greatest improvement comes from the attention mechanism, which helps distinguish the demands in the peak hours and off-peak hours. Additionally, the proposed model runs 10% to 40%-times faster than the other deep-learning-based models. We applied the models to participate in a taxi demand prediction challenge and won second place out of hundreds of teams.

原文???core.languages.en_GB???
主出版物標題VEHITS 2020 - Proceedings of the 6th International Conference on Vehicle Technology and Intelligent Transport Systems
編輯Karsten Berns, Markus Helfert, Oleg Gusikhin
發行者SciTePress
頁面268-275
頁數8
ISBN(電子)9789897584190
出版狀態已出版 - 2020
事件6th International Conference on Vehicle Technology and Intelligent Transport Systems, VEHITS 2020 - Virtual, Online
持續時間: 2 5月 20204 5月 2020

出版系列

名字VEHITS 2020 - Proceedings of the 6th International Conference on Vehicle Technology and Intelligent Transport Systems

???event.eventtypes.event.conference???

???event.eventtypes.event.conference???6th International Conference on Vehicle Technology and Intelligent Transport Systems, VEHITS 2020
城市Virtual, Online
期間2/05/204/05/20

指紋

深入研究「Taxi demand prediction based on LSTM with residuals and multi-head attention」主題。共同形成了獨特的指紋。

引用此