← Search

Zhejun Zhang

6 accepted papers

2025

Closed-Loop Supervised Fine-Tuning of Tokenized Traffic Models

CVPR 2025poster

Traffic simulation aims to learn a policy for traffic agents that, when unrolled in closed-loop, faithfully recovers the joint distribution of trajectories observed in the real world. Inspired by large language models, tokenized multi-agent policies have recently become the state-of-the-art in traff…

2025

Unraveling the Effects of Synthetic Data on End-to-End Autonomous Driving

ICCV 2025poster

End-to-end (E2E) autonomous driving (AD) models require diverse, high-quality data to perform well across various driving scenarios. However, collecting large-scale real-world data is expensive and time-consuming, making high-fidelity synthetic data essential for enhancing data diversity and model r…

2023

A Multiplicative Value Function for Safe and Efficient Reinforcement Learning

IROS 2023poster

An emerging field of sequential decision problems is safe Reinforcement Learning (RL), where the objective is to maximize the reward while obeying safety constraints. Being able to handle constraints is essential for deploying RL agents in real-world environments, where constraint violations can har…

Cited by 1SourcecodeScholar
2023

Real-Time Motion Prediction via Heterogeneous Polyline Transformer with Relative Pose Encoding

NeurIPS 2023poster

The real-world deployment of an autonomous driving system requires its components to run on-board and in real-time, including the motion prediction module that predicts the future trajectories of surrounding traffic participants. Existing agent-centric methods have demonstrated outstanding performan…

2023

TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Prediction

ICRA 2023poster

Data-driven simulation has become a favorable way to train and test autonomous driving algorithms. The idea of replacing the actual environment with a learned simulator has also been explored in model-based reinforcement learning in the context of world models. In this work, we show data-driven traf…

Cited by 48SourceScholar
2021

End-to-End Urban Driving by Imitating a Reinforcement Learning Coach

ICCV 2021poster

End-to-end approaches to autonomous driving commonly rely on expert demonstrations. Although humans are good drivers, they are not good coaches for end-to-end algorithms that demand dense on-policy supervision. On the contrary, automated experts that leverage privileged information can efficiently g…

Cited by 232PDFcodeScholar