Bayesian Spatio-Temporal Graph Convolutional Network for Traffic
  Forecasting
  In traffic forecasting, graph convolutional networks (GCNs), which model
traffic flows as spatio-temporal graphs, have achieved remarkable performance.
However, existing GCN-based methods heuristically define the graph structure as
the physical topology of the road network, ignoring potential dependence of the
graph structure over traffic data. And the defined graph structure is
deterministic, which lacks investigation of uncertainty. In this paper, we
propose a Bayesian Spatio-Temporal Graph Convolutional Network (BSTGCN) for
traffic prediction. The graph structure in our network is learned from the
physical topology of the road network and traffic data in an end-to-end manner,
which discovers a more accurate description of the relationship among traffic
flows. Moreover, a parametric generative model is proposed to represent the
graph structure, which enhances the generalization capability of GCNs. We
verify the effectiveness of our method on two real-world datasets, and the
experimental results demonstrate that BSTGCN attains superior performance
compared with state-of-the-art methods.
