Attention based graph convolutional networks for trajectory prediction

Jianxiao Chen, Guang Chen, Zhijun Li, Ya Wu, Alois Knoll

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

2 Scopus citations

Abstract

Predicting the future trajectory of different traffic agents in the complex traffic environments plays an important role in keeping the driving safety of self-driving cars, especially on urban roads. In the most of the existing works, researchers always use the long short-term memory network (LSTM) to solve this problem, since the LSTM has powerful capability for capturing the temporal dependency in motion trajectory. However, they only consider forward time cues and ignore the spatial-temporal correlations between traffic agents. Inspired by the previous work which utilizing the spatial-temporal graph, we design a spatial attention based spatial-temporal graph convolutional network, which assigns different attention weight to the the graph to take the different social interactions among the self-driving cars into consideration. We conduct extensive experiments on the benchmark InD to compare our method against many baselines. The experiment results indicate the superiority of our method than previous method, about 22% and 17% improvement on the metric of ADE and FDE respectively.

Original languageEnglish
Title of host publication2021 6th IEEE International Conference on Advanced Robotics and Mechatronics, ICARM 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages852-857
Number of pages6
ISBN (Electronic)9780738133645
DOIs
StatePublished - 3 Jul 2021
Event6th IEEE International Conference on Advanced Robotics and Mechatronics, ICARM 2021 - Chongqing, China
Duration: 3 Jul 20215 Jul 2021

Publication series

Name2021 6th IEEE International Conference on Advanced Robotics and Mechatronics, ICARM 2021

Conference

Conference6th IEEE International Conference on Advanced Robotics and Mechatronics, ICARM 2021
Country/TerritoryChina
CityChongqing
Period3/07/215/07/21

Fingerprint

Dive into the research topics of 'Attention based graph convolutional networks for trajectory prediction'. Together they form a unique fingerprint.

Cite this