NodeTrans: A Graph Transfer Learning Approach for Traffic Prediction
Recently, deep learning methods have made great progress in traffic
prediction, but their performance depends on a large amount of historical data.
In reality, we may face the data scarcity issue. In this case, deep learning
models fail to obtain satisfactory performance. Transfer learning is a
promising approach to solve the data scarcity issue. However, existing transfer
learning approaches in traffic prediction are mainly based on regular grid
data, which is not suitable for the inherent graph data in the traffic network.
Moreover, existing graph-based models can only capture shared traffic patterns
in the road network, and how to learn node-specific patterns is also a
challenge. In this paper, we propose a novel transfer learning approach to
solve the traffic prediction with few data, which can transfer the knowledge
learned from a data-rich source domain to a data-scarce target domain. First, a
spatial-temporal graph neural network is proposed, which can capture the
node-specific spatial-temporal traffic patterns of different road networks.
Then, to improve the robustness of transfer, we design a pattern-based transfer
strategy, where we leverage a clustering-based mechanism to distill common
spatial-temporal patterns in the source domain, and use these knowledge to
further improve the prediction performance of the target domain. Experiments on
real-world datasets verify the effectiveness of our approach.