ABSTRACT
Taxi demand prediction is of much importance, which enables the building of intelligent systems and smart city. It is necessary to predict taxi demand accurately to schedule taxi fleet in a reasonable and efficient way and to reduce the pressure of traffic jam. However, the taxi demand involves complex and non-linear spatial-temporal impacts. The superiority of deep learning makes people explore the possibility to apply it to traffic prediction. State-of-the-art methods on taxi demand prediction only capture static spatial correlations between regions (e.g., Using static graph embedding) and only take taxi demand data into consideration. We propose a Multi-Task Spatial-Temporal Graph Attention Network (MSTGAT-Net) framework which models the correlations between regions dynamically with graph-attention network and captures the correlation between taxi pick up and taxi drop off with multi-task training. To the best of our knowledge, it is the first paper to address the taxi demand prediction problem with graph attention network and multi-task learning. Experiments on real-world taxi data show that our model is superior to state-of-the-art methods.
- Moreira-Matias, L., Gama, J., Ferreira, M., Mendes-Moreira, J., & Damas, L. 2013. Predicting taxi-passenger demand using streaming data. IEEE Transactions on Intelligent Transportation Systems, 14(3), 1393--1402.Google ScholarDigital Library
- D. Wang, W. Cao, J. Li and J. Ye, DeepSD: Supply-Demand Prediction for Online Car-Hailing Services Using Deep Neural Networks, 2017. IEEE 33rd International Conference on Data Engineering, San Diego, CA, 2017, 243--254.Google Scholar
- Ren, S., He, K., Girshick, R., & Sun, J. 2015. Faster r-cnn: towards real-time object detection with region proposal networks. IEEE Transactions on Pattern Analysis & Machine Intelligence, 39(6), 1137--1149.Google ScholarDigital Library
- Devlin, J., Chang, M. W., Lee, K., & Toutanova, K. 2018. Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805.Google Scholar
- Zhang, J., Zheng, Y., & Qi, D. 2017. Deep spatio-temporal residual networks for citywide crowd flows prediction. In Thirty-First AAAI Conference on Artificial Intelligence.Google ScholarDigital Library
- LeCun, Y. 2015. LeNet-5, convolutional neural networks. URL: http://yann.lecun.com/exdb/lenet, 20, 5.Google Scholar
- Hochreiter, S., & Schmidhuber, J. 1997. Long short-term memory. Neural computation, 9(8), 1735--1780.Google Scholar
- Tang, J., Qu, M., Wang, M., Zhang, M., Yan, J., & Mei, Q. 2015. Line: Large-scale information network embedding. In Proceedings of the 24th international conference on world wide web, 1067--1077.Google ScholarDigital Library
- Veličković, P., Cucurull, G., Casanova, A., Romero, A., Lio, P., & Bengio, Y. 2017. Graph attention networks. arXiv preprint arXiv: 1710.10903.Google Scholar
- Yao, H., Wu, F., Ke, J., Tang, X., Jia, Y., Lu, S. 2018. Deep multi-view spatial-temporal network for taxi demand prediction. In Thirty-Second AAAI Conference on Artificial Intelligence.Google ScholarCross Ref
- Nair, V., & Hinton, G. E. 2010. Rectified linear units improve restricted boltzmann machines. In Proceedings of the 27th international conference on machine learning, 807--814.Google ScholarDigital Library
- Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Polosukhin, I. 2017. Attention is all you need. In Advances in neural information processing systems, 5998--6008.Google Scholar
- Li, Y., Fu, K., Wang, Z., Shahabi, C., Ye, J., & Liu, Y. 2018. Multi-task representation learning for travel time estimation. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, 1695--1704.Google ScholarDigital Library
- Kingma, D. P., & Ba, J. 2014. Adam: A method for stochastic optimization. arXiv preprint arXiv: 1412.6980.Google Scholar
- Box, G. E., & Pierce, D. A. 1970. Distribution of residual autocorrelations in autoregressive-integrated moving average time series models. Journal of the American statistical Association, 65(332), 1509--1526.Google ScholarCross Ref
Index Terms
- Multi-Task Spatial-Temporal Graph Attention Network for Taxi Demand Prediction
Recommendations
Exploring the Side Information Fusion Method with Spatial-temporal Model for Taxi Demand Prediction
DSIT 2020: Proceedings of the 3rd International Conference on Data Science and Information TechnologyTaxi is one of the most common public transport, predicting taxi demand precisely within an area is of great significance for improving efficiency of traffic. Taxi demand is spatial-temporal data, and highly influenced by many external factors, such as ...
Unifying Uber and taxi data via deep models for taxi passenger demand prediction
AbstractTaxi passenger demand prediction is of great significance to perceive citywide human mobility and make a lot of urban sensing applications more convenient. There are two major challenges to develop accurate predictive models, i.e., the complexity ...
Multi-Attribute Spatial-temporal Graph Convolutional Network for Taxi Demand Forecasting
ICBDT '22: Proceedings of the 5th International Conference on Big Data TechnologiesAccurate forecasting taxi demand help reduce waiting time for drivers and passengers as well as ease traffic congestion. However, most of the current research work has mostly ignored the impact of historical cab inflows and potential spatial ...
Comments