Deep spatio-temporal 3D densenet with multiscale ConvLSTM-Resnet network for citywide traffic flow forecasting

作者全名："He, Rui; Liu, Yanbing; Xiao, Yunpeng; Lu, Xingyu; Zhang, Song"

作者地址："[He, Rui; Liu, Yanbing; Xiao, Yunpeng; Lu, Xingyu; Zhang, Song] Chongqing Univ Posts & Telecommun, Coll Comp Sci & Technol, Chongqing 400065, Peoples R China; [Liu, Yanbing] Chongqing Med Univ, Chongqing 400016, Peoples R China"

通信作者："Liu, YB (通讯作者)，Chongqing Med Univ, Chongqing 400016, Peoples R China."

来源：KNOWLEDGE-BASED SYSTEMS

ESI学科分类：COMPUTER SCIENCE

WOS号：WOS:000833283600004

JCR分区：Q1

影响因子：8.8

年份：2022

卷号：250

期号：　

开始页：　

结束页：　

文献类型：Article

关键词：3D densenet; Traffic prediction; Spatio-temporal data mining; Neural network

摘要："Reliable traffic flow forecasting is paramount in Intelligent Transportation Systems (ITS) as it can effectively improve traffic efficiency and social security. Its vital challenge is to effectively integrate various factors (such as multiple temporal correlations, complex spatial correlation, high heterogeneous) to infer the evolution trend of future traffic flow. Inspired by spatio-temporal prediction in computer vision, we regard traffic data slices at each moment as ""traffic frames"". This paper presents an end-to-end architecture named Spatio-Temporal 3D Densenet Multiscale ConvLSTM-Resnet Network (ST-3DDMCRN) to predict future traffic flow accurately. Specifically, a 3D densenet network is applied simultaneously to capture the traffic frame's local regional spatio-temporal information. Traditional Resnet networks cannot capture long-range spatial correlation, a novel multiscale ConvLSTM-Resnet network is developed to overcome this problem, extracting traffic frame's nearby and long-range spatial dependencies. In addition, considering the spatio-temporal heterogeneity of traffic frames, a Region-Squeeze-and-Excitation (RSE) unit is designed to accurately quantify the difference of the contributions of the correlations in space. The experiment result on two datasets in the real world illustrates the ST-3DDMCRN model outperforms the state-of-art baselines for the citywide traffic flow prediction. Furthermore, to validate the model's generality, we utilize the model to predict the passenger pickupidropoff demand task, the prediction results are more accurate than the baseline methods. (C) 2022 Elsevier B.V. All rights reserved."

基金机构：National Natural Science Foundation of China [61772098]

基金资助正文：This work was supported in part by the National Natural Science Foundation of China under Grants 61772098