Taxi Demand Prediction Using Parallel Multi-Task Learning Model

Chizhan Zhang,Fenghua Zhu,Xiao Wang,Leilei Sun,Haina Tang,Yisheng Lv
DOI: https://doi.org/10.1109/tits.2020.3015542
IF: 8.5
2022-02-01
IEEE Transactions on Intelligent Transportation Systems
Abstract:Accurate and real-time taxi demand prediction can help managers pre-allocate taxi resources in cities, which assists drivers quickly finding passengers and reduce passengers' waiting time. Most of the existing studies focus on mining spatial-temporal characteristics of taxi demand distributions, while lacking in modeling the correlations between taxi pick-up demand and the drop-off demand from the perspective of multi-task learning. In this article, we propose a multi-task learning model containing three parallel LSTM layers to co-predict taxi pick-up and drop-off demands, and compare the performance of single demand prediction methodology and that of two demands' co-prediction methodology. Experimental results on real-world datasets demonstrate that the pick-up demand and the drop-off demand do depend on each other, and the effectiveness of the proposed co-prediction methods.
engineering, electrical & electronic,transportation science & technology, civil
What problem does this paper attempt to address?