时空特征变分学习的交通流预测模型.pdfVIP

第42卷第1期控制理论与应用Vol.42No.1

2025年1月ControlTheoryApplicationsJan.2025

时时时空空空特特特征征征变变变分分分学学学习习习的的的交交交通通通流流流预预预测测测模模模型型型

欧阳毅,汤文燕,邵泳博,黎晏伶

(浙江工商大学管理工程与电子商务学院,浙江杭州310018)

摘要:交通流量时空预测是智能交通系统的关键任务.针对城市交通流序列的非线性和多模态特性,本文提出了

一种基于时空特征融合的变分学习模型(ST-FVAE),采用局部时空特征融合和全局特征融合两个阶段对具有图空

间拓扑特性的交通流数据进行预测.局部特征融合模块由时间卷积残差单元和图卷积神经网络(GCN)构成,提取交

通节点的局部时间特征信息,并利用GCN将空间拓扑信息嵌入局部时间特征信息中.通过基于局部时空图特征融

合的变分自编码器交通流预测模型,学习全局时空相关性特征.在全局时空特征融合变分自编码器的学习过程中,

为使分布能够逼近实际数据分布,通过最大化似然函数的变分推断证据下界(ELBO)使得两个分布之间的KL散

度最小化,提出了计算分布期望的KL差异构造训练损失函数的方法,进一步提高预测准确率.通过对交通流数据集

和交通速度数据集的预测实验结果表明:本文提出的方法在交通流量和速度的预测方面都具有较好的预测特性,

对于30min和60min的预测鲁棒性更好.

关键词:交通流预测;时空融合;变分自编码器;图卷积

引用格式:欧阳毅,汤文燕,邵泳博,等.时空特征变分学习的交通流预测模型.控制理论与应用,2025,42(1):158

–166

DOI:10.7641/CTA.2023.21057

Spatial-temporalfeaturevariationalinferencemodelfor

trafficflowforecasting

OUYANGYi,TANGWen-yan,SHAOYong-bo,LIYan-ling

(SchoolofManagementEngineeringandE-Commerce,ZhejiangGongshangUniversity,HangzhouZhejiang310018,China)

Abstract:Trafficflowspatial-temporaldataforecastingisacrucialtaskforintelligenttransportationsystems.This

paperproposesavariationallearningmodelbasedonthespatio-temporalfeaturefusion(ST-FVAE).Themodelaimsto

addressthenonlinearandmulti-modalfeaturesofurbantrafficflowsequencesbyutilizinglocalspatiotemporalfeature

fusionandglobalfeaturefusion.Italsotakesintoaccountthegraphspatialtopologicalcharacteristicstopredicttr