Tang Jie, Zhu Rong, Wu Fengyun, He Xuansen, Huang Jing, Zhou Xianlai, Sun Yishuai
School of Modern Information Industry, Guangzhou College of Commerce, Guangzhou, 510000, China.
School of Information and Intelligent Engineering, Guangzhou Xinhua University, Guangzhou, 510000, China.
Sci Rep. 2025 Apr 6;15(1):11743. doi: 10.1038/s41598-025-95711-6.
With the rapid development of economy, the concept of intelligent transportation system (ITS) and smart city has been mentioned. The most important part of building them is whether they can accurately predict traffic flow. An accurate traffic flow forecast can help manage traffic, plan travel paths in advance, and rationally allocate public resources such as shared bicycles. The biggest difficulty in this task is how to solve the problem of spatial imbalance and the problem of temporal imbalance. In this paper, we propose a deep learning algorithm STDConvLSTM. Firstly, for spatial features, most scholars use convolutional neural networks (with fixed kernel size) to capture. However, this does not solve the problem of spatial imbalance, i.e. each region has a different size of correlated regions (e.g., the busy area has a wider range of correlated regions). In this paper, we design a space-dependent attention mechanism, which assigns a convolutional neural network with a different kernel size to each region through attention weights. Secondly, for time features, most scholars use time series prediction models, such as recurrent neural networks and their variants. However, in the actual forecasting process, the importance of historical data in different time steps is not the same. In this paper, we design a time-dependent attention mechanism that assigns different weights to historical data to solve the time imbalance. In the end, we ran experiments on two real-world data sets and achieve good performance.
随着经济的快速发展,智能交通系统(ITS)和智慧城市的概念被提了出来。构建它们最重要的部分在于能否准确预测交通流量。准确的交通流量预测有助于管理交通、提前规划出行路径以及合理分配共享单车等公共资源。这项任务最大的困难在于如何解决空间不平衡问题和时间不平衡问题。在本文中,我们提出了一种深度学习算法STDConvLSTM。首先,对于空间特征,大多数学者使用卷积神经网络(内核大小固定)来进行捕捉。然而,这并没有解决空间不平衡问题,即每个区域具有不同大小的相关区域(例如,繁忙区域具有更广泛的相关区域范围)。在本文中,我们设计了一种空间依赖注意力机制,通过注意力权重为每个区域分配具有不同内核大小的卷积神经网络。其次,对于时间特征,大多数学者使用时间序列预测模型,如递归神经网络及其变体。然而,在实际预测过程中,不同时间步长的历史数据的重要性并不相同。在本文中,我们设计了一种时间依赖注意力机制,为历史数据分配不同权重以解决时间不平衡问题。最后,我们在两个真实数据集上进行了实验并取得了良好的性能。