Skip to content

ShichengChen/Spatio-Temporal-Graph-Convolutional-Networks-A-Deep-Learning-Framework-for-Traffic-Forecasting

Repository files navigation

Spatio-Temporal-Graph-Convolutional-Networks-A-Deep-Learning-Framework-for-Traffic-Forecasting

reference

abstract

  • Spatio-Temporal Graph Convolutional Network
  • tackle the time series prediction problem in traffic domain
  • complete convolutional structures.

introduction

  • linear regression perform well on short interval forecast instead of long terms
  • this is a data-driven and using spotio-temporal information method.
  • fully utilize spatio-information instead of treating it as discrete units
  • $$\hat v_{t+1},...,\hat v_{t+H} = argmax log_{10} P(v_{t+1},...,v_{t+H}|v_{t-M},...,v_{t})$$
  • function one
  • where $$v_t \in R^n$$, n is an observation vector of n road segments at time step t

Convolutions on Graphs

convolution on graphs

  • normalized Laplacian

    • Random walk normalized Laplacian
    • analogy to The Multivariate Gaussian Distribution function one
    • Symmetric normalized Laplacian L: laplacian
  • first generation of GNC

  • second generation of GNC

    • if k == n, receptive field is n hop
  • third generation of GNC

    • where $$c_1$$, $$c_2$$ and $$c_3$$ are fixed
    • The only trainable parameters are $$\theta_0$$ and $$\theta_1$$
    • in the final version the authors even further fix $$\theta_0 = -\theta_1$$

Network Architecture

  • main architecture
  • GLU architecture
  • main equation
  • final equation

Experiments

  • linear interpolation method for missing values
  • normalized by standard score method((x-mean)/std)
  • adjacency matrix 10,0.5

result

About

presentation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages