Citation

Zhang, Yongxiang and Rashid, Sidra and Khan, Muazzam A. and Akram, Usman and Mumtaz, Raheel and Lee, It Ee and Syed, Toqeer Ali (2026) A non-local multi-head spatiotemporal attention LSTM for vehicle trajectory prediction. PLOS One, 21 (9). e0357735. ISSN 1932-6203

Text A non-local multi-head spatiotemporal attention LSTM for vehicle trajectory prediction.pdf - Published Version Restricted to Repository staff only Download (1MB)

Abstract

Vehicle trajectory prediction (VTP) is the primary part of the perception-planningcontrol pipeline in autonomous driving. The inter-vehicle interactions strongly shape the trajectory of vehicles in complex traffic scenarios. Neural networks that use recurrent or convolutional architectures often exhibit performance degradation in longer prediction horizons. Many existing approaches confine interaction modeling to a predefined spatial neighborhood and a narrow temporal window, while implicitly assuming that all neighboring vehicles have equal influence on the future trajectory of target vehicle. This results in error accumulation and the inability to capture long-range spatiotemporal dependencies. To overcome these limitations, a non-local multi-head spatiotemporal attention based long short-term memory model (NL-MHALSTM) is introduced which employs an attention mechanism to assign context weights to relevant neighbor vehicles. It extends beyond pairwise effects to model long-range dependencies. The model emphasizes the most influential vehicles and formulate position-aware interaction representations. A comprehensive set of experiments is conducted on the pubicly available HighD dataset. The results demonstrate that the proposed model outperforms all state-of-the-art methods, achieving a 53.4% reduction in RMSE at the 5s prediction horizon relative to the next-best model. In addition, a detailed ablation study is conducted to systematically evaluate the impact of different attention mechanisms and varying numbers of attention heads on prediction accuracy across multiple horizons.

Item Type:	Article
Uncontrolled Keywords:	Vehicle trajectory prediction
Subjects:	T Technology > TL Motor vehicles. Aeronautics. Astronautics
Divisions:	Faculty of Artificial Intelligence & Engineering (FAIE)
Depositing User:	Ms Rosnani Abd Wahab
Date Deposited:	01 Oct 2026 02:17
Last Modified:	01 Oct 2026 02:17
URII:	http://shdl.mmu.edu.my/id/eprint/16753

Downloads

Edit (login required)