arXiv · 2407.13431
Improving Out-of-Distribution Generalization of Trajectory Prediction for Autonomous Driving via Polynomial Representations
Abstract
Robustness against Out-of-Distribution (OoD) samples is a key performance indicator of a trajectory prediction model. However, the development and ranking of state-of-the-art (SotA) models are driven by their In-Distribution (ID) performance on individual competition datasets. We present an OoD testing protocol that homogenizes datasets and prediction tasks across two large-scale motion datasets. We introduce a novel prediction algorithm based on polynomial representations for agent trajectory and road geometry on both the input and output sides of the model. With a much smaller model size, training effort, and inference time, we reach near SotA performance for ID testing and significantly improve robustness in OoD testing. Within our OoD testing protocol, we further study two augmentation strategies of SotA models and their effects on model generalization. Highlighting the contrast between ID and OoD performance, we suggest adding OoD testing to the evaluation criteria of trajectory prediction models.
Explore related subjects
Keep this discovery
Yue Yao, Shengchao Yan, Daniel Goehring, Wolfram Burgard, Joerg Reichardt. 2024-07-18. Improving Out-of-Distribution Generalization of Trajectory Prediction for Autonomous Driving via Polynomial Representations. https://arxiv.org/abs/2407.13431
Cite the original work for its findings. Save a collection to share your selection of sources.