Flow prediction via adaptive dynamic graph with spatio-temporal correlations

被引:0
|
作者
Zhang, Hui [1 ,2 ,3 ]
Ding, Kun [1 ,2 ]
Xie, Jietao [3 ,4 ]
Xiao, Weidong [3 ]
Xie, Yuxiang [3 ]
机构
[1] Natl Univ Def Technol, Sixty Res Inst 3, Nanjing 210007, Peoples R China
[2] Natl Univ Def Technol, Lab Big Data & Decis, Changsha 410073, Peoples R China
[3] Natl Univ Def Technol, Coll Syst Engn, Changsha 410073, Peoples R China
[4] Baicheng Ordnance Test Ctr China, Jinzhou 121000, Peoples R China
关键词
Adaptive dynamic graph; Spatial relationship; Long-range correlation; Node adaptive learning; Graph adaptive generation;
D O I
10.1016/j.eswa.2024.125474
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Flow prediction has shown its significance in optimizing and smoothing resource allocation. However, it still faces challenges due to inherent temporal variability and dynamic spatial correlations of flow data. Although many methods have been introduced to solve the problem, they often rely on predefined graphs to capture shared patterns, which is likely to overlook the critical spatiotemporal dependencies that are necessary for mid-term and long-term prediction tasks. To address the issue, a novel flow prediction model, termed Dynamic Graph Multi-Step Prediction via Long Short-Term Memory Network and Adaptive Learning (DMLSA), is proposed. DMLSA employs Node Adaptive Learning (NAL) to learn node-specific patterns without relying on predefined static adjacency matrices. Moreover, DMLSA utilizes Graph Adaptive Generation (GAG) to extract dynamic features from node attributes and automatically infer interdependencies among flow sequences, thus enhancing the ability to predict complex flow dynamics. Additionally, DMLSA integrates multi-head self- attention mechanisms with gated recurrent units, effectively capturing both long-term and short-term node dependencies, therefore allowing for more precise identification of spatiotemporal patterns at the individual node level. Extensive experiments on five datasets demonstrate that DMLSA outperformed existing baseline models in terms of accuracy and stability for multi-step prediction tasks.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] Spatio-Temporal Dynamic Graph Relation Learning for Urban Metro Flow Prediction
    Xie, Peng
    Ma, Minbo
    Li, Tianrui
    Ji, Shenggong
    Du, Shengdong
    Yu, Zeng
    Zhang, Junbo
    arXiv, 2022,
  • [2] Spatio-Temporal Dynamic Graph Relation Learning for Urban Metro Flow Prediction
    Xie, Peng
    Ma, Minbo
    Li, Tianrui
    Ji, Shenggong
    Du, Shengdong
    Yu, Zeng
    Zhang, Junbo
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (10) : 9973 - 9984
  • [3] Dynamic Spatio-Temporal Graph-Based CNNs for Traffic Flow Prediction
    Chen, Ken
    Chen, Fei
    Lai, Baisheng
    Jin, Zhongming
    Liu, Yong
    Li, Kai
    Wei, Long
    Wang, Pengfei
    Tang, Yandong
    Huang, Jianqiang
    Hua, Xian-Sheng
    IEEE ACCESS, 2020, 8 : 185136 - 185145
  • [4] Dynamic traffic correlations based spatio-temporal graph convolutional network for urban traffic prediction
    Xu, Yuanbo
    Cai, Xiao
    Wang, En
    Liu, Wenbin
    Yang, Yongjian
    Yang, Funing
    INFORMATION SCIENCES, 2023, 621 : 580 - 595
  • [5] Dynamic Adaptive Spatio-Temporal Graph Convolution for fMRI Modelling
    El-Gazzar, Ahmed
    Thomas, Rajat Mani
    van Wingen, Guido
    MACHINE LEARNING IN CLINICAL NEUROIMAGING, 2021, 13001 : 125 - 134
  • [6] A Spatio-Temporal Traffic Flow Prediction Method Based on Dynamic Graph Convolution Network
    Yang, Guoliang
    Yu, Huasheng
    Xi, Hao
    2022 34TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2022, : 5364 - 5369
  • [7] Dynamic Spatio-Temporal Adaptive Graph Convolutional Recurrent Networks for Vacant Parking Space Prediction
    Gao, Liangpeng
    Fan, Wenli
    Jian, Wenliang
    APPLIED SCIENCES-BASEL, 2024, 14 (13):
  • [8] A spatio-temporal grammar graph attention network with adaptive edge information for traffic flow prediction
    Zhang, Zhao
    Jiao, Xiaohong
    APPLIED INTELLIGENCE, 2023, 53 (23) : 28787 - 28803
  • [9] A spatio-temporal grammar graph attention network with adaptive edge information for traffic flow prediction
    Zhao Zhang
    Xiaohong Jiao
    Applied Intelligence, 2023, 53 : 28787 - 28803
  • [10] Spatio-Temporal Contrastive Learning-Based Adaptive Graph Augmentation for Traffic Flow Prediction
    Zhang, Dingkai
    Wang, Pengfei
    Ding, Lu
    Wang, Xiaoling
    He, Jifeng
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2025, 26 (01) : 1304 - 1318