提出高效建模交通流多粒度异质依赖的神经网络,提升预测精度与效率。
STAHGNet: Modeling Hybrid-grained Heterogenous Dependency Efficiently for Traffic Prediction
- 设计混合粒度图注意力模块与粗粒度时序图生成器,联合捕捉多尺度依赖
- 在四个真实数据集上实现更低误差(如PeMSD3 MAE 14.82),优于8个基线和4个SOTA方法
- 引入领域知识特征工程与随机采样策略,计算量减少至少4倍,适合实际部署
交通流预测在智能交通系统中至关重要,但受复杂时空模式及随时间演变的异质性影响,仍具挑战。现有方法多仅关注时空依赖或隐式相似图构建,忽视了混合粒度演化过程。本文提出一种数据驱动的端到端框架——时空感知混合图网络(STAHGNet),通过精心设计的混合图注意力模块(HGAT)与粗粒度时序图生成器(CTG),同步建模序列中的多粒度异质相关性。此外,结合领域知识的汽车特征工程与随机邻居采样策略,显著提升效率并降低计算复杂度。采用MAE、RMSE、MAPE作为评估指标,在四个真实数据集上,本模型超越八种经典基线及四种SOTA方法(如PeMSD3上MAE为14.82,PeMSD4上MAE为18.92)。大量实验与可视化验证了各组件有效性。在计算成本方面,相比前代SOTA模型,内存消耗至少降低四倍。该模型将有助于实现更高效的交通流预测与智能交通系统建设。
原文摘要 · Abstract (English)
Traffic flow prediction plays a critical role in the intelligent transportation system, and it is also a challenging task because of the underlying complex Spatio-temporal patterns and heterogeneities evolving across time. However, most present works mostly concentrate on solely capturing Spatial-temporal dependency or extracting implicit similarity graphs, but the hybrid-granularity evolution is ignored in their modeling process. In this paper, we proposed a novel data-driven end-to-end framework, named Spatio-Temporal Aware Hybrid Graph Network (STAHGNet), to couple the hybrid-grained heterogeneous correlations in series simultaneously through an elaborately Hybrid Graph Attention Module (HGAT) and Coarse-granularity Temporal Graph (CTG) generator. Furthermore, an automotive feature engineering with domain knowledge and a random neighbor sampling strategy is utilized to improve efficiency and reduce computational complexity. The MAE, RMSE, and MAPE are used for evaluation metrics. Tested on four real-life datasets, our proposal outperforms eight classical baselines and four state-of-the-art (SOTA) methods (e.g., MAE 14.82 on PeMSD3; MAE 18.92 on PeMSD4). Besides, extensive experiments and visualizations verify the effectiveness of each component in STAHGNet. In terms of computational cost, STAHGNet saves at least four times the space compared to the previous SOTA models. The proposed model will be beneficial for more efficient TFP as well as intelligent transport system construction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。