Bridging the Gap: Improving Domain Generalization in Trajectory Prediction

In recent years, there has been a rapid rise in interest in trajectory prediction in the field of autonomous driving. However, the domain generalization of current works is often neglected, and their performance tends to degrade when transferred to a different dataset or scenario. In this article, we present a solution to this problem that accounts for the realistic conditions necessary for autonomous driving. Specifically, we identify velocity and environment as possible causes for the decline in domain generalization. Then, we propose incorporating a module for velocity refinement as a solution to the velocity issue. As a response to the environmental issue, we propose both self-distillation and an environment-specific loss. Our new model is named Lane Transformer++, with one plus representing velocity issues and the other representing environmental concerns. Comprehensive evaluations on both the Argoverse and INTERACTION datasets demonstrate that the proposed method can significantly enhance the performance of prediction.