arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.04440cs.LGcs.NE

使用基于小波的时间注意力进行交通流时空预测

Spatiotemporal Forecasting of Traffic Flow using Wavelet-based Temporal Attention

Sorbonne University(索邦大学);Indian Institute of Science, Education and Research, Kolkata(印度科学教育与研究学院(加尔各答));International Institute of Information Technology, Bangalore(班加罗尔国际信息技术学院)

Yash Jakhmola, Madhurima Panja, Nitish Kumar Mishra, Kripabandhu Ghosh, Uttam Kumar, Tanujit Chakraborty

AI总结:

本文提出W-DSTAGNN,利用小波分解与时间注意力建模动态时空依赖,在三个公开交通数据集上优于十个基线,并支持可靠长期点预测与区间概率预测。

摘要:

交通流数据的时空预测是机器学习领域的一个典型问题,对城市交通管理系统具有影响。一般而言,由于时间和空间维度相互交织,时空预测问题涉及复杂的相互作用、非线性和长程依赖。因此,传统统计和机器学习方法无法充分处理这些复杂交通流数据集中的时间与空间依赖。该领域一种流行方法将图卷积网络与多头注意力机制相结合,用于时空处理。本文提出一种基于小波的时间注意力模型,即基于小波的动态时空感知图神经网络(W-DSTAGNN),以解决交通预测问题。小波分解可通过将信号分解为能够独立分析的分量提供帮助,从而降低非平稳性的影响,并处理交通流数据集的长程依赖。使用三种常用统计指标进行的基准实验证实,所提方法有效捕捉了时空相关性,并在三个公开交通数据集上优于十个最先进模型(包括时间基准和时空基准)。所提出的集成方法能够更好地处理动态时间与空间依赖,并作出可靠的长期预测。除点预测外,所提模型还可生成区间预测,显著增强交通数据集的概率预测。

英文摘要:

Spatiotemporal forecasting of traffic flow data represents a typical problem in the field of machine learning, impacting urban traffic management systems. In general, spatiotemporal forecasting problems involve complex interactions, nonlinearities, and long-range dependencies due to the interwoven nature of the temporal and spatial dimensions. Due to this, traditional statistical and machine learning methods cannot adequately handle the temporal and spatial dependencies in these complex traffic flow datasets. A prevalent approach in the field combines graph convolutional networks and multi-head attention mechanisms for spatiotemporal processing. This paper proposes a wavelet-based temporal attention model, namely a wavelet-based dynamic spatiotemporal aware graph neural network (W-DSTAGNN), for tackling the traffic forecasting problem. Wavelet decomposition can help by decomposing the signal into components that can be analyzed independently, reducing the impact of non-stationarity and handling long-range dependencies of traffic flow datasets. Benchmark experiments using three popularly used statistical metrics confirm that our proposal efficiently captures spatiotemporal correlations and outperforms ten state-of-the-art models (including both temporal and spatiotemporal benchmarks) on three publicly available traffic datasets. Our proposed ensemble method can better handle dynamic temporal and spatial dependencies and make reliable long-term forecasts. In addition to point forecasts, our proposed model can generate interval forecasts that significantly enhance probabilistic forecasting for traffic datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑