Transfer Learning Study of Motion Transformer-based Trajectory Predictions

Lars Ullrich,Alex McMaster,Knut Graichen
DOI: https://doi.org/10.1109/IV55156.2024.10588422
2024-08-07
Abstract:Trajectory planning in autonomous driving is highly dependent on predicting the emergent behavior of other road users. Learning-based methods are currently showing impressive results in simulation-based challenges, with transformer-based architectures technologically leading the way. Ultimately, however, predictions are needed in the real world. In addition to the shifts from simulation to the real world, many vehicle- and country-specific shifts, i.e. differences in sensor systems, fusion and perception algorithms as well as traffic rules and laws, are on the agenda. Since models that can cover all system setups and design domains at once are not yet foreseeable, model adaptation plays a central role. Therefore, a simulation-based study on transfer learning techniques is conducted on basis of a transformer-based model. Furthermore, the study aims to provide insights into possible trade-offs between computational time and performance to support effective transfers into the real world.
Machine Learning,Robotics
What problem does this paper attempt to address?
The paper attempts to address the issue of how to improve the adaptability and performance of trajectory prediction models based on the Transformer architecture in different environments through transfer learning techniques in the field of autonomous driving. Specifically: 1. **Importance of Trajectory Prediction**: The paper points out that predicting the behavior trajectories of other road users is crucial for making reliable local decisions in autonomous driving. 2. **Limitations of Existing Methods**: Traditional physical methods and rule-based methods have shortcomings in handling complex real-world situations, whereas learning-based methods can better utilize contextual information and predict multiple behavior patterns. 3. **Distribution Shift Problem**: Although learning-based methods perform well in simulated environments, they encounter various distribution shift problems when applied in the real world, such as differences in sensor systems, fusion algorithms, and traffic regulations. 4. **Transfer Learning Research**: Therefore, the paper conducts a transfer learning study using a model based on the Transformer architecture, aiming to explore how to effectively transfer knowledge from the source dataset (Waymo Open Motion Dataset) to the target dataset (CarMaker Dataset) to support the model's application in the real world. 5. **Trade-off Between Computation Time and Performance**: Additionally, the research provides insights into the trade-off between computation time and performance to facilitate the effective transition of academic challenge results to practical applications. Through these studies, the paper aims to provide more reliable and scalable methods for trajectory prediction in the field of autonomous driving.