Diff-Pre: A Diffusion Framework for Trajectory Prediction

With the rapid development of intelligent transportation, accurately predicting vehicle trajectories is crucial for ensuring road safety and enhancing traffic efficiency. This paper proposes a trajectory prediction model that integrates a diffusion model framework with trajectory features of target and neighboring vehicles, as well as driving intentions. The model uses historical trajectories of the target and adjacent vehicles as input, employs Long Short-Term Memory (LSTM) networks to extract temporal features, and dynamically captures the interaction between the target and neighboring vehicles through a multi-head attention mechanism. An intention module regulates lateral offsets, and the diffusion framework selects the most probable trajectory from various possible predictions, thereby improving the model’s ability to handle complex scenarios. Experiments conducted on real traffic data demonstrate that the proposed method outperforms several representative models in terms of Average Displacement Error (ADE) and Final Displacement Error (FDE), without sacrificing efficiency. Notably, it exhibits higher robustness and predictive accuracy in high-interaction and uncertain scenarios, such as lane changes and overtaking. To the best of our knowledge, this is the first application of the diffusion framework in vehicle trajectory prediction. This study provides an efficient solution for vehicle trajectory prediction tasks. The average ADE within 1 to 5 s reached 0.199 m, while the average FDE within 1 to 5 s reached 0.437 m.