arXiv · 2405.13152
Interpretable Interaction Modeling for Trajectory Prediction via Agent Selection and Physical Coefficient
Abstract
A thorough understanding of the interaction between the target agent and surrounding agents is a prerequisite for accurate trajectory prediction. Although many methods have been explored, they assign correlation coefficients to surrounding agents in a purely learning-based manner. In this study, we present ASPILin, which manually selects interacting agents and replaces the attention scores in Transformer with a newly computed physical correlation coefficient, enhancing the interpretability of interaction modeling. Surprisingly, these simple modifications can significantly improve prediction performance and substantially reduce computational costs. We intentionally simplified our model in other aspects, such as map encoding. Remarkably, experiments conducted on the INTERACTION, highD, and CitySim datasets demonstrate that our method is efficient and straightforward, outperforming other state-of-the-art methods.
Explore related subjects
Keep this discovery
Shiji Huang, Lei Ye, Min Chen, Wenhai Luo, Dihong Wang, Chenqi Xu, Deyuan Liang. 2024-05-21. Interpretable Interaction Modeling for Trajectory Prediction via Agent Selection and Physical Coefficient. https://arxiv.org/abs/2405.13152
Cite the original work for its findings. Save a collection to share your selection of sources.