Research output: Contribution to Journal/Magazine › Journal article › peer-review
Research output: Contribution to Journal/Magazine › Journal article › peer-review
}
TY - JOUR
T1 - ST-InNet
T2 - Deep Spatio-Temporal Inception Networks for Traffic Flow Prediction in Smart Cities
AU - Dai, Fei
AU - Huang, Penggui
AU - Mo, Qi
AU - Xu, Xiaolong
AU - Bilal, Muhammad
AU - Song, Houbing
PY - 2022/10/1
Y1 - 2022/10/1
N2 - Traffic flow prediction plays a critical role in reducing traffic congestion in transportation systems. However, accurate traffic flow prediction becomes challenging due to the impact of complex spatio-temporal (ST) correlations and the diversity of ST correlations. When modeling complicated ST correlations, researchers usu did not take the diversity of ST correlations into consideration, resulting in poor prediction accuracy. In this paper, we propose ST-InNet, a deep spatio-temporal Inception network for collectively predicting traffic flow in each city region. Specifically, ST-InNet employs two Inception networks to simultaneously capture various spatial and temporal correlations of traffic data, including temporal closeness, temporal periodicity, nearby spatial dependencies, and distant spatial dependencies. For the diversity of spatial correlations, ST-InNet presents an improved variant of an Inception module to explicitly capture the different contributions of spatial correlations for each region. For the diversity of temporal correlations, ST-InNet designs a fusion component to explicitly model the varying contributions of temporal correlations on prediction. The experiments are conducted on a real-world traffic dataset in Nanjing, demonstrating that ST-InNet outperforms five state-of-the-art baselines in short-term and long-term traffic flow predictions with an average accuracy improvement of 32.09% and 30.97%, respectively.
AB - Traffic flow prediction plays a critical role in reducing traffic congestion in transportation systems. However, accurate traffic flow prediction becomes challenging due to the impact of complex spatio-temporal (ST) correlations and the diversity of ST correlations. When modeling complicated ST correlations, researchers usu did not take the diversity of ST correlations into consideration, resulting in poor prediction accuracy. In this paper, we propose ST-InNet, a deep spatio-temporal Inception network for collectively predicting traffic flow in each city region. Specifically, ST-InNet employs two Inception networks to simultaneously capture various spatial and temporal correlations of traffic data, including temporal closeness, temporal periodicity, nearby spatial dependencies, and distant spatial dependencies. For the diversity of spatial correlations, ST-InNet presents an improved variant of an Inception module to explicitly capture the different contributions of spatial correlations for each region. For the diversity of temporal correlations, ST-InNet designs a fusion component to explicitly model the varying contributions of temporal correlations on prediction. The experiments are conducted on a real-world traffic dataset in Nanjing, demonstrating that ST-InNet outperforms five state-of-the-art baselines in short-term and long-term traffic flow predictions with an average accuracy improvement of 32.09% and 30.97%, respectively.
KW - inception networks
KW - spatial correlations
KW - temporal correlations
KW - the diversity of spatio-temporal correlations
KW - Traffic flow prediction
U2 - 10.1109/TITS.2022.3179789
DO - 10.1109/TITS.2022.3179789
M3 - Journal article
AN - SCOPUS:85132775260
VL - 23
SP - 19782
EP - 19794
JO - IEEE Transactions on Intelligent Transportation Systems
JF - IEEE Transactions on Intelligent Transportation Systems
SN - 1524-9050
IS - 10
ER -