Probabilistic trajectory prediction of heterogeneous traffic agents based on layered spatio-temporal graph

In order to safely and comfortably navigate in the complex urban traffic, it is necessary to make multi-modal predictions of autonomous vehicles for the next trajectory of various traffic participants, with the continuous movement trend and inertia of the surrounding traffic agents taken into account. At present, most trajectory prediction methods focus on prediction on future behavior of traffic agents but with limited, consideration of the response of traffic agents to the future behavior of the ego-agent. Moreover, it can only predict the trajectory of single-type agents, which make it impossible to learn interaction in a complex environment between traffic agents. In this paper, we proposed a graph-based heterogeneous traffic agents trajectory prediction model LSTGHP, which consists of the following three parts: (1) layered spatio-temporal graph module; (2) ego-agent motion module; (3) trajectory prediction module, which can realize multi-modal prediction of future trajectories of traffic agents with different semantic categories in the scene. To evaluate its performance, we collected trajectory datasets of heterogeneous traffic agents in a time-varying, highly dynamic urban intersection environment, where vehicles, bicycles, and pedestrians interacted with each other in the scene. It can be drawn from experimental results that our model can improve its prediction accuracy while interacting at a close range. Compared with the previous prediction methods, the model has less prediction error in the trajectory prediction of heterogeneous traffic agents.