Journal of Computer Applications ›› 2026, Vol. 46 ›› Issue (5): 1507-1517.DOI: 10.11772/j.issn.1001-9081.2025050570

• Data science and technology • Previous Articles    

Time-interdependency-aware dynamic Bayesian network for traffic prediction

Huijie GUO1, Tianfeng DOU1, Zhenlin ZHANG1, Kaiyuan QI2, Dong WU2, Zhijian QU1, Zhao LI1, Chongguang REN1()   

  1. 1.School of Computer Science and Technology,Shandong University of Technology,Zibo Shandong 255049,China
    2.Inspur (Jinan) Data Technology Company Limited,Jinan Shandong 250101,China
  • Received:2025-05-27 Revised:2025-08-09 Accepted:2025-08-18 Online:2025-08-20 Published:2026-05-10
  • Contact: Chongguang REN
  • About author:GUO Huijie, born in 2000, M. S. candidate. Her research interests include artificial intelligence and intelligent systems.
    DOU Tianfeng, born in 2000, M. S. candidate. Her research interests include artificial intelligence and intelligent systems.
    ZHANG Zhenlin, born in 1999, M. S. candidate. His research interests include artificial intelligence and intelligent systems.
    QI Kaiyuan, born in 1984, Ph. D., senior engineer. His research interests include cloud computing, big data.
    WU Dong, born in 1984, engineer. His research interests include cloud computing.
    QU Zhijian, born in 1980, Ph.D., associate professor. His research interests include data analysis, intelligent computing, evolutionary algorithms.
    LI Zhao, born in 1983, Ph. D., associate professor. His research interests include artificial intelligence, edge computing.
  • Supported by:
    National Key Research and Development Program of China(2022YFE0107300);Outstanding Youth Innovation Teams in Higher Education of Shandong Province(2019KJN048)

基于时间依赖建模的动态贝叶斯网络交通预测

郭慧洁1, 窦天凤1, 张振琳1, 亓开元2, 吴栋2, 曲志坚1, 李钊1, 任崇广1()   

  1. 1.山东理工大学 计算机科学与技术学院,山东 淄博 255049
    2.济南浪潮数据技术有限公司,济南 250101
  • 通讯作者: 任崇广
  • 作者简介:郭慧洁(2000—),女,山东淄博人,硕士研究生,主要研究方向:人工智能与智能系统
    窦天凤(2000—),女,山东淄博人,硕士研究生,主要研究方向:人工智能与智能系统
    张振琳(1999—),男,山东聊城人,硕士研究生,主要研究方向:人工智能与智能系统
    亓开元(1984—),男,山东莱芜人,高级工程师,博士,主要研究方向:云计算、大数据
    吴栋(1984—),男,山东聊城人,工程师,主要研究方向:云计算
    曲志坚(1980—),男,山东青岛人,副教授,博士,主要研究方向:数据分析、智能计算、进化算法
    李钊(1983—),男,山东淄博人,副教授,博士,主要研究方向:人工智能、边缘计算
  • 基金资助:
    国家重点研发计划项目(2022YFE0107300);山东省高等学校优秀青年创新团队项目(2019KJN048)

Abstract:

Accurate traffic forecasting not only improves the efficiency and safety of the traffic system, but also promotes the sustainable social and economic development. Although a large number of studies have been devoted to modeling spatiotemporal correlation, existing methods still have significant limitations: most models tend to collectively predict the traffic flow of all regions in all time periods, ignoring spatio-temporal heterogeneity, especially the impact of the traffic status of the current region on the future traffic status of related regions. To address this problem, a Time-Interdependency-aware Dynamic Bayesian Network for traffic prediction (TIDBN) method was proposed. Using pre-trained modules, TIDBN employed a time-varying dynamic Bayesian network to capture the complex temporal relationships in time-series data arising from simultaneous and lagged effects. To further improve its ability to capture spatio-temporal correlation, a spatio-temporal attention mechanism was introduced for in-depth analysis. Subsequently, a Graph Convolutional Network (GCN) was utilized to model the spatio-temporal topological structure, generating more accurate traffic predictions. The experimental results show that TIDBN performs excellently on two real traffic prediction tasks, especially for 1-hour prediction. On the PeMS-BAY dataset, the Mean Absolute Error (MAE) of TIDBN is 4% lower than that of the second-best baseline method.

Key words: traffic prediction, Bayesian network, Graph Convolutional Network (GCN), attention mechanism, deep learning

摘要:

精准的交通预测不仅能提升交通系统的效率与安全性,还能促进社会和经济的可持续发展。尽管已有大量研究致力于建模时空相关性,但现有方法仍然存在明显不足:大多数模型倾向于集体预测所有区域在所有时间段的交通流量,忽略了时空异质性,特别是当前区域的交通状态对相关区域未来交通状态的影响。为了解决这一问题,提出基于时间依赖建模的动态贝叶斯网络交通预测(TIDBN)方法。TIDBN通过预训练模块,利用时变动态贝叶斯网络捕捉时间序列数据中由于同时和滞后影响而产生的复杂时序关系。为了进一步提升对时空相关性的捕捉能力,引入时空注意力机制进行深入分析;随后,采用图卷积网络(GCN)对时空拓扑结构进行建模,以生成更准确的交通预测。实验结果表明,TIDBN在2个真实的交通预测任务中均表现优异,尤其在1 h预测中,它在数据集PeMS-BAY上的平均绝对误差(MAE)比基线次优方法降低了4%。

关键词: 交通预测, 贝叶斯网络, 图卷积网络, 注意力机制, 深度学习

CLC Number: