Skip to content

TRC: ''Towards better traffic volume estimation: Jointly addressing the underdetermination and nonequilibrium problems with correlation-adaptive GNNs''.

License

Notifications You must be signed in to change notification settings

tongnie/GNN4Flow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Graph Neural Network for Traffic Volume Kriging

📝

This is the code repository for our publication ''Towards better traffic volume estimation: Jointly addressing the underdetermination and nonequilibrium problems with correlation-adaptive GNNs'' that is published on Transportation Research Part C.

Motivation

This project studies two key problems with regard to traffic volume estimation: (1) underdetermined traffic flows caused by fully undetected paths that can allow arbitrary volume values without violating the conservation law, where using local side information is insufficient to tackle, and (2) nonequilibrium traffic flows arise when traffic flows vary in density over space and time due to congestion propagation delay, which produce time-shifted volume readings to varying degrees. image

Methodology

  • Model structure: image This figure shows the holistic architecture of STCAGCN. It consists of two basic components: (1) spatial information aggregation block for alleviating underdetermined problem with learned speed-volume relationships. (2) time-asynchronous correlations extraction block for coping with nonequilibrium flows.
  • Inductive training: image We model the traffic volume estimation problem as a spatial-temporal kriging problem. GNNs are trained like a masked autoencoder, which can perform estimation at unseen locations during testing.

Run

  • We provide the Pytorch implementation of proposed STCAGCN in 'model.py' file.
  • A pretrained model is provided in the 'check_point' folder.
  • A JupyterNotebook demonstration of data loading, preprocessing, model configuration, training, evaluating, and visualizing is provided in 'run_STCAGCN.ipynb'.

Citation

Please cite our paper if this repository is helpful for your study.

@article{nie2023towards, title={Towards better traffic volume estimation: Jointly addressing the underdetermination and nonequilibrium problems with correlation-adaptive GNNs}, author={Nie, Tong and Qin, Guoyang and Wang, Yunpeng and Sun, Jian}, journal={Transportation Research Part C: Emerging Technologies}, volume={157}, pages={104402}, year={2023}, publisher={Elsevier} }

Acknowledgement

Our implementations are built on top of the IGNNK repository: https://github.com/Kaimaoge/IGNNK.

License

This project is released under the MIT license.

About

TRC: ''Towards better traffic volume estimation: Jointly addressing the underdetermination and nonequilibrium problems with correlation-adaptive GNNs''.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published