Please use this identifier to cite or link to this item: https://hdl.handle.net/10356/162966
Title: Multi-agent trajectory prediction with heterogeneous edge-enhanced graph attention network
Authors: Mo, Xiaoyu
Huang, Zhiyu
Xing, Yang
Lv, Chen
Keywords: Engineering::Mechanical engineering
Issue Date: 2022
Source: Mo, X., Huang, Z., Xing, Y. & Lv, C. (2022). Multi-agent trajectory prediction with heterogeneous edge-enhanced graph attention network. IEEE Transactions On Intelligent Transportation Systems, 23(7), 9554-9567. https://dx.doi.org/10.1109/TITS.2022.3146300
Project: W1925d0046
Journal: IEEE Transactions on Intelligent Transportation Systems
Abstract: Simultaneous trajectory prediction for multiple heterogeneous traffic participants is essential for safe and efficient operation of connected automated vehicles under complex driving situations. Two main challenges for this task are to handle the varying number of heterogeneous target agents and jointly consider multiple factors that would affect their future motions. This is because different kinds of agents have different motion patterns, and their behaviors are jointly affected by their individual dynamics, their interactions with surrounding agents, as well as the traffic infrastructures. A trajectory prediction method handling these challenges will benefit the downstream decision-making and planning modules of autonomous vehicles. To meet these challenges, we propose a three-channel framework together with a novel Heterogeneous Edge-enhanced graph ATtention network (HEAT). Our framework is able to deal with the heterogeneity of the target agents and traffic participants involved. Specifically, agents' dynamics are extracted from their historical states using type-specific encoders. The inter-agent interactions are represented with a directed edge-featured heterogeneous graph and processed by the designed HEAT network to extract interaction features. Besides, the map features are shared across all agents by introducing a selective gate-mechanism. And finally, the trajectories of multiple agents are predicted simultaneously. Validations using both urban and highway driving datasets show that the proposed model can realize simultaneous trajectory predictions for multiple agents under complex traffic situations, and achieve state-of-the-art performance with respect to prediction accuracy. The achieved final displacement error (FDE@3sec) is 0.66 meter under urban driving, demonstrating the feasibility and effectiveness of the proposed approach.
URI: https://hdl.handle.net/10356/162966
ISSN: 1524-9050
DOI: 10.1109/TITS.2022.3146300
Rights: © 2022 IEEE. All rights reserved.
Fulltext Permission: none
Fulltext Availability: No Fulltext
Appears in Collections:MAE Journal Articles

SCOPUSTM   
Citations 20

9
Updated on Nov 29, 2022

Web of ScienceTM
Citations 20

8
Updated on Dec 2, 2022

Page view(s)

11
Updated on Dec 3, 2022

Google ScholarTM

Check

Altmetric


Plumx

Items in DR-NTU are protected by copyright, with all rights reserved, unless otherwise indicated.