Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction

By Abduallah A. Mohamed, Kun Qian, Mohamed Elhoseiny, Christian G. Claudel. Computer Vision and Pattern Recognition, 2020

1,028 citations recorded by the source corpus.

Better machine understanding of pedestrian behaviors enables faster progress in modeling interactions between agents such as autonomous vehicles and humans. Pedestrian trajectories are not only influenced by the pedestrian itself but also by interaction with surrounding objects. Previous methods modeled these interactions by using a variety of aggregation methods that integrate different learned pedestrians states. We propose the Social Spatio-Temporal Graph Convolutional Neural Network (Social-STGCNN), which substitutes the need of aggregation methods by modeling the interactions as a graph. Our results show an improvement over the state of art by 20% on the Final Displacement Error (FDE) and an improvement on the Average Displacement Error (ADE) with 8.5 times less parameters and up to 48 times faster inference speed than previously reported methods. In addition, our model is data efficient, and exceeds previous state of the art on the ADE metric with only 20% of the training data. We propose a kernel function to embed the social interactions between pedestrians within the adjacency matrix. Through qualitative analysis, we show that our model inherited social behaviors that can be expected between pedestrians trajectories. Code is available at https://github.com/abduallahmohamed/Social-STGCNN.

Subfields: Automotive Engineering, Computer Vision and Pattern Recognition, Safety, Risk, Reliability and Quality

View Paper: https://www.semanticscholar.org/paper/1b24169304c4e85f60beae5e198e61038cfb3095

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Automotive Engineering

  1. Luyao Ye
  2. Jean Oh
  3. Gang Hua
  4. Kui Wu
  5. Jiwen Lu
  6. Zhenxing Niu
  7. Kejie Lu
  8. Aaron Steinfeld

In Computer Vision and Pattern Recognition

  1. Jie Zhou
  2. Jiwen Lu
  3. Yongming Rao
  4. Gang Hua
  5. Jean Oh
  6. Aaron Steinfeld
  7. Christoforos Mavrogiannis
  8. Francesca Baldini

In Safety, Risk, Reliability and Quality

  1. Jean Oh
  2. Aaron Steinfeld
  3. Pete Trautman
  4. Christoforos Mavrogiannis
  5. Francesca Baldini
  6. Dapeng Zhao
  7. Allan Wang

Citations in the corpus