HTSA-LSTM: Leveraging Driving Habits for Enhanced Long-Term Urban Traffic Trajectory Prediction

Authors:
Open Access
Publisher website
Google Scholar
Abstract
The rapid evolution of intelligent vehicle technology has significantly advanced autonomous decision-making and driving safety. However, the challenge of predicting long-term trajectories in complex urban traffic persists, as traditional methodologies usually handle spatiotemporal attention mechanisms in isolation and are typically limited to short-term trajectory predictions. This paper proposes a Habit-based Temporal–Spatial Attention Long Short-Term Memory (HTSA-LSTM) network, a novel framework that integrates a dual spatiotemporal attention mechanism to capture dynamic dependencies across time and space, coupled with a driving style analysis module. The driving style analysis module employs Sparse Inverse Covariance Clustering and Spectral Clustering (SICC-SC) to extract driving primitives and cluster trajectory data, thereby revealing diverse driving behavior patterns without relying on predefined labels. By segmenting real-world driving data into fundamental behavioral units that reflect individual driving preferences, this approach enhances the model’s adaptability. These behavioral units, in conjunction with the spatiotemporal attention outputs, serve as inputs to the model, ultimately improving prediction accuracy and robustness in multi-vehicle scenarios. The model was evaluated by using the NGSIM dataset and real driving data from Wuhan, China. In comparison to benchmark models, HTSA-LSTM achieved a 20.72% reduction in the root mean square error (RMSE) and a 24.98% reduction in the negative log likelihood (NLL) for 5 s predictions of long-term trajectories. Furthermore, HTSA-LSTM achieved R2 values exceeding 97.9% for 5 s predictions on highways and expressways and over 92.7% for 3 s predictions on urban roads, highlighting its excellent performance in long-term trajectory prediction and adaptability across diverse driving conditions.
Funding Information
  • National Key Research and Development Project (2021YFC3001502)

This publication has 27 references indexed in Scilit: