Traffic flow prediction is critical for intelligent transportation systems, facing challenges in modeling complex networks and extracting dynamic traffic patterns. As traffic networks scale, computational complexity and large hidden state spaces of models become issues, potentially leading to high computational costs and weak generalization, which can limit practical use. To tackle these challenges, in this paper, we propose a Spatio-temporal Dual Graph Network with Learnable Bases (STLB-GN), an efficient and effective model for complex spatio-temporal correlations: (i) Considering the evident periodicity in traffic flow data, a dynamic periodicity embedding learning approach is adopted for modeling the traffic network structure, which can better explore the data’s latent periodicity; (ii) We introduce learnable bases to replace Chebyshev polynomials in spectral graph convolution, thereby enhancing the model’s generalization ability; (iii) A dual graph convolution is proposed to automatically identify specific patterns for individual nodes and shared patterns between nodes, helping the model in capturing complex spatio-temporal features within traffic flow data; (iv) An improved multi-scale gated temporal convolution is proposed to grasp the temporal dynamics of traffic flow data and solve the temporal inconsistency problem in previous studies; (v) We performed a series of experiments and evaluations on real public transportation datasets. The outcomes show that STLB-GN performs more competitively than state-of-the-art models. Furthermore, STLB-GN shows high computational efficiency, which outperforms other high-performance models.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

STLB-GN: Spatio-Temporal Dual Graph Network with Learnable Bases

  • Sikai Lin,
  • Guanyuan Zeng,
  • Guoting Chen

摘要

Traffic flow prediction is critical for intelligent transportation systems, facing challenges in modeling complex networks and extracting dynamic traffic patterns. As traffic networks scale, computational complexity and large hidden state spaces of models become issues, potentially leading to high computational costs and weak generalization, which can limit practical use. To tackle these challenges, in this paper, we propose a Spatio-temporal Dual Graph Network with Learnable Bases (STLB-GN), an efficient and effective model for complex spatio-temporal correlations: (i) Considering the evident periodicity in traffic flow data, a dynamic periodicity embedding learning approach is adopted for modeling the traffic network structure, which can better explore the data’s latent periodicity; (ii) We introduce learnable bases to replace Chebyshev polynomials in spectral graph convolution, thereby enhancing the model’s generalization ability; (iii) A dual graph convolution is proposed to automatically identify specific patterns for individual nodes and shared patterns between nodes, helping the model in capturing complex spatio-temporal features within traffic flow data; (iv) An improved multi-scale gated temporal convolution is proposed to grasp the temporal dynamics of traffic flow data and solve the temporal inconsistency problem in previous studies; (v) We performed a series of experiments and evaluations on real public transportation datasets. The outcomes show that STLB-GN performs more competitively than state-of-the-art models. Furthermore, STLB-GN shows high computational efficiency, which outperforms other high-performance models.