基于时间图注意力的交通流量预测模型.

Traffic condition prediction is an important part of intelligent transportation system, and traffic flow is the most direct embodiment of traffic condition. Therefore, traffic flow prediction has important application value. On the one hand, the roads in the city have spatial topological properties, on the other hand, the traffic flow changes dynamically with time. Therefore, the key to the prediction of traffic flow is to model the time and space dependence in the data. In view of this characteristic, this paper used neural network model and attention mechanism to explore the temporal and spatial dependence relationship in traffic flow data, and proposed a traffic flow prediction model based on time map attention. In terms of spatial dependence, it used a learning algorithm combining graph convolution network and attention to assign different weights to nodes with different influence degrees, and added node adaptive learning to effectively extract spatial features. In terms of time dependence, it used the temporal convolution network to extract the temporal features, and expanded the sensing domain by expanding convolution, so as to capture the features of longer time series data. A spatial-temporal network layer was composed of graph attention network and time convolution network, which was finally connected to the output layer to output the prediction results. The model used the combination of graph convolution neural network and attention mechanism to extract spatial features, fully considered the spatial relationship between roads, and used temporal convolution network to capture temporal features. After experiments on two real datasets, it is found that it has good performance in the next 15, 30 and 60 minutes, and the results are better than the existing baselines. [ABSTRACT FROM AUTHOR]