Urban Vehicle Trajectory Generation Based on Generative Adversarial Imitation Learning

Min Wang,Jianqun Cui,Yew Wee Wong,Yanan Chang,Libing Wu,Jiong Jin
DOI: https://doi.org/10.1109/tvt.2024.3437412
IF: 6.8
2024-01-01
IEEE Transactions on Vehicular Technology
Abstract:With the rapid development of smart cities, the collection of vehicle trajectory data through sensors has increased significantly. While many studies have utilized calibrated physical car-following models (CFM) and machine learning techniques for trajectory prediction, these approaches often falter in complex, dynamic traffic scenarios. Addressing this gap, this paper introduces PS-TrajGAIL, a generative adversarial imitation learning framework tailored for urban vehicle trajectory generation. Contrary to conventional discriminative models, PS-TrajGAIL employs a generative model to capture the inherent distribution of urban vehicle trajectories. This framework models the tasks of trajectory generation as a partially observable Markov decision process based on imitation learning. PS-TrajGAIL's architecture features a generator, which simulates vehicle behavior to produce synthetic trajectories, and a discriminator that distinguishes between authentic and generated trajectories. In addition, the driving policy within the generator is fine-tuned using the Trust Region Policy Optimization (TRPO) algorithm, ensuring safety in vehicle driving. Experimental evaluations on both synthetic and real-world datasets highlight that PS-TrajGAIL notably surpasses existing baselines and state-of-the-art approaches in trajectory generation
What problem does this paper attempt to address?