PT-TDGCN: Pre-Trained Trend-Aware Dynamic Graph Convolutional Network for Traffic Flow Prediction
Accurate traffic flow prediction is vital for intelligent transportation systems, yet strong spatiotemporal coupling and multi-scale dynamics make modelling difficult. Existing methods often rely on static adjacency and short input windows, limiting adaptation to time-varying spatial relations and long-term patterns. To address these issues, we propose the Pre-trained Trend-aware Dynamic Graph Convolutional Network (PT-TDGCN), a two-stage framework. In the pre-training stage, a Transformer-based masked autoencoder learns segment-level temporal representations from historical sequences. In the prediction stage, three designs are integrated: (1) dynamic graph learning parameterized by tensor decomposition; (2) convolutional trend-aware attention that adds 1D convolutions to capture local trends while preserving global context; and (3) spatial graph convolution combined with lightweight fusion projection for aligning pre-trained, spatial, and temporal representations. Extensive experiments on four real-world datasets demonstrated that PT-TDGCN consistently outperformed 14 baseline models, achieving superior predictive accuracy and robustness.
Authors
- Yuanqing Wang (ORCID: https://orcid.org/0000-0003-4895-5271)
- Hanqing Yang
- Sen Wei
Institutions
- Chang'an University (CN)
Publication Details
- Journal
- Sensors
- Published
- 2025-11-03
- DOI
- https://doi.org/10.3390/s25216709
- Primary Topic
- Traffic Prediction and Management Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00