CtRL-Sim: Reactive and Controllable Driving Agents with Offline Reinforcement Learning
Abstract
Evaluating autonomous vehicle stacks (AVs) in simulation typically involves replaying driving logs from real-world recorded traffic. However, agents replayed from offline data are not reactive and hard to intuitively control. Existing approaches address these challenges by proposing methods that rely on heuristics or generative models of real-world data but these approaches either lack realism or necessitate costly iterative sampling procedures to control the generated behaviours. In this work, we take an alternative approach and propose CtRL-Sim, a method that leverages return-conditioned offline reinforcement learning to efficiently generate reactive and controllable traffic agents. Specifically, we process real-world driving data through a physics-enhanced Nocturne simulator to generate a diverse offline reinforcement learning dataset, annotated with various reward terms. We then train a return-conditioned multi-agent behaviour model that allows for fine-grained manipulation of agent behaviours by modifying the desired returns for the various reward components. This capability enables the generation of a wide range of driving behaviours beyond the scope of the initial dataset, including adversarial behaviours. We demonstrate that CtRL-Sim can generate diverse and realistic safety-critical scenarios while providing fine-grained control over agent behaviours.
Cite
Text
Rowe et al. "CtRL-Sim: Reactive and Controllable Driving Agents with Offline Reinforcement Learning." Proceedings of The 8th Conference on Robot Learning, 2024.Markdown
[Rowe et al. "CtRL-Sim: Reactive and Controllable Driving Agents with Offline Reinforcement Learning." Proceedings of The 8th Conference on Robot Learning, 2024.](https://mlanthology.org/corl/2024/rowe2024corl-ctrlsim/)BibTeX
@inproceedings{rowe2024corl-ctrlsim,
title = {{CtRL-Sim: Reactive and Controllable Driving Agents with Offline Reinforcement Learning}},
author = {Rowe, Luke and Girgis, Roger and Gosselin, Anthony and Carrez, Bruno and Golemo, Florian and Heide, Felix and Paull, Liam and Pal, Christopher},
booktitle = {Proceedings of The 8th Conference on Robot Learning},
year = {2024},
pages = {3600-3621},
volume = {270},
url = {https://mlanthology.org/corl/2024/rowe2024corl-ctrlsim/}
}