第49卷第2期2023年2月ComputerEngineering计算机工程基于时空注意力网络的动态高速路网交通速度预测邹国建1,2,赖子良1,2,李晔1,2(1.同济大学道路与交通工程教育部重点实验室,上海201804;2.同济大学交通运输工程学院,上海201804)摘要:交通速度是影响高速路网通行效率和安全的重要指标,精准预测高速路网交通速度可以减少交通事故和通行时间,预先为交通控制提供有价值的参考信息,对高速公路管理具有重要意义。基于时空注意力网络,提出一种由数据和长期预测任务驱动面向动态高速路网的交通速度预测模型(ST-ANet)。通过图注意力网络提取高速路网的动态空间关联特征,使用长短期记忆网络提取输入数据的时间关联特征。在此基础上,采用基于多头自注意力机制的时间注意力网络计算历史输入数据和预测值之间的相关性,并利用密集连接和层归一化方法进一步提升模型性能。基于中国宁夏回族自治区银川市高速路网监测数据进行实验,结果表明,与GCN-LSTM模型相比,ST-ANet模型预测未来1h、2h和3h内高速路网交通速度的平均绝对误差分别降低4.0%、3.6%和3.9%。关键词:交通工程;高速路网交通速度预测;ST-ANet预测模型;时间注意力;空间注意力;长短期记忆网络开放科学(资源服务)标志码(OSID):中文引用格式:邹国建,赖子良,李晔.基于时空注意力网络的动态高速路网交通速度预测[J].计算机工程,2023,49(2):303-313.英文引用格式:ZOUGJ,LAIZL,LIY.Trafficspeedpredictionbasedonspatio-temporalattentionnetworkfordynamicexpresswaynetwork[J].ComputerEngineering,2023,49(2):303-313.TrafficSpeedPredictionBasedonSpatio-TemporalAttentionNetworkforDynamicExpresswayNetworkZOUGuojian1,2,LAIZiliang1,2,LIYe1,2(1.TheKeyLaboratoryofRoadandTrafficEngineering,MinistryofEducation,TongjiUniversity,Shanghai201804,China;2.CollegeofTransportationEngineering,TongjiUniversity,Shanghai201804,China)【Abstract】Trafficspeedisanimportantindicatoroftheefficiencyandsafetyofexpresswaynetworktraffic.Theaccuratepredictionofexpresswaynetworktrafficspeedcanreducetrafficaccidentsandtransittime,providevaluablereferenceinformationfortrafficcontrolinadvance,andissignificantforexpresswaymanagement.ThispaperproposesaspeedpredictionmodelbasedonSpatio-TemporalAttentionNetwork...