GNN-RM: A Trajectory Completion Algorithm Based on Graph Neural Networks and Regeneration Modules

Jiyuan Zhang,Zhenjiang Zhang,Lin Hui
DOI: https://doi.org/10.1016/j.ijcce.2024.07.001
2024-07-16
International Journal of Cognitive Computing in Engineering
Abstract:Highlights • Introduced a feature extraction method for trajectory completion, capable of preserving adequate vehicle and environmental features with minimal data. • Employed a subgraph construction approach based on Long Short-Term Memory (LSTM) to represent vehicles and environments as interpretable subgraphs. • Leveraged self-attention mechanisms to efficiently extract spatial interactions between vehicles and environments. • Proposed a regeneration module that integrates temporal sequence features with spatial interaction features to complete trajectories. Data about vehicle trajectories assumes a crucial role in applications such as intelligent connected vehicles. However, missing values resulting from sensors and other factors frequently affect real trajectory data. Currently, it is challenging to utilize trajectory completion methods to generate accurate real-time results at an affordable computing cost. This paper proposes GNN-RM, a trajectory completion algorithm based on graph neural networks and regeneration modules, encompassing feature extraction, subgraph construction, spatial interaction graph, and trajectory regeneration modules. The feature extraction algorithm extracts influential data as feature vectors based on certain conditions and organizes these feature vectors into different subgraphs according to categories. The spatial interaction graph constructed through graph neural networks extracts spatial interaction features between vehicles and the environment, while the regeneration modules constructed by multi-head attention mechanisms extract temporal features of vehicles, thereby completing the missing trajectories. The experimental results demonstrate that GNN-RM can achieve higher trajectory completion accuracy with fewer input parameters than multiple baseline models.
What problem does this paper attempt to address?