TerraZero:用于零演示大规模自博弈的程序化驾驶模拟

TerraZero: Procedural Driving Simulation for Zero-Demonstration Self-Play at Scale

精选理由

TerraZero 用程序化模拟+零演示强化学习搞定了自动驾驶长尾场景,跑得巨快,还拿了 InterPlan 和 Waymo 的 SOTA,你不想看看怎么做到的?

AI 摘要

TerraZero 是一个程序化驾驶模拟器与自博弈训练框架,在单 GPU 上达到每秒 130 万 agent-step,远超同类模拟器。它基于真实地图几何随机生成交通场景,无需人类演示,仅靠强化学习从头训练策略。该策略在 InterPlan 长尾基准上首次超越所有学习型规划器,并在 Waymo 开放模拟智能体任务中优于其他无演示方法。此外,在常规驾驶基准 val14 上,TerraZero 取得最低碰撞率和最佳碰撞时间分数。

原文 · arXiv cs.LG

TerraZero: Procedural Driving Simulation for Zero-Demonstration Self-Play at Scale

Training robust autonomous driving agents requires a simulator that is fast enough for reinforcement learning at scale, realistic enough to ground behavior in real-world map structure, and diverse enough to cover the safety-critical long tail that logged data rarely contains. We present TerraZero, a procedural driving simulator and self-play training stack. A configurable C engine runs simulation on the CPU and policy inference on the GPU over a zero-copy path, sustaining 1.3M agent-steps per second on a single server-grade GPU, far faster than existing object-level simulators, while keeping fidelity lighter single-agent systems omit: heterogeneous agents, multiple dynamics models, and full traffic-rule enforcement. TerraZero treats logged data only as a source of real-world map geometry, populating each map with randomized rule-based road users and signal controllers and randomizing agent dynamics, rewards, and sizes per episode, so a map yields an unbounded set of scenarios. Every reported policy trains from scratch by reinforcement learning alone on a compute-efficient self-play recipe across GPUs, with zero human demonstrations and no fallback planner at inference. Policies generalize zero-shot across cities and datasets, including emergent left-hand-traffic driving without explicit supervision. As an ego policy, TerraZero is the first fully learned policy to top the InterPlan long-tail benchmark, ahead of larger learned planners; on routine-driving val14 it ranks among the best approaches and is the safest, posting the best collision and time-to-collision scores. On Waymo Open Sim Agents realism the same recipe outperforms other demonstration-free methods and is competitive with the strongest reference-anchored self-play method. One stack serves both roles: driving policies across dynamics for cars and trucks, and sim agents that jointly control vehicles, pedestrians, and cyclists.