Back to Search
Start Over
HSTA: A Hierarchical Spatio-Temporal Attention Model for Trajectory Prediction.
- Source :
-
IEEE Transactions on Vehicular Technology . Nov2021, Vol. 70 Issue 11, p11295-11307. 13p. - Publication Year :
- 2021
-
Abstract
- Predicting the future trajectories of surrounding agents has become an crucial problem to be solved for the safety of autonomous vehicles. Recent studies based on Long Short Term Memory (LSTM) networks have shown powerful abilities to model social interactions. However, many of these approaches focus on spatial interactions of the neighborhood agents but ignore temporal interactions that accompany spatial interactions. In this paper, we propose a Hierarchical Spatio-Temporal Attention architecture (HSTA), which activates the utilization of spatial interactions with different weights, and jointly considers the temporal interactions across time steps of all agents. More specially, the graph attention mechanism (GAT) is presented to capture spatial interactions, the multi-head attention mechanism (MHA) is conducted to encode temporal correlations of interactions and a state gated fusion (SGF) layer is used to integrate spatial and temporal interactions. We evaluate our proposed method against baselines on both pedestrian and vehicle datasets. The results show that our model is effective and achieves state-of-the-art achievements. [ABSTRACT FROM AUTHOR]
Details
- Language :
- English
- ISSN :
- 00189545
- Volume :
- 70
- Issue :
- 11
- Database :
- Academic Search Index
- Journal :
- IEEE Transactions on Vehicular Technology
- Publication Type :
- Academic Journal
- Accession number :
- 153732431
- Full Text :
- https://doi.org/10.1109/TVT.2021.3115018