|国家预印本平台
首页|基于图神经网络的轻量化\par 时空交通流预测方法

基于图神经网络的轻量化\par 时空交通流预测方法

Lightweight Spatiotemporal Traffic Flow Prediction Method Based on Graph Neural Network

中文摘要英文摘要

\justifying交通流量预测技术是智能交通系统的重要组成部分。目前基于深度学习的时空交通流预测已成为主流。然而深度学习方法往往具有大量的参数。良好的预测方法一方面应能保证预测的精度,另一方面又要适应目前边缘智能硬件的条件,尽可能缩小模型的尺寸,而这两者往往难以兼顾。目前,许多时空交通流预测模型忽视模型大小与预测精度间的平衡问题,在一定程度上限制了它们的实用性。针对这一问题,本文提出了一种轻量化的时空交通流预测算法:图和时间卷积网络(Graph Convolution and Temporal Convolution Network,GCTCN)。GCTCN包含一个针对交通预测改进的图卷积模块,可以有效建模长距离顶点间的依赖关系,并高效提取路网空间特征;还包含基于膨胀因果卷积的时间特征提取模块,可以提取不同尺度的时间特征。结合上述方法,在不增加参数量的前提下,同时建模长、短距离的时空依赖关系,有效地提升了预测能力。实验证明,该方法能兼顾到时空效率与预测精度。

\justifying Traffic flow prediction technology is an important component of intelligent transportation systems. Currently, spatiotemporal traffic flow prediction based on deep learning has become mainstream. However, deep learning methods often have a large number of parameters. A good prediction method should ensure both accuracy and adapt to the current conditions of edge intelligent hardware, trying to minimize the model's size. However, these two goals are often difficult to balance. Currently, many spatiotemporal traffic flow prediction models ignore the balance between model size and prediction accuracy, which limits their practicality to some extent. In response to this problem, this article proposes a lightweight spatiotemporal traffic flow prediction algorithm: Graph Convolution and Temporal Convolution Network (GCTCN). GCTCN includes a graph convolution module that is improved for traffic prediction, which can effectively model the dependence relationship between long-distance vertices and efficiently extract spatial features of the road network; it also includes a time feature extraction module based on dilated causal convolution, which can extract time features of different scales. By combining the above methods, it can model long- and short-distance spatiotemporal dependence relationships without increasing the number of parameters, effectively improving the prediction ability. Experimental results have shown that this method can balance spatiotemporal efficiency and prediction accuracy.

孙晶茹、邱子瑜

公路运输工程

深度学习图神经网络卷积神经网络

eep learning Graph neural network Convolutional neural network

孙晶茹,邱子瑜.基于图神经网络的轻量化\par 时空交通流预测方法[EB/OL].(2023-05-19)[2025-08-22].http://www.paper.edu.cn/releasepaper/content/202305-160.点此复制

评论