Learning to Drive via Asymmetric Self-Play

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Chris, Biswas, Sourav, Wong, Kelvin, Fallah, Kion, Zhang, Lunjun, Chen, Dian, Casas, Sergio, Urtasun, Raquel
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913520693542912
author Zhang, Chris
Biswas, Sourav
Wong, Kelvin
Fallah, Kion
Zhang, Lunjun
Chen, Dian
Casas, Sergio
Urtasun, Raquel
author_facet Zhang, Chris
Biswas, Sourav
Wong, Kelvin
Fallah, Kion
Zhang, Lunjun
Chen, Dian
Casas, Sergio
Urtasun, Raquel
contents Large-scale data is crucial for learning realistic and capable driving policies. However, it can be impractical to rely on scaling datasets with real data alone. The majority of driving data is uninteresting, and deliberately collecting new long-tail scenarios is expensive and unsafe. We propose asymmetric self-play to scale beyond real data with additional challenging, solvable, and realistic synthetic scenarios. Our approach pairs a teacher that learns to generate scenarios it can solve but the student cannot, with a student that learns to solve them. When applied to traffic simulation, we learn realistic policies with significantly fewer collisions in both nominal and long-tail scenarios. Our policies further zero-shot transfer to generate training data for end-to-end autonomy, significantly outperforming state-of-the-art adversarial approaches, or using real data alone. For more information, visit https://waabi.ai/selfplay .
format Preprint
id arxiv_https___arxiv_org_abs_2409_18218
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning to Drive via Asymmetric Self-Play
Zhang, Chris
Biswas, Sourav
Wong, Kelvin
Fallah, Kion
Zhang, Lunjun
Chen, Dian
Casas, Sergio
Urtasun, Raquel
Robotics
Computer Vision and Pattern Recognition
Machine Learning
Large-scale data is crucial for learning realistic and capable driving policies. However, it can be impractical to rely on scaling datasets with real data alone. The majority of driving data is uninteresting, and deliberately collecting new long-tail scenarios is expensive and unsafe. We propose asymmetric self-play to scale beyond real data with additional challenging, solvable, and realistic synthetic scenarios. Our approach pairs a teacher that learns to generate scenarios it can solve but the student cannot, with a student that learns to solve them. When applied to traffic simulation, we learn realistic policies with significantly fewer collisions in both nominal and long-tail scenarios. Our policies further zero-shot transfer to generate training data for end-to-end autonomy, significantly outperforming state-of-the-art adversarial approaches, or using real data alone. For more information, visit https://waabi.ai/selfplay .
title Learning to Drive via Asymmetric Self-Play
topic Robotics
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2409.18218