首页|基于动态扩散图卷积的交通流量预测算法

基于动态扩散图卷积的交通流量预测算法

扫码查看
为了得到准确的交通流量预测结果,提出一种基于动态扩散图卷积的交通流量预测模型。首先,利用扩散图卷积模型对不同节点间的空间特征进行学习;其次,通过引入动态邻接矩阵,以确保各节点在不同时刻间的特征都得到充分学习;再次,采用门控循环单元,对交通流量数据进行时间特征提取;最后,通过模型层级间的残差连接,传递更多原始信息以增强模型的稳定性。在4个公开数据集上的实验结果证明本文算法在交通流量预测任务中的有效性。
Traffic flow prediction algorithm based on dynamic diffusion graph convolution
In order to obtain accurate traffic flow prediction results,a traffic flow prediction algorithm based on dynamic diffusion graph convolution was proposed.Firstly,the model used the diffusion graph convolution model to learn the spatial characteristics between different nodes.Secondly,the dynamic adjacency matrix was introduced to ensure that the characteristics of each node at each time can be learned.Once more,the model used the gated recurrent unit to extract the time characteristics of traffic flow data.Finally,residual connection between model levels was used to transfer more original information and enhance the stability of the model.The experimental results on four open data sets can prove the effectiveness of the algorithm in traffic flow prediction tasks.

artificial intelligencetraffic flow predictiongated recurrent unitdiffusion graph convolution

井佩光、田雨豆、汪少初、李云、苏育挺

展开 >

天津大学 电气自动化与信息工程学院,天津 300072

天津市测绘院有限公司,天津 300072

广西财经学院 大数据与人工智能学院,南宁 530001

人工智能 交通流量预测 门控循环单元 扩散图卷积

国家自然科学基金项目辽宁省自然科学基金项目

623610022023-MS-139

2024

吉林大学学报(工学版)
吉林大学

吉林大学学报(工学版)

CSTPCD北大核心
影响因子:0.792
ISSN:1671-5497
年,卷(期):2024.54(6)
  • 1