LLM2D
GraphSparseNet:一种新型的大规模交通流预测方法
GraphSparseNet: a Novel Method for Large Scale Traffic Flow Prediction
作者: Weiyang Kong, Kaiqi Wu, Sen Zhang, Yubao Liu
发布日期: 5/14/2025
arXiv ID: oai:arXiv.org:2502.19823v2

摘要

arXiv:2502.19823v2 通知类型: 替换-交叉 摘要:交通流量预测是智能路线规划和动态交通管理中广泛应用于时空数据挖掘的关键任务。最近深度学习的发展,特别是通过图神经网络(GNNs),在捕捉复杂的时空动态方面显著提高了这些预测的准确性。然而,由于随着图中节点数增加模型复杂性的指数增长,GNN 的可扩展性仍是一个挑战。现有解决这一问题的方法,包括稀疏化、分解和核基方法,要么没有完全解决复杂性问题,要么有损预测准确性。本文提出了 GraphSparseNet(GSNet),这是一种新型框架,旨在提高基于 GNN 的交通预测模型的可扩展性和准确性。GraphSparseNet 包含两个核心模块:特征提取器和关系压缩器。这两个模块具有线性时间和空间复杂度,从而将模型的整体计算复杂度降低到线性规模。我们对多个真实世界数据集进行的广泛实验表明,GraphSparseNet 不仅将训练时间显著缩短了 3.51 倍,与最先进的线性模型相比,而且保持了高度的预测性能。