MSAE: A Multitask Learning Approach for Traffic Flow Prediction Using Deep Neural Network

  • Di Yang
  • Hua-Min YangEmail author
  • Peng Wang
  • Song-Jiang Li
Conference paper
Part of the Smart Innovation, Systems and Technologies book series (SIST, volume 156)


Traffic  flow prediction has been regarded as one of the key problems in Intelligent Transportation Systems. Neural networks are widely leveraged in traffic prediction task, but with the limitation of (1) single task learning, which would ignore shared information among traffic network; and (2) initialization problem that would directly affect the training efficiency. In this work, we propose a deep neural network based multitask learning approach for traffic flow prediction, called MSAE, which incorporates stacked autoencoders (SAE) for the neural network initialization, and jointly and adaptively predicts network-scale traffic flow with shared information. The experiments on real traffic data indicate that the proposed model, jointly considering shared information among different traffic flow prediction tasks and neural network initialization, is capable and promising of dealing with complex traffic flow forecasting with satisfying accuracy and effectiveness.


Intelligent transportation systems Traffic flow prediction Multitask learning Stacked autoencoders 


  1. 1.
    Ahmed, M.S., Cook, A.R.: Analysis of freeway traffic time-series data by using box-jenkins techniques. Trans. Res. Rec. 722 (1979)Google Scholar
  2. 2.
    Williams, B.M., Hoel, L.A.: Modeling and forecasting vehicular traffic flow as a seasonal arima process: theoretical basis and empirical results. J. Trans. Eng. ASCE 129(6), 664–672 (2003)CrossRefGoogle Scholar
  3. 3.
    Vlahogianni, E.I., Karlaftis, M.G., Golias, J.C.: Optimized and meta-optimized neural networks for short-term traffic flow prediction: a genetic approach. Trans. Res. Part C Emerg. Technol. 13(3), 211–234 (2005)CrossRefGoogle Scholar
  4. 4.
    Jiang, X., Adeli, H.: Dynamic wavelet neural network model for traffic flow forecasting. J. Trans. Eng. ASCE 131(10), 771–779 (2005)CrossRefGoogle Scholar
  5. 5.
    Jin, F., Sun, S.: Neural network multitask learning for traffic flow forecasting. In: 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence), pp. 1897–1901 (2008)Google Scholar
  6. 6.
    Lv, Y., Duan, Y., Kang, W., Li, Z., Wang, F.Y.: Traffic flow prediction with big data: a deep learning approach. IEEE Trans. Intell. Trans. Syst. 16(2), 865–873 (2015)Google Scholar
  7. 7.
    Zhao, Z., Chen, W., Wu, X., Chen, P.C.Y., Liu, J.: Lstm network: a deep learning approach for short-term traffic forecast. IET Intell. Trans. Syst. 11(2), 68–75 (2017)CrossRefGoogle Scholar
  8. 8.
    Caruana, R.: Multitask learning. Mach. Learn. 28(1), 41–75 (1997) (Special issue on inductive transfer archive)Google Scholar
  9. 9.
    Bengio, Y., Lamblin, P., Popovici, D., Larochelle, H.: Greedy layer-wise training of deep networks. Adv. Neural Inf. Process. Syst. 19, 153–160 (2006)Google Scholar
  10. 10.
    Xia, Y., Hu, R.: Fuzzy neural network based energy efficiencies control in the heating energy supply system responding to the changes of user demands. J. Netw. Intell. 2(2), 186–194 (2017)MathSciNetGoogle Scholar
  11. 11.
    Liu, L., Chen, R.C.: A novel passenger flow prediction model using deep learning methods. Trans. Res. Part C Emerg. Technol. 84, 74–91 (2017)CrossRefGoogle Scholar
  12. 12.
    Kumar, A., Daum, H.: Learning task grouping and overlap in multi-task learning. In: International Conference on Machine Learning, pp. 1723–1730 (2012)Google Scholar
  13. 13.
    Pan, S.J., Yang, Q.: A survey on transfer learning. IEEE Trans. Knowl. Data Eng. 22(10), 1345–1359 (2010)CrossRefGoogle Scholar
  14. 14.
    California Departement of Transportation. Caltrans Performance Measurement System. (2018). Accessed 8 Apr 2018
  15. 15.
    Highway Capacity Manual. Volumes 1–4: Transporation Research Board (2010)Google Scholar

Copyright information

© Springer Nature Singapore Pte Ltd. 2020

Authors and Affiliations

  • Di Yang
    • 1
  • Hua-Min Yang
    • 1
    Email author
  • Peng Wang
    • 1
  • Song-Jiang Li
    • 1
  1. 1.School of Computer Science and TechnologyChangchun University of Science and TechnologyChangchunChina

Personalised recommendations