Pairwise and Hyper-correlations Based Spatiotemporal Neural Networks for Traffic Speed Predictions

The problem of traffic speed predictions is still very challenging due to the complex and dynamic urban traffic conditions. Many existing works have implied the importance of integrating spatial correlations into models to explore nonlinear spatio-temporal dependencies and make traffic predictions in near future. However, some of the works only consider pairwise correlations and cannot model the hidden information among multiple nodes well, and the others only consider hyper-correlations (that can be shared by more than two nodes) and discount the role of the pairwise ones for propagating spatial dependencies. Therefore, we propose a Spatio-Temporal neural nEtwork based on both Pairwise and Hyper-correlations (STEPH) for traffic speed predictions. It is distinguished primarily by incorporating both types of spatial correlations into temporal information and designing new hybrid spatio-temporal blocks in neural networks to effectively overcome the challenge. Experiments on two real-world traffic datasets demonstrate the effectiveness of the proposed model, and show its superiority performance compared to other state-of-the-art baselines.

[1]  K. Zeitouni,et al.  Graph convolutional networks for traffic forecasting with missing values , 2022, Data Mining and Knowledge Discovery.

[2]  Baocai Yin,et al.  Multitask Hypergraph Convolutional Networks: A Heterogeneous Traffic Prediction Framework , 2022, IEEE Transactions on Intelligent Transportation Systems.

[3]  Depeng Jin,et al.  Crowd Flow Prediction for Irregular Regions with Semantic Graph Attention Network , 2022, ACM Trans. Intell. Syst. Technol..

[4]  Xiaoyi Luo,et al.  Directed hypergraph attention network for traffic forecasting , 2021, IET Intelligent Transport Systems.

[5]  Jingyuan Wang,et al.  LibCity: An Open Library for Traffic Prediction , 2021, SIGSPATIAL/GIS.

[6]  Yanjie Fu,et al.  Dynamic and Multi-faceted Spatio-temporal Deep Learning for Traffic Speed Forecasting , 2021, KDD.

[7]  Yanfeng Sun,et al.  Hierarchical Graph Convolution Network for Traffic Forecasting , 2021, AAAI.

[8]  Yongli Hu,et al.  Metro Passenger Flow Prediction via Dynamic Hypergraph Convolution Networks , 2021, IEEE Transactions on Intelligent Transportation Systems.

[9]  Chi-Yin Chow,et al.  GAMIT: A New Encoder-Decoder Framework with Graphical Space and Multi-grained Time for Traffic Predictions , 2020, 2020 IEEE International Conference on Big Data (Big Data).

[10]  Chi-Yin Chow,et al.  STNN: A Spatio-Temporal Neural Network for Traffic Predictions , 2020, IEEE Transactions on Intelligent Transportation Systems.

[11]  Jinkyoo Park,et al.  Hypergraph Convolutional Recurrent Neural Network , 2020, KDD.

[12]  Lina Yao,et al.  Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting , 2020, NeurIPS.

[13]  Xiaojun Chang,et al.  Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks , 2020, KDD.

[14]  Gaofeng Meng,et al.  Spatio-Temporal Graph Structure Learning for Traffic Forecasting , 2020, AAAI.

[15]  Natalia Gimelshein,et al.  PyTorch: An Imperative Style, High-Performance Deep Learning Library , 2019, NeurIPS.

[16]  Cheng Wang,et al.  GMAN: A Graph Multi-Attention Network for Traffic Prediction , 2019, AAAI.

[17]  Jieping Ye,et al.  Spatiotemporal Multi-Graph Convolution Network for Ride-Hailing Demand Forecasting , 2019, AAAI.

[18]  Dafang Zhang,et al.  Dynamic Spatial-Temporal Graph Convolutional Neural Networks for Traffic Forecasting , 2019, AAAI.

[19]  Ning Feng,et al.  Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting , 2019, AAAI.

[20]  Chi-Yin Chow,et al.  STCNN: A Spatio-Temporal Convolutional Neural Network for Long-Term Traffic Prediction , 2019, 2019 20th IEEE International Conference on Mobile Data Management (MDM).

[21]  Jing Jiang,et al.  Graph WaveNet for Deep Spatial-Temporal Graph Modeling , 2019, IJCAI.

[22]  Wonjong Rhee,et al.  DDP-GCN: Multi-graph convolutional network for spatiotemporal traffic forecasting , 2019, Transportation Research Part C: Emerging Technologies.

[23]  Yu Zheng,et al.  Predicting Citywide Crowd Flows in Irregular Regions Using Multi-View Graph Convolutional Networks , 2019, IEEE Transactions on Knowledge and Data Engineering.

[24]  Yue Gao,et al.  Hypergraph Neural Networks , 2018, AAAI.

[25]  Zhanxing Zhu,et al.  Spatio-temporal Graph Convolutional Neural Network: A Deep Learning Framework for Traffic Forecasting , 2017, IJCAI.

[26]  Cyrus Shahabi,et al.  Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting , 2017, ICLR.

[27]  Yann Dauphin,et al.  Convolutional Sequence to Sequence Learning , 2017, ICML.

[28]  Tianrui Li,et al.  Predicting Citywide Crowd Flows Using Deep Spatio-Temporal Residual Networks , 2017, Artif. Intell..

[29]  Fei-Yue Wang,et al.  Travel time prediction with LSTM neural network , 2016, 2016 IEEE 19th International Conference on Intelligent Transportation Systems (ITSC).

[30]  Xavier Bresson,et al.  Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering , 2016, NIPS.

[31]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[32]  Vladlen Koltun,et al.  Multi-Scale Context Aggregation by Dilated Convolutions , 2015, ICLR.

[33]  Dit-Yan Yeung,et al.  Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting , 2015, NIPS.

[34]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[35]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[36]  Quoc V. Le,et al.  Sequence to Sequence Learning with Neural Networks , 2014, NIPS.

[37]  Wenhao Huang,et al.  Deep Architecture for Traffic Flow Prediction: Deep Belief Networks With Multitask Learning , 2014, IEEE Transactions on Intelligent Transportation Systems.

[38]  Joan Bruna,et al.  Spectral Networks and Locally Connected Networks on Graphs , 2013, ICLR.

[39]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[40]  Yoshua Bengio,et al.  Deep Sparse Rectifier Neural Networks , 2011, AISTATS.

[41]  Dongjoo Park,et al.  Forecasting Freeway Link Travel Times with a Multilayer Feedforward Neural Network , 1999 .

[42]  S. Hochreiter,et al.  Long Short-Term Memory , 1997, Neural Computation.

[43]  Jitendra Malik,et al.  Normalized cuts and image segmentation , 1997, Proceedings of IEEE Computer Society Conference on Computer Vision and Pattern Recognition.

[44]  A. Krizhevsky ImageNet Classification with Deep Convolutional Neural Networks , 2022 .