HDGT: Heterogeneous Driving Graph Transformer for Multi-Agent Trajectory Prediction via Scene Encoding
- Xiaosong Jia, Penghao Wu, Li Chen, Hongyang Li, Yu Liu, Junchi Yan
- Paper | Blog (coming soon)
HDGT is an unified heterogeneous transformer-based graph neural network for driving scene encoding. It is a SOTA method on INTERACTION and Waymo Motion Prediction Chanllege.
By time of release in April 2022, the proposed method achieves new state-of-the-art on INTERACTION Prediction Challenge and Waymo Open Motion Challenge, in which we rank the first and second respectively in terms of the minADE/minFDE metric.
All content within this repository is under Apache License 2.0.