arXiv · 2404.13347
Augmenting Safety-Critical Driving Scenarios while Preserving Similarity to Expert Trajectories
Abstract
Trajectory augmentation serves as a means to mitigate distributional shift in imitation learning. However, imitating trajectories that inadequately represent the original expert data can result in undesirable behaviors, particularly in safety-critical scenarios. We propose a trajectory augmentation method designed to maintain similarity with expert trajectory data. To accomplish this, we first cluster trajectories to identify minority yet safety-critical groups. Then, we combine the trajectories within the same cluster through geometrical transformation to create new trajectories. These trajectories are then added to the training dataset, provided that they meet our specified safety-related criteria. Our experiments exhibit that training an imitation learning model using these augmented trajectories can significantly improve closed-loop performance.
Explore related subjects
Keep this discovery
Hamidreza Mirkhani, Behzad Khamidehi, Kasra Rezaee. 2024-04-20. Augmenting Safety-Critical Driving Scenarios while Preserving Similarity to Expert Trajectories. https://arxiv.org/abs/2404.13347
Cite the original work for its findings. Save a collection to share your selection of sources.