Goal-CurveNet: A pedestrian trajectory prediction network using heterogeneous graph attention goal prediction and curve fitting

Xiangchen Wang,Xin Yang,Dake Zhou
DOI: https://doi.org/10.1016/j.engappai.2024.108323
IF: 8
2024-04-13
Engineering Applications of Artificial Intelligence
Abstract:Pedestrian trajectory prediction in dynamic and strongly interactive scenes has become one of the most challenging problems in fields such as automated driving. In this paper, we propose Goal-CurveNet, a multimodal trajectory prediction network combining heterogeneous graph attention goal prediction and curve fitting. The model addresses the problems of pedestrian interaction modeling, multimodal trajectory prediction, and performance of predicted trajectories in pedestrian trajectory prediction. Goal-CurveNet can better model the historical trajectories and interaction behaviors in the scene systematically based on heterogeneous graph attention. It predicts the complete trajectories by curve fitting, which effectively improves the quality of predicted trajectories. The model architecture of "goal first and then trajectory" and targeted training paradigm also enhance the final performance. Through detailed training and testing on ETH & UCY datasets, we validate the effectiveness of each contribution of Goal-CurveNet. Compared to many state-of-the-art models, Goal-CurveNet achieves performance improvements in key metrics and effective prediction of pedestrian trajectories.
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary
What problem does this paper attempt to address?