Transfer Learning With Spatial–Temporal Graph Convolutional Network for Traffic Prediction

Accurate spatial-temporal traffic modeling and prediction play an important role in intelligent transportation systems (ITS). Recently, various deep learning methods such as graph convolutional networks (GCNs) and recurrent neural networks (RNNs) have been widely adopted in traffic prediction tasks to extract spatial-temporal dependencies based on a large volume of high-quality training data. However, there exist data scarcity problems in some transportation networks, and in these cases, the performance of traditional GCNs and RNNs based approaches will degrade sharply. To address this problem, this paper proposes an adversarial domain adaptation with spatial-temporal graph convolutional network (Ada-STGCN) model to predict traffic indicators for a data-scarce target road network by transferring the knowledge from a data-sufficient source road network. Specifically, Ada-STGCN first develops a spatial-temporal graph convolutional network that combines the GCN and gated recurrent unit (GRU) to extract spatial-temporal dependencies from source and target road networks. Then, the technique of adversarial domain adaptation is integrated with the spatial-temporal graph convolutional network to learn discriminative and domain-invariant features to facilitate knowledge transfer. Experimental results on the real-world traffic datasets in the traffic flow prediction task demonstrate that the authors' model yields the best prediction performance compared to state-of-the-art baseline methods.

• Record URL:
  • http://dx.doi.org/10.1109/TITS.2023.3250424
• Availability:
  • Find a library where document is available. Order URL: http://worldcat.org/oclc/41297384
• Supplemental Notes:
  • Copyright © 2023, IEEE.
• Authors:
  • Yao, Zhixiu
  • Xia, Shichao
  • Li, Yun
  • Wu, Guangfu
  • Zuo, Linli
• Publication Date: 2023-8

Language

• English

Media Info

• Media Type: Web
• Features: References;
• Pagination: pp 8592-8605
• Serial:
  • IEEE Transactions on Intelligent Transportation Systems
  • Volume: 24
  • Issue Number: 8
  • Publisher: Institute of Electrical and Electronics Engineers (IEEE)
  • ISSN: 1524-9050
  • Serial URL: http://ieeexplore.ieee.org/xpl/RecentIssue.jsp?punumber=6979

Subject/Index Terms

• TRT Terms: Machine learning; Neural networks; Predictive models; Time; Traffic forecasting
• Subject Areas: Data and Information Technology; Highways; Operations and Traffic Management;

Filing Info

• Accession Number: 01901195
• Record Type: Publication
• Files: TRIS
• Created Date: Dec 1 2023 9:13AM