An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms
Accurate trajectory prediction for multiple vehicles in complex social interaction environments is essential for ensuring the safety of autonomous vehicles and improving the quality of their planning and control. The social interactions between vehicles significantly influence their future trajector...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2024-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10368009/ |
| _version_ | 1797361407850708992 |
|---|---|
| author | Senyao Qiao Fei Gao Jianghang Wu Rui Zhao |
| author_facet | Senyao Qiao Fei Gao Jianghang Wu Rui Zhao |
| author_sort | Senyao Qiao |
| collection | DOAJ |
| description | Accurate trajectory prediction for multiple vehicles in complex social interaction environments is essential for ensuring the safety of autonomous vehicles and improving the quality of their planning and control. The social interactions between vehicles significantly influence their future trajectories. However, traditional trajectory prediction models based on Recurrent Neural Networks (RNN) or Convolutional Neural Networks (CNN) often overlook or simplify these interactions. Although these models may exhibit high performance in short-term predictions, they fail to achieve high prediction accuracy in scenarios with long-term dynamic interactions. To address this limitation, we propose a Social-Attention Long Short-Term Memory (LSTM) model which predicts the future trajectories of neighboring vehicles and achieves increased accuracy. Our proposed model employs a Social-Pooling layer to effectively capture cooperative behaviors and mutual influences between vehicles. Additionally, we incorporate a self-attention mechanism to weight the inputs and outputs of the Social-Pooling layer, which is significant for assessing the influence between vehicles in different positions. This combination allows our model to take into consideration both the dependencies within the sequence and the social relationships between vehicles, providing a more comprehensive scene understanding. The efficacy of our model is tested on two real-world freeway trajectory datasets, namely NGSIM and HighD. Our model surpasses various baseline methods, exhibiting exceptional accuracy in both prediction and tracking. |
| first_indexed | 2024-03-08T15:54:16Z |
| format | Article |
| id | doaj.art-08d05925bcbe4cf3b04fa0b2bddd9fa6 |
| institution | Directory Open Access Journal |
| issn | 2169-3536 |
| language | English |
| last_indexed | 2024-03-08T15:54:16Z |
| publishDate | 2024-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj.art-08d05925bcbe4cf3b04fa0b2bddd9fa62024-01-09T00:04:36ZengIEEEIEEE Access2169-35362024-01-01121718172610.1109/ACCESS.2023.334564310368009An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention MechanismsSenyao Qiao0https://orcid.org/0009-0005-4675-5833Fei Gao1https://orcid.org/0000-0001-9020-6720Jianghang Wu2https://orcid.org/0009-0002-2383-3200Rui Zhao3https://orcid.org/0000-0003-1597-1961College of Automotive Engineering, Jilin University, Jilin, Changchun, ChinaState Key Laboratory of Automotive Simulation and Control, Jilin University, Jilin, Changchun, ChinaCollege of Automotive Engineering, Jilin University, Jilin, Changchun, ChinaCollege of Automotive Engineering, Jilin University, Jilin, Changchun, ChinaAccurate trajectory prediction for multiple vehicles in complex social interaction environments is essential for ensuring the safety of autonomous vehicles and improving the quality of their planning and control. The social interactions between vehicles significantly influence their future trajectories. However, traditional trajectory prediction models based on Recurrent Neural Networks (RNN) or Convolutional Neural Networks (CNN) often overlook or simplify these interactions. Although these models may exhibit high performance in short-term predictions, they fail to achieve high prediction accuracy in scenarios with long-term dynamic interactions. To address this limitation, we propose a Social-Attention Long Short-Term Memory (LSTM) model which predicts the future trajectories of neighboring vehicles and achieves increased accuracy. Our proposed model employs a Social-Pooling layer to effectively capture cooperative behaviors and mutual influences between vehicles. Additionally, we incorporate a self-attention mechanism to weight the inputs and outputs of the Social-Pooling layer, which is significant for assessing the influence between vehicles in different positions. This combination allows our model to take into consideration both the dependencies within the sequence and the social relationships between vehicles, providing a more comprehensive scene understanding. The efficacy of our model is tested on two real-world freeway trajectory datasets, namely NGSIM and HighD. Our model surpasses various baseline methods, exhibiting exceptional accuracy in both prediction and tracking.https://ieeexplore.ieee.org/document/10368009/Vehicle trajectory predictionautonomous drivinginteraction modelingspatial-temporal attention |
| spellingShingle | Senyao Qiao Fei Gao Jianghang Wu Rui Zhao An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms IEEE Access Vehicle trajectory prediction autonomous driving interaction modeling spatial-temporal attention |
| title | An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms |
| title_full | An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms |
| title_fullStr | An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms |
| title_full_unstemmed | An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms |
| title_short | An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms |
| title_sort | enhanced vehicle trajectory prediction model leveraging lstm and social attention mechanisms |
| topic | Vehicle trajectory prediction autonomous driving interaction modeling spatial-temporal attention |
| url | https://ieeexplore.ieee.org/document/10368009/ |
| work_keys_str_mv | AT senyaoqiao anenhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT feigao anenhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT jianghangwu anenhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT ruizhao anenhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT senyaoqiao enhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT feigao enhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT jianghangwu enhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms AT ruizhao enhancedvehicletrajectorypredictionmodelleveraginglstmandsocialattentionmechanisms |