Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (12): 2765-2774    DOI: 10.3785/j.issn.1008-973X.2026.12.022
土木工程、交通工程     
因果先验引导的交通流时空预测模型
侯越1(),王艳1,董雨璇1,赵建国1,张琳栋1,马静2,3,宿增强3
1. 兰州交通大学 电子与信息工程学院,甘肃 兰州 730070
2. 西安交通大学 计算机科学与技术学院,陕西 西安 710000
3. 甘肃省交通规划勘察设计院股份有限公司,甘肃 兰州 730070
Causal prior-guided spatiotemporal traffic flow prediction model
Yue HOU1(),Yan WANG1,Yuxuan DONG1,Jianguo ZHAO1,Lindong ZHANG1,Jing MA2,3,Zengqiang SU3
1. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
2. School of Computer Science and Technology, Xi’an Jiaotong University, Xi’an 710000, China
3. Gansu Province Transportation Planning Survey andb Design Institute Limited Company, Lanzhou 730070, China
 全文: PDF(1253 KB)   HTML
摘要:

针对交通网络中流量变化驱动因素难以识别、预测性能有待提升的问题,提出因果先验引导的交通流时空预测模型. 采用时空特征融合模块,对空间维度进行离散化建模,并与周期性嵌入相融合,增强模型对复杂非线性特征的表征能力. 利用潜在因果发现模块,引入时滞表征因果卷积和因果筛选器,挖掘时间序列间的潜在因果关系. 采用因果验证模块,在假设检验框架下筛选真实因果关系,计算具有因果效应的节点之间的时延信息. 将经因果验证的结构作为先验知识嵌入动态图卷积模块,指导信息传播与特征聚合. 实验结果表明,与基线模型相比,所提模型在PeMSD4、PeMSD8和成都-滴滴数据集上的MAE指标分别降低了4.9197%、3.9518%、2.1968%,验证了因果机制在交通流预测中的有效性.

关键词: 交通流预测;  因果发现;  因果验证;  深度学习;  时空性    
Abstract:

A causal prior-guided spatiotemporal traffic flow prediction model was proposed in order to address the challenges of identifying the driving factors behind traffic flow variations and improving prediction performance. The spatiotemporal feature fusion module was used to discretize the spatial dimension and integrate it with periodic embeddings in order to enhance the representation of complex nonlinear feature. The latent causal discovery module was utilized to introduce a lag-aware causal convolution and a causal filter in order to capture potential causal relationship among time series. The causal validation module was used to identify true causal relationship under a hypothesis testing framework in order to calculate time delay between nodes with causal effect. The causally validated structure was embedded as prior knowledge into a dynamic graph convolution module in order to guide information propagation and feature aggregation. The experimental results demonstrated that the proposed model achieved reduction in MAE by 4.9197%, 3.9518% and 2.1968% on PeMSD4, PeMSD8 and Chengdu-DiDi datasets, respectively, compared with baseline models. The effectiveness of the causal mechanism in traffic flow prediction was verified.

Key words: traffic flow prediction    causal discovery    causal validation    deep learning    spatio-temporality
收稿日期: 2025-09-12 出版日期: 2026-10-21
CLC:  U 491  
基金资助: 国家自然科学基金资助项目(62363020);甘肃省交通运输厅科技项目(2025-28).
作者简介: 侯越(1979—),女,教授,博士,从事大数据智能交通研究. orcid.org/0000-0002-8289-329X. E-mail:houyue@mail.lzjtu.cn
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
侯越
王艳
董雨璇
赵建国
张琳栋
马静
宿增强

引用本文:

侯越,王艳,董雨璇,赵建国,张琳栋,马静,宿增强. 因果先验引导的交通流时空预测模型[J]. 浙江大学学报(工学版), 2026, 60(12): 2765-2774.

Yue HOU,Yan WANG,Yuxuan DONG,Jianguo ZHAO,Lindong ZHANG,Jing MA,Zengqiang SU. Causal prior-guided spatiotemporal traffic flow prediction model. Journal of ZheJiang University (Engineering Science), 2026, 60(12): 2765-2774.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.12.022    或    https://www.zjujournals.com/eng/CN/Y2026/V60/I12/2765

图 1  CP-STN模型的整体架构
图 2  原始流量序列离散前、后的对比图
图 3  潜在因果发现模块图
图 4  因果验证模块图
模型PeMSD4PeMSD8成都-滴滴
MAERMSER2MAERMSER2MAERMSER2
TCN23.0636.030.945619.2230.160.95462.794.050.8234
LSTM22.1436.110.946920.1731.500.95372.784.040.8263
DCRNN22.2834.290.916616.7425.630.92882.784.090.5818
ASTGCN20.4432.810.919316.4925.360.92852.353.490.6920
SCINet19.7531.690.959816.5025.710.96792.253.450.6905
STG-NCDE20.2931.360.950915.8324.890.97082.323.440.6932
DeepSTUQ(23)19.1531.360.933915.0423.970.93882.293.440.6933
BGCN(24)18.8331.730.959414.8223.840.97322.353.340.7523
FGI(24)19.3131.680.928214.9323.640.93962.193.310.7142
GINTRIP(24)18.6631.010.957214.9124.050.97382.293.370.7075
CP-STN18.3630.230.967814.3423.600.96992.143.180.8469
表 1  不同模型在3个交通流数据集上的预测准确度
图 5  CP-STN与基准模型在3个数据集上的预测性能对比
模型MAERMSER2
w/o D18.3730.650.9634
w/o DW19.3231.510.9417
w/o CD18.4230.730.9601
w/o IQR18.4330.710.9599
w/o CV18.4030.600.9631
CP-STN18.3630.230.9678
表 2  CP-STN模型在PeMSD4数据集上的消融实验结果
模型MAERMSER2
w/o D14.4423.850.9683
w/o DW14.5924.930.9510
w/o CD14.5123.980.9613
w/o IQR14.4724.010.9611
w/o CV14.4923.910.9612
CP-STN14.3423.610.9699
表 3  CP-STN模型在PeMSD8数据集上的消融实验结果
图 6  部分节点因果注意力分数热力图
图 7  部分节点在不同时间段的流量变化
图 8  部分节点间的因果时延图
LMAERMSER2
114.5823.630.9651
214.3423.600.9699
314.4523.610.9613
表 4  不同模型层数在PeMSD8数据集上的平均预测结果
mMAERMSER2
0.814.5323.910.9591
1.014.4123.720.9621
1.214.3423.600.9699
1.414.3923.690.9617
表 5  不同超参数m在PeMSD8数据集上的平均预测结果
αMAERMSER2
0.2014.3523.710.9683
0.2514.3423.600.9699
0.3014.3823.820.9613
0.3514.4723.930.9551
表 6  不同超参数α在PeMSD8数据集上的平均预测结果
模型ttr/sMAENp/103
ASTGCN12716.49179.46
SCINet6416.502 778.62
STG-NCDE24115.832 547.28
DeepSTUQ3515.04198.64
GINTRIP2114.91415.87
CP-STN6314.34218.63
表 7  基于时间、精度与参数量的模型效率对比
1 崔建勋, 要甲, 赵泊媛 基于深度学习的短期交通流预测方法综述[J]. 交通运输工程学报, 2024, 24 (2): 50- 64
Cui Jianxun, Yao Jia, Zhao Boyuan A review of short-term traffic flow prediction methods based on deep learning[J]. Journal of Transportation Engineering, 2024, 24 (2): 50- 64
doi: 10.19818/j.cnki.1671-1637.2024.02.003
2 秦凡. 基于时空图神经网络的交通流量预测方法研究[D]. 北京: 北京化工大学, 2025.
Qin Fan. Research on traffic flow prediction method based on spatio-temporal graph neural network [D]. Beijing: Beijing University of Chemical Technology, 2025.
3 Li Y G, Yu R, Shahabi C, et al. Diffusion convolutional recurrent neural network: data-driven traffic forecasting [PP/OL]. arXiv (2017-07-06) [2025-05-27]. https://arxiv.org/abs/1707.01926.
4 Zhang H, Lu G, Zhan M, et al Semi-supervised classification of graph convolutional networks with Laplacian rank constraints[J]. Neural Process Lett 54, 2022, 54 (4): 2645- 2656
doi: 10.1007/s11063-020-10404-7
5 ZAREMBA W, SUTSKEVER I, VINYALS O. Recurrent neural network regularization [PP/OL]. arXiv (2015-02-19) [2025-05-27]. https://arxiv.org/abs/1409.2329.
6 Guo S, Lin Y, Feng N, et al. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting [C]//Proceedings of the AAAI Conference on Artificial Intelligence. Honolulu: AAAI Press, 2019: 922−929.
7 林霆堉, 邬群勇, 王珂月 基于时间图卷积网络的交通流时空预测模型[J]. 贵州大学学报(自然科学版), 2025, 42 (4): 88- 93
Lin Tingyu, Wu Qunyong, Wang Keyue Traffic flow spatiotemporal prediction model based on temporal graph convolutional network[J]. Journal of Guizhou University (Natural Science Edition), 2025, 42 (4): 88- 93
doi: 10.15958/j.cnki.gdxbzrb.2025.04.11
8 Song C, Lin Y, Guo S, et al. Spatial-temporal synchronous graph convolutional networks: a new framework for spatial-temporal network data forecasting [C]// Proceedings of the AAAI Conference on Artificial Intelligence. Palo Alto: AAAI Press, 2020: 914–921.
9 贺文武, 裴博彧, 李雅婷, 等 基于双向自适应门控图卷积网络的交通流预测[J]. 交通运输系统工程与信息, 2023, 23 (1): 187- 197
He Wenwu, Pei Boyu, Li Yating, et al Traffic flow prediction based on bidirectional adaptive gated graph convolutional network[J]. Journal of Transportation Systems Engineering and Information, 2023, 23 (1): 187- 197
doi: 10.16097/j.cnki.1009-6744.2023.01.020
10 Zhao Y, Deng P, Liu J, et al. Causal conditional hidden Markov model for multimodal traffic prediction [C]//Proceedings of the AAAI Conference on Artificial Intelligence. Washington, DC: AAAI Press, 2023: 4929-4936.
11 Liu W, Zheng Y, Chawla S, et al. Discovering spatio-temporal causal interactions in traffic data streams [C]//Proceedings of the 17th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. New York: ACM, 2011: 1010–1018.
12 Kapoor V, Saxena D, Raychoudhury V, et al. Real time building and maintaining causal congestion graph for intelligent traffic management [C]//2018 IEEE International Conference on Pervasive Computing and Communications Workshops. Athens: IEEE, 2018: 770-775.
13 Granger C W J Investigating causal relations by econometric models and cross-spectral methods[J]. Econometrica: Journal of the Econometric Society, 1969, 37 (3): 424- 438
14 Li L, Su X, Wang Y, et al Robust causal dependence mining in big data network and its application to traffic flow predictions[J]. Transportation Research Part C: Emerging Technologies, 2015, 58: 292- 307
doi: 10.1016/j.trc.2015.03.003
15 He S, Luo Q, Du R, et al STGC-GNNs: a GNN-based traffic prediction framework with a spatial–temporal Granger causality graph[J]. Physica A: Statistical Mechanics and its Applications, 2023, 623: 128913
doi: 10.1016/j.physa.2023.128913
16 Zhang Z, Ren S, Qian X, et al. Learning flexible time-windowed granger causality integrating heterogeneous interventional time series data [C]//Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining. New York: ACM, 2024: 4408-4418.
17 Liu M, Zeng A, Chen M, et al Scinet: time series modeling and forecasting with sample convolution and interaction[J]. Advances in Neural Information Processing Systems, 2022, 35: 5816- 5828
18 Choi J, Choi H, Hwang J, et al Graph neural controlled differential equations for traffic forecasting[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2022, 36 (6): 6367- 6374
doi: 10.21203/rs.3.rs-3972614/v1
19 Qian W Z, Zhang D L, Zhao Y, et al. Uncertainty quantification for traffic forecasting: a unified approach [C]//2023 IEEE 39th International Conference on Data Engineering. Anaheim: IEEE, 2023: 992−1004.
20 Fu J, Zhou W, Chen Z Bayesian graph convolutional network for traffic prediction[J]. Neurocomputing, 2024, 582: 127507
doi: 10.1016/j.neucom.2024.127507
21 Hou Y, Zhang D, Li D, et al Regional traffic flow combination prediction model considering virtual space of the road network[J]. Physica A: Statistical Mechanics and its Applications, 2024, 637: 129598
doi: 10.1016/j.physa.2024.129598
[1] 杨恒,王韶涵,董青,赵科渊,杨明亮. 基于注意力增强的跨模态多层融合网络的物体位姿估计[J]. 浙江大学学报(工学版), 2026, 60(9): 1872-1880.
[2] 侯越,谢金龙,张琳栋,尹杰,王甜甜. 异质性解耦与特征分层建模驱动的交通流预测[J]. 浙江大学学报(工学版), 2026, 60(6): 1185-1195.
[3] 林洪彬,吕思进,王晨阳,蔡天放,骆鹏伟. 基于残差/梯度高斯自适应采样的径向基网络[J]. 浙江大学学报(工学版), 2026, 60(5): 1119-1127.
[4] 李子祥,陆克成,蔡海兵,解伟帅,张广东. 基于触觉仿生技术的黑暗环境混凝土裂缝检测[J]. 浙江大学学报(工学版), 2026, 60(5): 915-925.
[5] 吴杰,韩贝林,张舣航,邹超,辛莉峰,黄仕平. 微调稳定扩散模型的裂缝图像数据集扩充方法[J]. 浙江大学学报(工学版), 2026, 60(5): 926-934.
[6] 武晓春,郭宁. 基于HMARU-net的隧道渗漏水轻量化检测方法[J]. 浙江大学学报(工学版), 2026, 60(3): 468-477.
[7] 林乐平,李量,欧阳宁. 基于双偏振雷达的高时空分辨率临近降水预报[J]. 浙江大学学报(工学版), 2026, 60(3): 574-584.
[8] 杨明辉,宋牧原,付大喜,郭炎伟,卢贤锥,张文聪,郑伟龙. 基于多头自注意力-Bi-LSTM模型的盾构掘进引发的土体沉降预测[J]. 浙江大学学报(工学版), 2026, 60(2): 415-424.
[9] 韦超,陈凌峰,王雨农,鲁仕博,徐元欣. 基于频域双维协同特征的AUV多传感器融合导航[J]. 浙江大学学报(工学版), 2026, 60(12): 2673-2682.
[10] 韩同春,令狐知博,汪全震,周建,余锦地. 基于地层图像与时序数据的CNN-BiLSTM路堤沉降预测模型[J]. 浙江大学学报(工学版), 2026, 60(11): 2503-2512.
[11] 曹铭,谌道宽,原鹏飞,周锦鹏,周奎. 基于深度学习的桌面级3D打印参数实时优化方法[J]. 浙江大学学报(工学版), 2026, 60(11): 2355-2363.
[12] 吴厚明,陈岭. 加速大模型训练的双向交错流水线并行方法[J]. 浙江大学学报(工学版), 2026, 60(11): 2395-2405.
[13] 胡欣,阎希玥,常娅姝,程鸿亮,肖剑,罗诗伟,马亮. 基于上下文增强与多目标语义感知的电力缺陷检测算法[J]. 浙江大学学报(工学版), 2026, 60(10): 2176-2185.
[14] 朱志航,闫云凤,齐冬莲. 基于扩散模型多模态提示的电力人员行为图像生成[J]. 浙江大学学报(工学版), 2026, 60(1): 43-51.
[15] 孙月,张兴兰. 基于双重引导的目标对抗攻击方法[J]. 浙江大学学报(工学版), 2026, 60(1): 81-89.