Graph Convolutional Modules for Traffic Forecasting

Graph convolutional network is a generalization of convolutional network for learning graph-structured data. In some of the recent works on traffic networks, a few graph convolutional blocks have been designed and shown to be useful. In this work, we extend the ideas and provide a systematic way of creating graph convolutional modules. The method consists of designing basic weighted adjacency matrices as the smallest building blocks, defining partition functions that can partition a weighted adjacency matrix into M matrices that can also serve as building blocks, and finally designing graph convolutional modules using the building blocks. We evaluate some of the designed modules by replacing the graph convolutional parts in STGCN and DCRNN, and find 8.4% to 25.0% reduction in speed estimation error.

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

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

[3]  Jianqiang Huang,et al.  Dynamic Spatio-temporal Graph-based CNNs for Traffic Prediction , 2018, ArXiv.

[4]  Qiang Yang,et al.  Bike flow prediction with multi-graph convolutional networks , 2018, SIGSPATIAL/GIS.

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

[6]  Bin Yu,et al.  Spatio-temporal Graph Convolutional Neural Network: A Deep Learning Framework for Traffic Forecasting , 2017 .

[7]  Yinhai Wang,et al.  Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting , 2018, IEEE Transactions on Intelligent Transportation Systems.

[8]  Zhiyong Cui,et al.  High-Order Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting , 2018, ArXiv.

[9]  Haifeng Li,et al.  Temporal Graph Convolutional Network for Urban Traffic Flow Prediction Method , 2018, ArXiv.

[10]  W. Tobler.  A Computer Movie Simulating Urban Growth in the Detroit Region , 1970 .

[11]  Zhanxing Zhu,et al.  ST-UNet: A Spatio-Temporal U-Network for Graph-structured Time Series Modeling , 2019, ArXiv.

[12]  Max Welling,et al.  Semi-Supervised Classification with Graph Convolutional Networks , 2016, ICLR.

[13]  Philip S. Yu,et al.  A Comprehensive Survey on Graph Neural Networks , 2019, IEEE Transactions on Neural Networks and Learning Systems.

[14]  Cyrus Shahabi,et al.  Graph Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting , 2017, ArXiv.

[15]  Pascal Frossard,et al.  Signal Processing on Graphs: Extending High-Dimensional Data Analysis to Networks and Other Irregular Data Domains , 2012, ArXiv.

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

[17]  Wei Xu,et al.  DeepTransport: Learning Spatial-Temporal Dependency for Traffic Condition Forecasting , 2017, 2018 International Joint Conference on Neural Networks (IJCNN).

[18]  Alexandre M. Bayen,et al.  Traffic state estimation on highway: A comprehensive survey , 2017, Annu. Rev. Control..

[19]  Edsger W. Dijkstra,et al.  A note on two problems in connexion with graphs , 1959, Numerische Mathematik.