{"type":"ScholarlyArticle","title":"Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting","authors":[{"name":"Bing Yu","slug":"bing-yu","given":"Bing","family":"Yu","affiliation":"School of Mathematical Sciences , Peking University , Beijing , China"},{"name":"Haoteng Yin","slug":"haoteng-yin","given":"Haoteng","family":"Yin","affiliation":"Academy for Advanced Interdisciplinary Studies , Peking University , Beijing , China; Center for Data Science , Peking University , Beijing , China"},{"name":"Zhanxing Zhu","slug":"zhanxing-zhu","given":"Zhanxing","family":"Zhu","affiliation":"Center for Data Science , Peking University , Beijing , China; Beijing Institute of Big Data Research (BIBDR) , Beijing , China"}],"abstract":"Timely accurate traffic forecast is crucial for urban traffic control and guidance. Due to the high nonlinearity and complexity of traffic flow, traditional methods cannot satisfy the requirements of mid-and-long term prediction tasks and often neglect spatial and temporal dependencies. In this paper, we propose a novel deep learning framework, Spatio-Temporal Graph Convolutional Networks (STGCN), to tackle the time series prediction problem in traffic domain. Instead of applying regular convolutional and recurrent units, we formulate the problem on graphs and build the model with complete convolutional structures, which enable much faster training speed with fewer parameters. Experiments show that our model STGCN effectively captures comprehensive spatio-temporal correlations through modeling multi-scale traffic networks and consistently outperforms state-of-the-art baselines on various real-world traffic datasets.","doi":"10.24963/ijcai.2018/505","url":"https://ijcai.org/proceedings/2018/505","pdf":"https://ijcai.org/proceedings/2018/0505.pdf","markdown":"https://ijcai.org/proceedings/2018/505.md","bibtex":"https://ijcai.org/proceedings/2018/bibtex/505","volume":{"slug":"2018","title":"Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence","short_title":"IJCAI 2018","url":"https://ijcai.org/proceedings/2018/","year":2018,"location":"Stockholm","dates":"13-19 July 2018","editors":"Jérôme Lang","isbn":"978-0-9992411-2-7","publisher":"International Joint Conferences on Artificial Intelligence Organization","doi_prefix":"10.24963/ijcai.2018"},"section":"Machine Learning Applications","track":"Main track","number":505,"pages":{"first":3634,"last":3640},"keywords":["Machine Learning: Neural Networks","Machine Learning Applications: Applications of Supervised Learning","Machine Learning: Time-series;Data Streams"],"videos":[],"references":[{"position":1,"raw":"[Ahmed and Cook, 1979] Mohammed S Ahmed and Allen R Cook. Analysis of freeway traffic time-series data by using Box-Jenkins techniques. 1979.","title":"Analysis of freeway traffic time-series data by using Box-Jenkins techniques","authors":"S Mohammed, Allen R Ahmed, Cook","year":"1979","venue":null,"doi":null,"url":null,"internal_url":null},{"position":2,"raw":"[Bruna et al., 2013] Joan Bruna, Wojciech Zaremba, Arthur Szlam, and Yann LeCun. Spectral networks and lo- cally connected networks on graphs. arXiv preprint arXiv:1312.6203, 2013.","title":"Spectral networks and locally connected networks on graphs","authors":"Joan Bruna, Wojciech Zaremba, Arthur Szlam, Yann Lecun","year":"2013","venue":null,"doi":null,"url":null,"internal_url":null},{"position":3,"raw":"[Chen et al., 2001] Chao Chen, Karl Petty, Alexander Sk- abardonis, Pravin Varaiya, and Zhanfeng Jia. Freeway per- formance measurement system: mining loop detector data. Transportation Research Record: Journal of the Trans- portation Research Board, (1748):96-102, 2001.","title":"Freeway performance measurement system: mining loop detector data","authors":"Chao Chen, Karl Petty, Alexander Skabardonis, Pravin Varaiya, Zhanfeng Jia","year":"1748","venue":"Transportation Research Record: Journal of the Transportation Research Board","doi":null,"url":null,"internal_url":null},{"position":4,"raw":"[Chen et al., 2016] Quanjun Chen, Xuan Song, Harutoshi Yamada, and Ryosuke Shibasaki. Learning deep represen- tation from big and heterogeneous data for traffic accident inference. In AAAI, pages 338-344, 2016.","title":"Learning deep representation from big and heterogeneous data for traffic accident inference","authors":"Chen","year":"2016","venue":"AAAI","doi":null,"url":null,"internal_url":null},{"position":5,"raw":"[Defferrard et al., 2016] Michaël Defferrard, Xavier Bres- son, and Pierre Vandergheynst. Convolutional neural net- works on graphs with fast localized spectral filtering. In NIPS, pages 3844-3852, 2016.","title":"Convolutional neural networks on graphs with fast localized spectral filtering","authors":"Michaël Defferrard, Xavier Bresson, Pierre Vandergheynst","year":"2016","venue":"NIPS","doi":null,"url":null,"internal_url":null},{"position":6,"raw":"[Gehring et al., 2017] Jonas Gehring, Michael Auli, David Grangier, Denis Yarats, and Yann N Dauphin. Convo- lutional sequence to sequence learning. arXiv preprint arXiv:1705.03122, 2017.","title":"Convolutional sequence to sequence learning","authors":"Jonas Gehring, Michael Auli, David Grangier, Denis Yarats, Yann N Dauphin","year":"2017","venue":null,"doi":null,"url":null,"internal_url":null},{"position":7,"raw":"[Hammond et al., 2011] David K Hammond, Pierre Van- dergheynst, and Rémi Gribonval. Wavelets on graphs via spectral graph theory. Applied and Computational Har- monic Analysis, 30(2):129-150, 2011.","title":"Wavelets on graphs via spectral graph theory","authors":"Pierre David K Hammond, Rémi Vandergheynst, Gribonval","year":"2011","venue":"Applied and Computational Harmonic Analysis","doi":null,"url":null,"internal_url":null},{"position":8,"raw":"[Hochreiter and Schmidhuber, 1997] Sepp Hochreiter and Jürgen Schmidhuber. Long short-term memory. Neural computation, 9(8):1735-1780, 1997.","title":"Long short-term memory","authors":"Schmidhuber Hochreiter, Sepp Hochreiter, Jürgen Schmidhuber","year":"1997","venue":"Neural computation","doi":null,"url":null,"internal_url":null},{"position":9,"raw":"[Huang et al., 2014] Wenhao Huang, Guojie Song, Haikun Hong, and Kunqing Xie. Deep architecture for traffic flow prediction: deep belief networks with multitask learning. IEEE Transactions on Intelligent Transportation Systems, 15(5):2191-2201, 2014.","title":"Deep architecture for traffic flow prediction: deep belief networks with multitask learning","authors":"Wenhao Huang, Guojie Song, Haikun Hong, Kunqing Xie","year":"2014","venue":"IEEE Transactions on Intelligent Transportation Systems","doi":null,"url":null,"internal_url":null},{"position":10,"raw":"[Jia et al., 2016] Yuhan Jia, Jianping Wu, and Yiman Du. Traffic speed prediction using deep learning method. In ITSC, pages 1217-1222. IEEE, 2016.","title":"Traffic speed prediction using deep learning method","authors":"Yuhan Jia, Jianping Wu, Yiman Du","year":"2016","venue":"ITSC","doi":null,"url":null,"internal_url":null},{"position":11,"raw":"[Kipf and Welling, 2016] Thomas N Kipf and Max Welling. Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907, 2016.","title":"Semi-supervised classification with graph convolutional networks","authors":"N Thomas, Max Kipf, Welling","year":"2016","venue":null,"doi":null,"url":null,"internal_url":null},{"position":12,"raw":"[Li et al., 2015] Yexin Li, Yu Zheng, Huichu Zhang, and Lei Chen. Traffic prediction in a bike-sharing system. In SIGSPATIAL, page 33. ACM, 2015.","title":"Traffic prediction in a bike-sharing system","authors":"Yexin Li, Yu Zheng, Huichu Zhang, Lei Chen","year":"2015","venue":"SIGSPATIAL","doi":null,"url":null,"internal_url":null},{"position":13,"raw":"[Li et al., 2018] Yaguang Li, Rose Yu, Cyrus Shahabi, and Yan Liu. Diffusion convolutional recurrent neural net- work: Data-driven traffic forecasting. In ICLR, 2018.","title":"Diffusion convolutional recurrent neural network: Data-driven traffic forecasting","authors":"Yaguang Li, Rose Yu, Cyrus Shahabi, Yan Liu","year":"2018","venue":"ICLR","doi":null,"url":null,"internal_url":null},{"position":14,"raw":"[Lv et al., 2015] Yisheng Lv, Yanjie Duan, Wenwen Kang, Zhengxi Li, and Fei-Yue Wang. Traffic flow prediction with big data: a deep learning approach. IEEE Trans- actions on Intelligent Transportation Systems, 16(2):865- 873, 2015.","title":"Traffic flow prediction with big data: a deep learning approach","authors":"Yisheng Lv, Yanjie Duan, Wenwen Kang, Zhengxi Li, Fei-Yue Wang","year":"2015","venue":"IEEE Transactions on Intelligent Transportation Systems","doi":null,"url":null,"internal_url":null},{"position":15,"raw":"[Niepert et al., 2016] Mathias Niepert, Mohamed Ahmed, and Konstantin Kutzkov. Learning convolutional neural networks for graphs. In ICML, pages 2014-2023, 2016.","title":"Learning convolutional neural networks for graphs","authors":"Mathias Niepert, Mohamed Ahmed, Konstantin Kutzkov","year":"2016","venue":"ICML","doi":null,"url":null,"internal_url":null},{"position":16,"raw":"[Seo et al., 2016] Youngjoo Seo, Michaël Defferrard, Pierre Vandergheynst, and Xavier Bresson. Structured sequence modeling with graph convolutional recurrent networks. arXiv preprint arXiv:1612.07659, 2016.","title":"Structured sequence modeling with graph convolutional recurrent networks","authors":"Seo","year":"2016","venue":null,"doi":null,"url":null,"internal_url":null},{"position":17,"raw":"[Shi et al., 2015] Xingjian Shi, Zhourong Chen, Hao Wang, Dit-Yan Yeung, Wai-Kin Wong, and Wang-chun Woo. Convolutional lstm network: A machine learning approach for precipitation nowcasting. In NIPS, pages 802-810, 2015.","title":"Convolutional lstm network: A machine learning approach for precipitation nowcasting","authors":"Shi","year":"2015","venue":"NIPS","doi":null,"url":null,"internal_url":null},{"position":18,"raw":"[Shuman et al., 2013] David I Shuman, Sunil K Narang, Pas- cal Frossard, Antonio Ortega, and Pierre Vandergheynst. The emerging field of signal processing on graphs: Ex- tending high-dimensional data analysis to networks and other irregular domains. IEEE Signal Processing Maga- zine, 30(3):83-98, 2013.","title":"The emerging field of signal processing on graphs: Extending high-dimensional data analysis to networks and other irregular domains","authors":"David I Shuman, K Sunil, Pascal Narang, Antonio Frossard, Pierre Ortega, Vandergheynst","year":"2013","venue":"IEEE Signal Processing Magazine","doi":null,"url":null,"internal_url":null},{"position":19,"raw":"[Sutskever et al., 2014] Ilya Sutskever, Oriol Vinyals, and Quoc V Le. Sequence to sequence learning with neural networks. In NIPS, pages 3104-3112, 2014.","title":"Quoc V Le. Sequence to sequence learning with neural networks","authors":"Sutskever","year":"2014","venue":"NIPS","doi":null,"url":null,"internal_url":null},{"position":20,"raw":"[Vlahogianni, 2015] Eleni I Vlahogianni. Computational in- telligence and optimization for transportation big data: challenges and opportunities. In Engineering and Applied Sciences Optimization, pages 107-128. Springer, 2015.","title":"Computational intelligence and optimization for transportation big data: challenges and opportunities","authors":"Eleni I Vlahogianni","year":"2015","venue":"Engineering and Applied Sciences Optimization","doi":null,"url":null,"internal_url":null},{"position":21,"raw":"[Williams and Hoel, 2003] Billy M Williams and Lester A Hoel. Modeling and forecasting vehicular traffic flow as a seasonal arima process: Theoretical basis and em- pirical results. Journal of transportation engineering, 129(6):664-672, 2003.","title":"Modeling and forecasting vehicular traffic flow as a seasonal arima process: Theoretical basis and empirical results","authors":"Hoel ; ] Williams, M Billy, Lester A Williams, Hoel","year":"2003","venue":"Journal of transportation engineering","doi":null,"url":null,"internal_url":null},{"position":22,"raw":"[Wu and Tan, 2016] Yuankai Wu and Huachun Tan. Short- term traffic flow forecasting with spatial-temporal correla- tion in a hybrid deep learning framework. arXiv preprint arXiv:1612.01022, 2016.","title":"Shortterm traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework","authors":"Tan ; Wu, Yuankai Wu, Huachun Tan","year":"2016","venue":null,"doi":null,"url":null,"internal_url":null}],"license_note":"Copyright International Joint Conferences on Artificial Intelligence."}