Abstract
The prediction of traffic flow is of great importance to urban planning and intelligent transportation systems. Recently, deep learning models have been applied to study this problem. However, there still exist two main limitations: (1) They do not effectively model dynamic traffic patterns in irregular regions; (2) The traffic flow of a region is strongly correlated to the transition-flow between different regions, while this issue is largely ignored by existing approaches. To address these issues, we propose a multitask deep learning model called MTGCN for a more accurate traffic flow prediction. First, to process the input traffic network data, we propose using graph convolution in place of traditional grid-based convolution to model spatial dependencies between irregular regions. Second, as original graph convolution can not well respond to traffic dynamics, we design a novel attention mechanism to capture dynamic traffic patterns. At last, to obtain a more accurate prediction result, we integrate two correlated tasks which respectively predict two types of traffic flows (region-flow and transition-flow) as a whole, by combining the representations learned from each task in a rational way. We conduct extensive experiments on two real-world datasets and the results show that our proposed method achieves better performance compared with other baseline models.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Notes
References
Bruna, J., Zaremba, W., Szlam, A., LeCun, Y.: Spectral networks and locally connected networks on graphs. In: ICLR (2014)
Dai, J., Liu, C., Xu, J., Ding, Z.: On personalized and sequenced route planning. World Wide Web 19(4), 679–705 (2015). https://doi.org/10.1007/s11280-015-0352-2
Defferrard, M., Bresson, X., Vandergheynst, P.: Convolutional neural networks on graphs with fast localized spectral filtering. In: NIPS, pp. 3837–3845 (2016)
Guo, S., Lin, Y., Feng, N., Song, C., Wan, H.: Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In: AAAI, pp. 922–929 (2019)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: CVPR, pp. 770–778. IEEE Computer Society (2016)
Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. In: ICLR (2017)
Lin, Z., Feng, J., Lu, Z., Li, Y., Jin, D.: Deepstn+: Context-aware spatial-temporal neural network for crowd flow prediction in metropolis. In: AAAI, pp. 1020–1027 (2019)
Lv, Z., Xu, J., Zheng, K., Yin, H., Zhao, P., Zhou, X.: LC-RNN: a deep learning model for traffic speed prediction. In: IJCAI, pp. 3470–3476 (2018)
Ma, X., Tao, Z., Wang, Y., Yu, H., Wang, Y.: Long short-term memory neural network for traffic speed prediction using remote microwave sensor data. Transp. Res. Part C Emerg. Technol. 54, 187–197 (2015). https://doi.org/10.1016/j.trc.2015.03.014
Niepert, M., Ahmed, M., Kutzkov, K.: Learning convolutional neural networks for graphs. In: ICML, pp. 2014–2023 (2016)
Pan, B., Demiryurek, U., Shahabi, C.: Utilizing real-world transportation data for accurate traffic prediction. In: ICDM, pp. 595–604. IEEE Computer Society (2012)
Peng, S., Shen, Y., Zhu, Y., Chen, Y.: A frequency-aware spatio-temporal network for traffic flow prediction. In: Li, G., Yang, J., Gama, J., Natwichai, J., Tong, Y. (eds.) DASFAA 2019. LNCS, vol. 11447, pp. 697–712. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-18579-4_41
Shi, X., Chen, Z., Wang, H., Yeung, D., Wong, W., Woo, W.: Convolutional LSTM network: A machine learning approach for precipitation nowcasting. In: NIPS, pp. 802–810 (2015)
Sun, J., Zhang, J., Li, Q., Yi, X., Zheng, Y.: Predicting citywide crowd flows in irregular regions using multi-view graph convolutional networks. CoRR abs/1903.07789 (2019). http://arxiv.org/abs/1903.07789
Velickovic, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., Bengio, Y.: Graph attention networks. In: ICLR (2018)
Wu, C., Ho, J., Lee, D.: Travel-time prediction with support vector regression. IEEE Trans. Intell. Transp. Syst. 5(4), 276–281 (2004)
Xu, J., Chen, J., Zhou, R., Fang, J., Liu, C.: On workflow aware location-based service composition for personal trip planning. Future Gener. Comput. Syst. 98, 274–285 (2019)
Yu, B., Yin, H., Zhu, Z.: Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. In: IJCAI, pp. 3634–3640 (2018)
Yuan, J., Zheng, Y., Xie, X.: Discovering regions of different functions in a city using human mobility and POIs. In: SIGKDD, pp. 186–194. ACM (2012)
Zhang, J., Zheng, Y., Sun, J., Qi, D.: Flow prediction in spatio-temporal networks based on multitask deep learning. In: TKDE, pp. 1 (2019). https://doi.org/10.1109/TKDE.2019.2891537
Zhang, J., Zheng, Y., Qi, D.: Deep spatio-temporal residual networks for citywide crowd flows prediction. In: AAAI, pp. 1655–1661 (2017)
Zhang, J., Zheng, Y., Qi, D., Li, R., Yi, X.: DNN-based prediction model for spatio-temporal data. In: SIGSPATIAL, pp. 92:1–92:4. ACM (2016)
Zhang, J., Wang, F., Wang, K., Lin, W., Xu, X., Chen, C.: Data-driven intelligent transportation systems: a survey. IEEE Trans. Intell. Transp. Syst. 12(4), 1624–1639 (2011)
Zhao, J., Xu, J., Zhou, R., Zhao, P., Liu, C., Zhu, F.: On prediction of user destination by sub-trajectory understanding: a deep learning based approach. In: CIKM, pp. 1413–1422 (2018)
Zonoozi, A., Kim, J., Li, X., Cong, G.: Periodic-CRN: a convolutional recurrent model for crowd density prediction with recurring periodic patterns. In: IJCAI, pp. 3732–3738 (2018)
Acknowledgements
This work was supported by the National Natural Science Foundation of China under Grant Nos. 61872258, 61772356, 61876117, and 61802273, the Australian Research Council discovery projects under grant numbers DP170104747, DP180100212, and the Open Program of State Key Laboratory of Software Architecture under item number SKLSAOP1801.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Wang, F., Xu, J., Liu, C., Zhou, R., Zhao, P. (2020). MTGCN: A Multitask Deep Learning Model for Traffic Flow Prediction. In: Nah, Y., Cui, B., Lee, SW., Yu, J.X., Moon, YS., Whang, S.E. (eds) Database Systems for Advanced Applications. DASFAA 2020. Lecture Notes in Computer Science(), vol 12112. Springer, Cham. https://doi.org/10.1007/978-3-030-59410-7_30
Download citation
DOI: https://doi.org/10.1007/978-3-030-59410-7_30
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-59409-1
Online ISBN: 978-3-030-59410-7
eBook Packages: Mathematics and StatisticsMathematics and Statistics (R0)