Search papers, labs, and topics across Lattice.
This research addresses the spatial generalization gap in Graph Neural Networks (GNNs) for the Traffic Assignment Problem by introducing a network-agnostic initialization layer called GUIDED, which embeds travel demand as a scalar attribute on auxiliary virtual links. This approach allows for seamless transferability across different urban environments and enhances the model's robustness to out-of-distribution demand patterns. Experimental results show that integrating GUIDED with a Heterogeneous Graph Attention Network (HetGAT) not only maintains state-of-the-art predictive accuracy but also achieves a 50% reduction in training time per epoch compared to traditional methods.
A novel initialization layer enables GNNs to adaptively transfer across urban environments, achieving superior performance with significantly reduced training times.
The Traffic Assignment Problem is a fundamental but computationally expensive component of transportation planning. While Graph Neural Networks have emerged as fast, data-driven surrogates, their practical deployment is severely constrained by a spatial generalization gap. Standard models rely on transductive feature initializations that tie travel demand to fixed network topologies, preventing seamless transfer to new urban environments. To overcome this structural limitation, this research proposes a network-agnostic initialization layer, termed Geometrically Unconstrained Inductive Demand EmbeDding (GUIDED). By injecting travel demand as a scalar attribute on auxiliary virtual links rather than as specific node features, this modular framework standardizes the input space regardless of network scale. Extensive experimental evaluation across multiple urban topologies demonstrates that a Heterogeneous Graph Attention Network (HetGAT) model integrated with the proposed GUIDED layer maintains state-of-the-art predictive accuracy on single-network tasks, while demonstrating superior robustness to out-of-distribution demand patterns and maintaining a distinct performance advantage over the baseline even under severe data scarcity. Notably, the proposed feature initialization enables highly parameter-efficient domain adaptation for inter-network transfer learning without artificial input homogenization, establishing a robust foundation for truly inductive models. At the same time, the optimized scatter operations of the initialization layer yield an approximate 50% reduction in training time per epoch compared to the baseline approach. Furthermore, while demonstrated on vehicular traffic, this fundamental abstraction of spatial topology provides a versatile blueprint for generalized origin-destination spatial problems, such as freight logistics and multimodal network optimization.
