# Heterogeneous Graph Sequence Neural Networks for Dynamic Traffic Assignment

arXiv: https://arxiv.org/abs/2408.04131 · HTML: https://mlsanity.com/paper/2408.04131/

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 6 of its 27 references that are in the map.
- First posted on arXiv 7 Aug 2024.

## Builds on (most cited first, 6 of 6)

- [Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting](https://mlsanity.com/paper/1709.04875/) (2017)
- [Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting](https://mlsanity.com/paper/2106.12931/) (2021)
- [GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal Graphs](https://mlsanity.com/paper/1803.07294/) (2018)
- [Graph Transformer for Graph-to-Sequence Learning](https://mlsanity.com/paper/1911.07470/) (2019)
- [End-to-End Heterogeneous Graph Neural Networks for Traffic Assignment](https://mlsanity.com/paper/2310.13193/) (2023)
- [Graph Neural Network Surrogate for Seismic Reliability Analysis of Highway Bridge Systems](https://mlsanity.com/paper/2210.06404/) (2022)

## Built on by (most cited first, 2 of 2)

- [Multi-Class Traffic Assignment using Multi-View Heterogeneous Graph Attention Networks](https://mlsanity.com/paper/2501.09117/) (2025)
- [Scalable and Reliable Multi-agent Reinforcement Learning for Traffic Assignment](https://mlsanity.com/paper/2506.17029/) (2025)

## Abstract (arXiv)

Traffic assignment and traffic flow prediction provide critical insights for urban planning, traffic management, and the development of intelligent transportation systems. An efficient model for calculating traffic flows over the entire transportation network could provide a more detailed and realistic understanding of traffic dynamics. However, existing traffic prediction approaches, such as those utilizing graph neural networks, are typically limited to locations where sensors are deployed and cannot predict traffic flows beyond sensor locations. To alleviate this limitation, inspired by fundamental relationship that exists between link flows and the origin-destination (OD) travel demands, we proposed the Heterogeneous Spatio-Temporal Graph Sequence Network (HSTGSN). HSTGSN exploits dependency between origin and destination nodes, even when it is long-range, and learns implicit vehicle route choices under different origin-destination demands. This model is based on a heterogeneous graph which consists of road links, OD links (virtual links connecting origins and destinations) and a spatio-temporal graph encoder-decoder that captures the spatio-temporal relationship between OD demands and flow distribution. We will show how the graph encoder-decoder is able to recover the incomplete information in the OD demand, by using node embedding from the graph decoder to predict the temporal changes in flow distribution. Using extensive experimental studies on real-world networks with complete/incomplete OD demands, we demonstrate that our method can not only capture the implicit spatio-temporal relationship between link traffic flows and OD demands but also achieve accurate prediction performance and generalization capability.