Affiliation:
1. Yonsei University, South Korea
Abstract
Traffic forecasting is one of the most popular spatio-temporal tasks in the field of machine learning. A prevalent approach in the field is to combine graph convolutional networks and recurrent neural networks for the spatio-temporal processing. There has been fierce competition and many novel methods have been proposed. In this article, we present the method of spatio-temporal graph neural rough differential equation (STG-NRDE). Neural rough differential equations (NRDEs) are a breakthrough concept for processing time-series data. Their main concept is to use the log-signature transform to convert a time-series sample into a relatively shorter series of feature vectors. We extend the concept and design two NRDEs: one for the temporal processing and the other for the spatial processing. After that, we combine them into a single framework. We conduct experiments with 6 benchmark datasets and 27 baselines. STG-NRDE shows the best accuracy in all cases, outperforming all those 27 baselines by non-trivial margins.
Funder
Yonsei University Research Fund of 2022
Institute of Information & Communications Technology Planning & Evaluation
Korean government
Artificial Intelligence Graduate School Program
Context and Activity Analysis-based Solution for Safe Childcare
Publisher
Association for Computing Machinery (ACM)
Subject
Artificial Intelligence,Theoretical Computer Science
Reference61 articles.
1. STG2Seq: Spatial-Temporal Graph to Sequence Model for Multi-step Passenger Demand Forecasting
2. Lei Bai, Lina Yao, Can Li, Xianzhi Wang, and Can Wang. 2020. Adaptive graph convolutional recurrent network for traffic forecasting. In Proceedings of the International Conference on Advances in Neural Information Processing Systems (NeurIPS’20). 17804–17815.
3. An empirical evaluation of generic convolutional and recurrent networks for sequence modeling;Bai Shaojie;arXiv:1803.01271,2018
4. Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting
5. Freeway Performance Measurement System: Mining Loop Detector Data
Cited by
2 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献