3D-ConvLSTMNet: A Deep Spatio-Temporal Model for Traffic Flow Prediction

Lihua He, Wuman Luo

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

5 Citations (Scopus)

Abstract

Spatiotemporal correlations are crucial for traffic flow prediction. So far, various traffic flow prediction methods based on convolutional neural network (CNN) and long short-term memory (LSTM) network have been proposed. However, the common CNN - based models cannot preserve the temporal information after the first layer. Although the 3D CNN-based models can effectively capture short-term spatial and tempo-ral features, they are not suitable for long-term information capturing. LSTM is excellent at long-term features extraction. However, it alone cannot be used for spatial information extraction. To address these issues, we propose a deep architecture called 3D-ConvLSTMNet to better capture the spatiotemporal correlations among the traffic data. Specifically, we proposed a short-long term spatiotemporal feature extraction module called 3D-ConvLSTM, which uses 3D CNN to extract short-term spatiotemporal correlations, and uses ConvLSTM to extract the long-term spatiotemporal correlations. To get the long-distance spatial features, we adopt the residual neural network to develop the depth of 3D-ConvLSTMNet. Finally, we utilize a channel-wise attention mechanism to quantify the contribution of each grid in space domain. To evaluate the performances of ConvLSTMNet, we conduct extensive experiments on two real-world datasets. The experiment results show that our model gets better performances than the other state-of-the-art methods.

Original languageEnglish
Title of host publicationProceedings - 2022 23rd IEEE International Conference on Mobile Data Management, MDM 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages147-152
Number of pages6
ISBN (Electronic)9781665451765
DOIs
Publication statusPublished - 2022
Event23rd IEEE International Conference on Mobile Data Management, MDM 2022 - Virtual, Paphos, Cyprus
Duration: 6 Jun 20229 Jun 2022

Publication series

NameProceedings - IEEE International Conference on Mobile Data Management
Volume2022-June
ISSN (Print)1551-6245

Conference

Conference23rd IEEE International Conference on Mobile Data Management, MDM 2022
Country/TerritoryCyprus
CityVirtual, Paphos
Period6/06/229/06/22

Keywords

  • 3D CNN
  • ConvLSTM
  • Traffic flow prediction
  • channel-wise attention
  • residual neural network

Fingerprint

Dive into the research topics of '3D-ConvLSTMNet: A Deep Spatio-Temporal Model for Traffic Flow Prediction'. Together they form a unique fingerprint.

Cite this