Projects per year
Abstract
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.
Original language | English |
---|---|
Title of host publication | VEHITS 2020 - Proceedings of the 6th International Conference on Vehicle Technology and Intelligent Transport Systems |
Editors | Karsten Berns, Markus Helfert, Oleg Gusikhin |
Publisher | SciTePress |
Pages | 268-275 |
Number of pages | 8 |
ISBN (Electronic) | 9789897584190 |
State | Published - 2020 |
Event | 6th International Conference on Vehicle Technology and Intelligent Transport Systems, VEHITS 2020 - Virtual, Online Duration: 2 May 2020 → 4 May 2020 |
Publication series
Name | VEHITS 2020 - Proceedings of the 6th International Conference on Vehicle Technology and Intelligent Transport Systems |
---|
Conference
Conference | 6th International Conference on Vehicle Technology and Intelligent Transport Systems, VEHITS 2020 |
---|---|
City | Virtual, Online |
Period | 2/05/20 → 4/05/20 |
Keywords
- Deep Learning
- LSTM
- Multi-head Attention
- Residual Connection
- Taxi Demand
Fingerprint
Dive into the research topics of 'Taxi demand prediction based on LSTM with residuals and multi-head attention'. Together they form a unique fingerprint.Projects
- 1 Finished
-
A Study on the Multi-Objective Recommender Systems Based on Deep Learning(2/3)
Chen, H.-H. (PI)
1/08/19 → 31/07/20
Project: Research