CityTrans: Domain-Adversarial Training with Knowledge Transfer for Spatio-Temporal Prediction Across Cities

Xiaocao Ouyang,Yan Yang,Wei Zhou,Yiling Zhang,Hao Wang,Wei Huang
DOI: https://doi.org/10.1109/tkde.2023.3283520
IF: 9.235
2024-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:As the spatio-temporal data of a city is not always available, insufficient data would lead to poor performance in some urban prediction tasks. Existing works utilize transfer learning to solve the data scarcity problem, but they ignore the differences in data distributions across cities, which leads to the ineffectiveness of knowledge transfer. In this paper, we propose a domain adversarial model with knowledge transfer for spatio-temporal prediction across cities, entitled CityTrans . Specifically, 1) the self-adaptive spatio-temporal knowledge (namely ST-Knowledge) is mined, to learn the latent spatial and temporal patterns among cities; 2) the domain-adversarial training strategy is introduced to enhance domain invariance; 3) a knowledge attention mechanism is proposed to extract the transferable information from the ST-Knowledge. Note that our CityTrans is an end-to-end domain adversarial spatio-temporal network without two-stage training (i.e., pre-training and fine-tuning). Finally, we conduct extensive experiments on two spatio-temporal prediction tasks: traffic (flow and speed) prediction, and air quality prediction. Experimental results demonstrate that CityTrans outperforms state-of-the-art models on all tasks by a significant margin.
What problem does this paper attempt to address?