Transfer Knowledge Between Sub-regions for Traffic Prediction Using Deep Learning Method

  • Yi RenEmail author
  • Kunqing Xie
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11871)


In modern society, traffic is a very important aspect of our social lives, but a lot of problems arise such as traffic accident, congestion, air pollution, etc. The emergence of ITS (Intelligent Transportation System) brings a range of scientific ways to tackle these problems, among which there is a crucial need to study prediction methods in modern traffic systems. Fortunately, big data in ITS and advanced machine learning technologies motivate us to implement extensive researches in this area. In this paper, we introduce an innovative method for traffic prediction. We split the whole traffic system into separate sub-regions. Then, each of them would be modeled by deep graph neural networks, which could extract unique characters of the region as well as preserve the topology property of it. Moreover, we transfer the useful knowledge between these regions and share the basic information in order to improve the performance of our model. Finally, we conduct experiment on real world data set and achieve the best performance, which proves the effectiveness of our method.


Data mining Deep learning Big data Traffic prediction Intelligent Transportation System Smart city 


  1. 1.
    Kaysi, I., Benakiva, M., Koutsopoulos, H.-N., et al.: Integrated approach to vehicle routing and congestion prediction for real-time driver guidance. J. Transp. Res. Record. 1408, 66–74 (1993)Google Scholar
  2. 2.
    Cetin, M., Comert, G.: Short-term traffic flow prediction with regime switching models. J. Transp. Res. Record. 1965(1), 23–31 (2006)CrossRefGoogle Scholar
  3. 3.
    Vanajakshi, L., Rilett, L.-R.: A comparison of the performance of artificial neural networks and support vector machines for the prediction of traffic speed. In: Intelligent Vehicles Symposium, pp. 194–199 (2004)Google Scholar
  4. 4.
    Castillo, E., Menéndez, J.-M., Sánchez-Cambronero, S.: Predicting traffic flow using Bayesian networks. J. Transp. Res. Part B-Methodol. 42(5), 482–509 (2008)CrossRefGoogle Scholar
  5. 5.
    Lv, Y.-S., Duan, Y.-J., Kang, W.-W., Li, Z.-X., Wang, F.-Y.: Traffic flow prediction with big data: a deep learning approach. J. IEEE Trans. Intell. Transp. Syst. 16(2), 865–873 (2015)Google Scholar
  6. 6.
    Cui, Z.-Y., Ke, R.-M., Wang, Y.-H.: Deep stacked bidirectional and unidirectional LSTM recurrent neural network for network-wide traffic speed prediction. In: Proceedings of the International Workshop on Urban Computing (2017)Google Scholar
  7. 7.
    Zhang, J.-B., Zheng, Y., Qi. D.-K.: Deep spatio-temporal residual networks for citywide crowd flows prediction. In: Proceeding of the Thirty-First AAAI Conference on Artificial Intelligence (2017)Google Scholar
  8. 8.
    Perozzi, B., Al-Rfou, R., Skiena, S.: Deep walk: online learning of social representations. In: Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 701–710. ACM, New York (2014)Google Scholar
  9. 9.
    Grover, A., Leskovec, J.: node2vec: scalable feature learning for networks. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (2016)Google Scholar
  10. 10.
    Kipf, T.-N., Welling, M.: Semi-supervised classification with graph convolutional networks. In: Proceedings of International Conference on Learning Representations (ICLR 2017) (2017)Google Scholar
  11. 11.
    Niepert, M., Ahmed, M., Kutzkov, K.: Learning convolutional neural networks for graphs. In: Proceedings of the 33rd International Conference on Machine Learning, PMLR 48, pp. 2014–2023 (2016)Google Scholar
  12. 12.
    Velickovic, P., Cucurull, G., Casanova, A., Romero, A., et al.: Graph attention networks. In: Proceedings of ICLR (2018)Google Scholar
  13. 13.
    Hammond, D.-K., Vandergheynst, P., Gribonval, R.: Wavelets on graphs via spectral graph theory. J. Appl. Comput. Harmonic Anal. 30(2), 129–150 (2011)MathSciNetCrossRefGoogle Scholar
  14. 14.
    Zeiler, M.D., Fergus, R.: Visualizing and understanding convolutional networks. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8689, pp. 818–833. Springer, Cham (2014). Scholar
  15. 15.
    Yosinski, J., Clune, J., Bengio, Y., Lipson, H.: How transferable are features in deep neural networks? In: Advances in Neural Information Processing Systems, vol. 27, pp. 3320–3328 (2014)Google Scholar
  16. 16.
    Long, M., Cao, Y., Wang, J., et al.: Learning transferable features with deep adaptation networks. In: 32nd International Conference on Machine Learning (ICML 2015) (2015)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.Key Laboratory of Machine Perception (Ministry of Education), School of Electronics Engineering and Computer SciencePeking UniversityBeijingChina

Personalised recommendations