• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science

Previous Articles     Next Articles

Short-term passenger flow prediction in Shanghai
subway system based on stacked autoencoder

XU Yizhi1,2,PENG Ling1,LIN Hui1,2,LI Xiang1,2   

  1. (1.Institute of Remote Sensing and Digital Earth,Chinese Academy of Sciences,Beijing 100010;
    2.University of Chinese Academy of Sciences,Beijing 100049,China)
  • Received:2017-01-06 Revised:2017-04-25 Online:2018-07-25 Published:2018-07-25

Abstract:

Urban public traffic networks have been carrying huge passenger flow all the time. And the increase of passenger flow brings great pressure to public traffic networks and traffic intelligent dispatch. Shortterm passenger flow forecasting on subway station provides an important technical support for decisionmaking in the intelligent subway dispatch system. With the historical data of metro cards, we propose a shortterm passenger flow prediction method based on deep learning which is able to extract inherent and deep features from data. So a deep learning network model is built based on stacked autoencoder (SAE) and we pretrain the model in a downtop fashion. After pretraining, we use the BP algorithm to finetune and update the whole network’s parameters in a topdown fashion. Results on the data of metro cards for a month period of Shanghai subway show that the proposed method outperforms  the wavelet neural network (waveletNN) and the autoregressive integrated moving average (ARIMA) in terms of prediction performance.
 

Key words: deep learning, stacked autoencoder, subway passenger flow, shortterm prediction