一种利用ConvGRU解决交通流量预测问题的方法.pdfVIP

小型微型计算机系统２０２４年１０月第１０期

ＤＯＩ:１０.２０００９/ｊ.ｃｎｋｉ.２１￣１１０６/ＴＰ.２０２３￣０２３８

ＪｏｕｒｎａｌｏｆＣｈｉｎｅｓｅＣｏｍｐｕｔｅｒＳｙｓｔｅｍｓＶｏｌ.４５Ｎｏ.１０２０２４

一种利用ＣｏｎｖＧＲＵ解决交通流量预测问题的方法

１２ꎬ３１１ꎬ４１ꎬ４１２１

王玉森ꎬ景志勇ꎬ卫琳ꎬ高宇飞ꎬ石磊ꎬ王清贤ꎬ陶永才ꎬ王向杰

１(郑州大学网络空间安全学院ꎬ郑州４５０００２)

２(郑州大学计算机与人工智能学院ꎬ郑州４５０００１)

３(郑州轻工业大学软件学院ꎬ郑州４５０００１)

４(嵩山实验室ꎬ郑州４５０００３)

Ｅ￣ｍａｉｌ:ｗｙｓ＠ｇｓ.ｚｚｕ.ｅｄｕ.ｃｎ

摘要:准确的交通流量预测可协助交通管理部门工作ꎬ减少交通压力.但现有方法对交通流时间特征与空间特征学习不充

分ꎬ将二者割裂讨论ꎬ忽视了时间与空间的动态相关性.针对以上问题ꎬ本文提出了基于ＣｏｎｖＧＲＵ的多时间尺度时空卷积交通

流预测方法(ＭＴＳＴＣ)ꎬ设计浅层时空卷积模块对数据中的时空相关性进行初步提取ꎻ提出以ＣｏｎｖＧＲＵ为核心特征提取器的

深层时空卷积模块ꎬ对数据的时空特征进行更深层次挖掘ꎻ并从３种时间尺度范围的数据中提炼交通流的周期性特征ꎻ结合注

意力机制设计了时空注意力模块辅助模型训练ꎬ提升模型收敛速率.在公开数据集ＰＥＭＳ０４和ＰＥＭＳ０８上进行实验验证ꎬ结果

表明采用ＭＡＥ和ＲＭＳＥ评价指标时ꎬ本文方法的准确率相较基线方法在两个数据集上提升了３.２３％~５.６４％.

关键词:交通流量预测ꎻ时空卷积模块ꎻ注意力机制ꎻＣｏｎｖＧＲＵ

中图分类号:ＴＰ１８３文献标识码:Ａ文章编号:１０００￣１２２０(２０２４)１０￣２３５５￣０７

ＮｏｖｅｌＴｒａｆｆｉｃＦｌｏｗＰｒｅｄｉｃｔｉｏｎＭｅｔｈｏｄＢａｓｅｄｏｎＣｏｎｖＧＲＵ

１２ꎬ３１１ꎬ４１ꎬ４１２１

ＷＡＮＧＹｕｓｅｎꎬＪＩＮＧＺｈｉｙｏｎｇꎬＷＥＩＬｉｎꎬＧＡＯＹｕｆｅｉꎬＳＨＩＬｅｉꎬＷＡＮＧＱｉｎｇｘｉａｎꎬＴＡＯＹｏｎｇｃａｉꎬＷＡＮＧＸｉａｎｇｊｉｅ

１(ＳｃｈｏｏｌｏｆＣｙｂｅｒＳｃｉｅｎｃｅａｎｄＥｎｇｉｎｅｅｒｉｎｇꎬＺｈｅｎｇｚｈｏｕＵｎｉｖｅｒｓｉｔｙꎬＺｈｅｎｇｚｈｏｕ４５０００２ꎬＣｈｉｎａ)

２(ＳｃｈｏｏｌｏｆＣｏｍｐｕｔｅｒａｎｄＡｒｔｉｆｉｃｉａｌＩｎｔｅｌｌｉｇｅｎｃｅꎬＺｈｅｎｇｚｈｏｕＵｎｉｖｅｒｓｉｔｙꎬＺｈｅｎｇｚｈｏｕ４５０００１ꎬＣｈｉｎａ)

３(ＣｏｌｌｅｇｅｏｆＳｏｆｔｗａｒｅＥｎｇｉｎｅｅｒｉｎｇꎬＺｈｅｎｇｚｈｏｕＵｎｉｖｅｒｓｉｔｙｏｆＬｉｇｈｔＩｎｄｕｓｔｒｙꎬＺｈｅｎｇｚｈｏｕ４５０００１ꎬＣｈｉｎａ)

４(ＳｏｎｇｓｈａｎＬａｂｏｒａｔｏｒｙꎬＺｈｅｎｇｚｈｏｕ４５０００３ꎬＣｈｉｎａ)

Ａｂｓｔｒａｃｔ:Ｔｒａｆｆｉｃｆｌｏｗｐｒｅｄｉｃｔｉｏｎｉｓａｎｅｓｓｅｎｔｉａｌｐａｒｔｏｆｉｎｔｅｌｌｉｇｅｎｔｔｒａｎｓｐｏｒｔａｔｉｏｎｓｙｓｔｅｍ.Ａｃｃｕｒａｔｅｔｒａｆｆｉｃｆｌｏｗｐｒｅｄｉｃｔｉｏｎｃａｎｎｏｔｏｎｌｙ

ａｓｓｉｓｔｔｒａｆｆｉｃｍａｎａｇｅｍｅｎｔｔｏｒｅｄｕｃｅｔｒａｆｆｉｃｐｒｅｓｓｕｒｅｂｕｔａｌｓｏａｖｏｉｄｃｏｎｇｅｓｔｉｏｎａｎｄｔｒａｆｆｉｃａｃｃｉｄｅｎｔｓ.Ｔｈｅｃｕｒｒｅｎｔｍｅｔｈｏｄｓｄｏｎｏｔｌｅａｒｎｔｈｅ

ｔｅｍｐｏｒａｌａｎｄｓｐａｔｉａｌｆｅａｔｕｒｅｓｏｆｔｒａｆｆｉｃｆｌｏｗａｄｅｑｕａｔｅｌｙꎬａｎｄｌｅａｒｎｔｈｅｍｉｎｉｓｏｌａｔｉｏｎꎬｉｇｎｏｒｉｎｇ