Trajectory Prediction for Autonomous Driving Using Spatial-Temporal Graph Attention Transformer

For autonomous vehicles driving on roads, future trajectories of surrounding traffic agents (e.g., vehicles, bicycles, pedestrians) are essential information. The prediction of future trajectories is challenging as the motion of traffic agents is constantly affected by spatial-temporal interactions from agents and road infrastructure. To take those interactions into account, this study proposes a Graph Attention Transformer (Gatformer) in which a traffic scene is represented by a sparse graph. To maintain the spatial and temporal information of traffic agents in a traffic scene, Convolutional Neural Networks (CNNs) are utilized to extract spatial features and a position encoder is proposed to encode the spatial features and the corresponding temporal features. Based on the encoded features, a Graph Attention Network (GAT) block is employed to model the agent-agent and agent-infrastructure interactions with the help of attention mechanisms. Finally, a Transformer network is introduced to predict trajectories for multiple agents simultaneously. Experiments are conducted over the Lyft dataset and state-of-the-art methods are introduced for comparison. The results show that the proposed Gatformer could make more accurate predictions while requiring less inference time than its counterparts.

• Record URL:
  http://dx.doi.org/10.1109/TITS.2022.3164450
• Availability:
  • Find a library where document is available. Order URL: http://worldcat.org/oclc/41297384
• Supplemental Notes:
  • Copyright © 2022, IEEE.
• Authors:
  • Zhang, Kunpeng
  • Feng, Xiaoliang
  • Wu, Lan
  • He, Zhengbing
• Publication Date: 2022-11

Language

• English

Media Info

• Media Type: Web
• Features: References;
• Pagination: pp 22343-22353
• Serial:
  • IEEE Transactions on Intelligent Transportation Systems
  • Volume: 23
  • Issue Number: 11
  • Publisher: Institute of Electrical and Electronics Engineers (IEEE)
  • ISSN: 1524-9050
  • Serial URL: http://ieeexplore.ieee.org/xpl/RecentIssue.jsp?punumber=6979

Subject/Index Terms

• TRT Terms: Autonomous vehicles; Predictive models; Trajectory; Vehicle to infrastructure communications; Vehicle to vehicle communications
• Subject Areas: Data and Information Technology; Highways; Vehicles and Equipment;

Filing Info

• Accession Number: 01876321
• Record Type: Publication
• Files: TRIS
• Created Date: Mar 21 2023 9:27AM