结合变种残差模型和Transformer的城市公路短时交通流预测.pdfVIP

JournalofComputerApplicationsISSN1001⁃90812024⁃09⁃10

计算机应用,2024,44(9):2947-2951CODENJYIIDUhttp：//www.joca.cn

文章编号：1001-9081（2024）09-2947-05DOI：10.11772/j.issn.1001-9081.2023091262

结合变种残差模型和Transformer的

城市公路短时交通流预测

112*1

杨鑫，陈雪妮，吴春江，周世杰

（1.电子科技大学信息与软件工程学院，成都610054；2.成都信息工程大学软件工程学院，成都610225）

（∗通信作者电子邮箱chunjiangwu126.com）

摘要：城市公路交通流的预测受到历史交通流量和相邻车道交通流量的影响，蕴含了复杂的时空特征。针对传

统交通流预测模型卷积长短时记忆（ConvLSTM）网络进行交通流预测时，未将时空特征分开提取而造成的特征提取

不充分、特征信息混淆和特征信息缺失等问题，对ConvLSTM模型作出改进。首先，提取每个采样时刻的交通流数据

的短期时间特征和空间特征，并在特定的维度下将交通流的短期时空特征融合；其次，进行残差映射；最后，将映射后

的短期时空特征交由Transformer模型捕捉交通流数据长期的时空特征，并根据所捕捉的长期特征对未来时刻每个采

样点交通流进行预测。使用加州城市快速路数据对模型进行验证，以平均绝对误差（MAE）作为模型评价指标时，所

提模型相较于Conv-Transformer模型，预测精度提高了18%，验证了所提模型的有效性。

关键词：短时交通流预测；交通流；时空特征提取；残差结构；Transformer；组合模型

中图分类号：TP18；U491.1文献标志码：A

Short-termtrafficflowpredictionofurbanhighwaybasedon

variantresidualmodelandTransformer

112*1

YANGXin，CHENXueni，WUChunjiang，ZHOUShijie

（1.SchoolofInformationandSoftwareEngineering，UniversityofElectronicScienceandTechnologyofChina，ChengduSichuan610054，China；

2.SchoolofSoftwareEngineering，ChengduUniversityofInformationTechnology，ChengduSichuan610225，China）

Abstract:Thepredictionofurbanhighwaytrafficflowisinflu