提出新型图卷积模型,高效预测大城市路网交通流。
Efficient Traffic Forecasting on Large-Scale Road Network by Regularized Adaptive Graph Convolution
- 用余弦相似度构建线性复杂度图卷积,提升大规模路网处理能力。
- 结合自适应图卷积与正则化嵌入,精度优于现有方法。
- 适合需要高精度且实时性的城市交通预测场景。
交通预测在时空建模中至关重要,广泛应用于出行规划和城市管理。为捕捉交通数据中的复杂时空依赖关系,时空图卷积网络(STGCNs)被广泛应用并取得先进性能。然而,在大规模路网中,传统图卷积的二次计算复杂度严重限制了其可扩展性。现有方法通过近似、压缩或空间分割缓解此问题,但往往难以兼顾计算效率与预测精度。为此,本文提出正则化自适应图卷积(RAGC)模型:首先,设计高效余弦算子(ECO),基于节点嵌入的余弦相似度实现线性时间复杂度的图卷积;其次,引入结合随机共享嵌入(SSE)与自适应图卷积的正则化框架,通过残差差异机制学习高质量节点嵌入,兼顾精度与效率。在四个大规模真实交通数据集上的实验表明,RAGC在预测精度上持续超越当前最优方法,同时具备良好的计算效率。代码已公开于 https://github.com/wkq-wukaiqi/RAGC。
原文摘要 · Abstract (English)
Traffic prediction is a critical task in spatial-temporal forecasting with broad applications in travel planning and urban management. To model the complex spatial-temporal dependencies in traffic data, Spatial-Temporal Graph Convolutional Networks (STGCNs) have been widely employed, achieving advanced performance. However, when applied to large-scale road networks, the quadratic computational complexity of traditional graph convolution operations severely limits their scalability. Several methods attempt to address this issue through approximation, compression, or spatial partitioning. Nevertheless, these methods often either fail to achieve sufficient computational efficiency or compromise prediction accuracy. To address these challenges, we propose a Regularized Adaptive Graph Convolution (RAGC) model. First, to ensure scalability on large road networks, we develop the Efficient Cosine Operator (ECO), which performs graph convolution based on the cosine similarity of node embeddings with linear time complexity. Second, we introduce a regularized adaptive graph convolution framework that combines Stochastic Shared Embedding (SSE) and adaptive graph convolution through a residual difference mechanism. This design enables the model to learn high-quality node embeddings, thereby improving prediction accuracy while maintaining computational efficiency. Extensive experiments on four large-scale real-world traffic datasets show that RAGC consistently outperforms state-of-the-art methods in terms of prediction accuracy and exhibits competitive computational efficiency. The code is available at: https://github.com/wkq-wukaiqi/RAGC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。