Seed: Bridging Sequence and Diffusion Models for Road Trajectory Generation
Track: Systems and infrastructure for Web, mobile, and WoT
Keywords: Trajectory, Data Synthesis, Transformer, Diffusion Model
TL;DR: A novel framework that bridges sequence and diffusion models for road trajectory generation, simultaneously addressing consistency, regularity, and diversity.
Abstract: Road trajectory generation creates synthetic yet realistic trajectories to tackle data collection costs and privacy concerns. Existing methods generate a trajectory either segment-by-segment using sequence models or holistically in one step using diffusion models. Sequence-based models have good regularity and consistency (i.e., resemble the input trajectories) but lack diversity, while diffusion-based models enhance diversity but sacrifice regularity and consistency. To combine the merits of existing methods, we propose $\textit{Seed}$, by bridging $\underline{\textit{se}}$quenc$\underline{\textit{e}}$ and $\underline{\textit{d}}$iffusion models for trajectory generation. In particular, Seed adopts a \textit{conditional diffusion structure}, where a Transformer models the movement of each trajectory along the road segments, and conditioned on the Transformer's output, a diffusion model recovers the next road segment from random noise. The rationale is that the Transformer captures sequential movement patterns for regularity and consistency, while the diffusion model introduces diversity by recovering from noise. To train Seed, we adopt Node2vec to learn embeddings for the road segments to prepare model inputs, supervise learning using the task of trajectory reconstruction, and design a curriculum learning strategy to accelerate convergence. We compare Seed with 8 state-of-the-art trajectory generation methods on 3 datasets, and the results show that Seed improves the best-performing baseline by over 50\%.
Submission Number: 976
Loading