Wavelet-Inspired Multiscale Graph Convolutional Recurrent Network for Traffic Forecasting (2401.06040v3)
Abstract: Traffic forecasting is the foundation for intelligent transportation systems. Spatiotemporal graph neural networks have demonstrated state-of-the-art performance in traffic forecasting. However, these methods do not explicitly model some of the natural characteristics in traffic data, such as the multiscale structure that encompasses spatial and temporal variations at different levels of granularity or scale. To that end, we propose a Wavelet-Inspired Graph Convolutional Recurrent Network (WavGCRN) which combines multiscale analysis (MSA)-based method with Deep Learning (DL)-based method. In WavGCRN, the traffic data is decomposed into time-frequency components with Discrete Wavelet Transformation (DWT), constructing a multi-stream input structure; then Graph Convolutional Recurrent networks (GCRNs) are employed as encoders for each stream, extracting spatiotemporal features in different scales; and finally the learnable Inversed DWT and GCRN are combined as the decoder, fusing the information from all streams for traffic metrics reconstruction and prediction. Furthermore, road-network-informed graphs and data-driven graph learning are combined to accurately capture spatial correlation. The proposed method can offer well-defined interpretability, powerful learning capability, and competitive forecasting performance on real-world traffic data sets.
- E. Cascetta, Transportation systems engineering: theory and methods, vol. 49, Springer Science & Business Media, 2013.
- T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” in Proc. Int. Conf. Learn. Representations, 2017.
- S. Hochreiter and J. Schmidhuber, “Long short-term memory,” Neural Comput., vol. 9, no. 8, pp. 1735–1780, 1997.
- “Empirical evaluation of gated recurrent neural networks on sequence modeling,” arXiv preprint arXiv:1412.3555, 2014.
- “Attention is all you need,” in Proc. Adv. Neural Inf. Process. Syst., 2017, vol. 30, p. 5998–6008.
- “Diffusion convolutional recurrent neural network: Data-driven traffic forecasting,” arXiv preprint arXiv:1707.01926, 2017.
- “Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting,” in Proc. 27th Int. Joint Conf. Artif. Intell., 2018, p. 3634–3640.
- “Dynamic graph convolutional recurrent network for traffic prediction: Benchmark and solution,” ACM Trans. Knowl. Discov. Data, vol. 17, no. 1, pp. 1–21, 2023.
- “Hierarchical graph convolution network for traffic forecasting,” in Proc. 35th AAAI Conf. Artif. Intell., 2021, vol. 35, pp. 151–159.
- “An improved wavelet–arima approach for forecasting metal prices,” Resour. Policy, vol. 39, pp. 32–41, 2014.
- “Multilevel wavelet decomposition network for interpretable time series analysis,” in Proc. 24th ACM SIGKDD Int. Conf. Knowl. Discov. Data Min., 2018, pp. 2437–2446.
- “A multiscale interactive recurrent network for time-series forecasting,” IEEE Trans. Cybern., vol. 52, no. 9, pp. 8793–8803, 2021.
- A. J. Huang and S. Agarwal, “Physics-informed deep learning for traffic state estimation: Illustrations with lwr and ctm models,” IEEE Open J. Intell. Transp. Syst., vol. 3, pp. 503–518, 2022.
- F. RK. Chung, Spectral graph theory, vol. 92, American Mathematical Soc., 1997.
- “Convolutional neural networks on graphs with fast localized spectral filtering,” in Proc. Adv. Neural Inf. Process. Syst., 2016, pp. 3844–3852.
- “Triformer: Triangular, variable-specific attentions for long sequence multivariate time series forecasting–full version,” in Proc. 31th Int. Joint Conf. Artif. Intell., 2022.
- “Connecting the dots: Multivariate time series forecasting with graph neural networks,” in Proc. 26th ACM SIGKDD Int. Conf. Knowl. Discov. Data Min., 2020, pp. 753–763.
- “Dags with no tears: Continuous optimization for structure learning,” in Proc. Int. Conf. Neural Inf. Process. Syst., 2018, pp. 9472–9483.
- “A limited memory algorithm for bound constrained optimization,” SIAM J. Sci. Comput., vol. 16, no. 5, pp. 1190–1208, 1995.
- “Big data and its technical challenges,” Commun. ACM, vol. 57, no. 7, pp. 86–94, 2014.
- “Graph wavenet for deep spatial-temporal graph modeling,” in Proc. 28th Int. Joint Conf. Artif. Intell., 2019, pp. 1–7.
- “Adaptive graph convolutional recurrent network for traffic forecasting,” in Proc. Adv. Neural Inf. Process. Syst., 2020, vol. 33, pp. 17804–17815.