KiGRAS: Kinematic-Driven Generative Model for Realistic Agent Simulation

Jianbo Zhao1,2*, Jiaheng Zhuang2,3*, Qibin Zhou2*, Taiyu Ban1*, Ziyao Xu2†, Hangning Zhou2†, Junhe Wang2, Guoan Wang2, Zhiheng Li3, Bin Li1
1University of Science and Technology of China, 2Mach Drive, 3Tsinghua University

Abstract

Trajectory generation is a pivotal task in autonomous driving. Recent studies have introduced the autoregressive paradigm, leveraging the state transition model to approximate future trajectory distributions. This paradigm closely mirrors the real-world trajectory generation process and has achieved notable success. However, its potential is limited by the ineffective representation of realistic trajectories within the redundant state space. To address this limitation, we propose the Kinematic-Driven Generative Model for Realistic Agent Simulation (KiGRAS). Instead of modeling in the state space, KiGRAS factorizes the driving scene into action probability distributions at each time step, providing a compact space to represent realistic driving patterns. By establishing physical causality from actions (cause) to trajectories (effect) through the kinematic model, KiGRAS eliminates massive redundant trajectories. All states derived from actions in the cause space are constrained to be physically feasible. Furthermore, redundant trajectories representing identical action sequences are mapped to the same representation, reflecting their underlying actions. This approach significantly reduces task complexity and ensures physical feasibility. KiGRAS achieves state-of-the-art performance in Waymo's SimAgents Challenge, ranking first on the WOMD leaderboard with significantly fewer parameters than other models.

Video

BibTeX

@misc{zhao2024kigraskinematicdrivengenerativemodel,
      title={KiGRAS: Kinematic-Driven Generative Model for Realistic Agent Simulation}, 
      author={Jianbo Zhao and Jiaheng Zhuang and Qibin Zhou and Taiyu Ban and Ziyao Xu and Hangning Zhou and Junhe Wang and Guoan Wang and Zhiheng Li and Bin Li},
      year={2024},
      eprint={2407.12940},
      archivePrefix={arXiv},
      primaryClass={cs.RO},
      url={https://arxiv.org/abs/2407.12940}, 
}

Acknowledgement

We express our sincere gratitude to Ziyao Xu for his invaluable contributions to our work on KiGRAS. His central role in providing overall direction, technical verification, theoretical guidance, and writing supervision has been indispensable.

This website is adapted from Nerfies, licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.