English

ReinDriveGen: Reinforcement Post-Training for Out-of-Distribution Driving Scene Generation

Computer Vision and Pattern Recognition 2026-04-02 v1

Abstract

We present ReinDriveGen, a framework that enables full controllability over dynamic driving scenes, allowing users to freely edit actor trajectories to simulate safety-critical corner cases such as front-vehicle collisions, drifting cars, vehicles spinning out of control, pedestrians jaywalking, and cyclists cutting across lanes. Our approach constructs a dynamic 3D point cloud scene from multi-frame LiDAR data, introduces a vehicle completion module to reconstruct full 360{\deg} geometry from partial observations, and renders the edited scene into 2D condition images that guide a video diffusion model to synthesize realistic driving videos. Since such edited scenarios inevitably fall outside the training distribution, we further propose an RL-based post-training strategy with a pairwise preference model and a pairwise reward mechanism, enabling robust quality improvement under out-of-distribution conditions without ground-truth supervision. Extensive experiments demonstrate that ReinDriveGen outperforms existing approaches on edited driving scenarios and achieves state-of-the-art results on novel ego viewpoint synthesis.

Keywords

Cite

@article{arxiv.2604.01129,
  title  = {ReinDriveGen: Reinforcement Post-Training for Out-of-Distribution Driving Scene Generation},
  author = {Hao Zhang and Lue Fan and Weikang Bian and Zehuan Wu and Lewei Lu and Zhaoxiang Zhang and Hongsheng Li},
  journal= {arXiv preprint arXiv:2604.01129},
  year   = {2026}
}

Comments

Project page: https://drive-sim.github.io/ReinDriveGen/