Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

RAP: 3D Rasterization Augmented End-to-End Planning

About

Imitation learning for end-to-end driving trains policies only on expert demonstrations. Once deployed in a closed loop, such policies lack recovery data: small mistakes cannot be corrected and quickly compound into failures. A promising direction is to generate alternative viewpoints and trajectories beyond the logged path. Prior work explores photorealistic digital twins via neural rendering or game engines, but these methods are prohibitively slow and costly, and thus mainly used for evaluation. In this work, we argue that photorealism is unnecessary for training end-to-end planners. What matters is semantic fidelity and scalability: driving depends on geometry and dynamics, not textures or lighting. Motivated by this, we propose 3D Rasterization, which replaces costly rendering with lightweight rasterization of annotated primitives, enabling augmentations such as counterfactual recovery maneuvers and cross-agent view synthesis. To transfer these synthetic views effectively to real-world deployment, we introduce a Raster-to-Real feature-space alignment that bridges the sim-to-real gap. Together, these components form Rasterization Augmented Planning (RAP), a scalable data augmentation pipeline for planning. RAP achieves state-of-the-art closed-loop robustness and long-tail generalization, ranking first on four major benchmarks: NAVSIM v1/v2, Waymo Open Dataset Vision-based E2E Driving, and Bench2Drive. Our results show that lightweight rasterization with feature alignment suffices to scale E2E training, offering a practical alternative to photorealistic rendering. Project page: https://alan-lanfeng.github.io/RAP/.

Lan Feng, Yang Gao, Eloi Zablocki, Quanyi Li, Wuyang Li, Sichao Liu, Matthieu Cord, Alexandre Alahi• 2025

Related benchmarks

TaskDatasetResultRank
Closed-loop PlanningBench2Drive
Driving Score66.42
194
Autonomous DrivingNAVSIM v1 (test)
NC99.1
147
Autonomous Driving PlanningNAVSIM navhard v2
NC97.1
108
Autonomous Driving Trajectory PlanningNAVSIM navhard-two-stage v2 (test)
Stage 1 NC97.1
48
Trajectory PlanningNAVSIM v2 (navhard)
NC Rate97.1
43
End-to-end DrivingWaymo E2E Driving Challenge (Leaderboard)
RFS (Overall)8.043
28
PlanningNAVSIM v1
PDMS93.8
23
Autonomous Driving Policy EvaluationNAVSIM navhard v2
NC97.1
20
Closed-loop Autonomous DrivingBench2Drive official (test)
Driving Score66.42
17
End-to-end Autonomous DrivingBridgeSim NavHard
DS52.08
15
Showing 10 of 22 rows

Other info

Follow for update