English

Augmenting Safety-Critical Driving Scenarios while Preserving Similarity to Expert Trajectories

Machine Learning 2024-04-23 v1 Artificial Intelligence

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.

Keywords

Cite

@article{arxiv.2404.13347,
  title  = {Augmenting Safety-Critical Driving Scenarios while Preserving Similarity to Expert Trajectories},
  author = {Hamidreza Mirkhani and Behzad Khamidehi and Kasra Rezaee},
  journal= {arXiv preprint arXiv:2404.13347},
  year   = {2024}
}

Comments

Accepted to 35th IEEE Intelligent Vehicles Symposium, 2024