Diff-Pre: A Diffusion Framework for Trajectory Prediction

doi:10.3390-s25154603

Diff-Pre: A Diffusion Framework for Trajectory Prediction

2025-08-22

0

PRE

AI

DOI：10.3390-s25154603

Original

Original request for help

Save

Abstract

En 中文

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.

Keywords:

trajectory prediction

diffusion model

LSTM

multi-head attention

driving intention

Diff-Pre: A Diffusion Framework for Trajectory Prediction

Abstract

Journal

Organization

Cited Papers