Abstract
The advancement of intelligent transportation systems underscores the importance of data-driven approaches in traffic forecasting, which plays a crucial role in tasks such as traffic signal control and route guidance, among others. However, the inherent uncertainty stemming from regional traffic dynamics, coupled with intricate spatio-temporal correlations, poses formidable challenges to accurate traffic prediction. Moreover, the complexities inherent in sequence forecasting across varying scales further exacerbate the accuracy dilemma. Recognizing the need for integrating information across spatial and temporal dimensions to enhance forecasting precision, a novel solution termed Spatial Temporal Masked Autoencoder (STMAE) is introduced. The STMAE framework addresses these challenges through a two-stage learning process. In the pre-training phase, an autoencoder architecture is employed to extract spatio-temporal features from the data. In the fine-tuning phase, the pre-trained encoder of the STMAE model undergoes further refinement to specifically target traffic forecasting tasks. Extensive evaluations validate the effectiveness of the proposed STMAE model. Notably, STMAE demonstrates competitive performance, achieving 3.32 Vehs MAE for long-term (60 min) traffic forecasting while operating within a reduced computational budget.
Original language | English |
---|---|
Title of host publication | Pattern Recognition |
Subtitle of host publication | 27th International Conference, ICPR 2024, Kolkata, India, December 1–5, 2024, Proceedings, Part V |
Editors | Apostolos Antonacopoulos, Subhasis Chaudhuri, Rama Chellappa, Cheng-Lin Liu, Saumik Bhattacharya, Umapada Pal |
Place of Publication | Cham |
Publisher | Springer |
Pages | 209-223 |
Number of pages | 15 |
ISBN (Electronic) | 9783031781698 |
ISBN (Print) | 9783031781681 |
DOIs | |
Publication status | Published - 30 Nov 2024 |
Event | 27th International Conference on Pattern Recognition - Kolkata, India Duration: 1 Dec 2024 → 5 Dec 2024 https://link.springer.com/book/10.1007/978-3-031-78107-0 (Conference proceedings) https://icpr2024.org/ (Conference website) |
Publication series
Name | Lecture Notes in Computer Science |
---|---|
Volume | 15305 |
ISSN (Print) | 0302-9743 |
ISSN (Electronic) | 1611-3349 |
Name | ICPR: International Conference on Pattern Recognition |
---|
Conference
Conference | 27th International Conference on Pattern Recognition |
---|---|
Abbreviated title | ICPR 2024 |
Country/Territory | India |
City | Kolkata |
Period | 1/12/24 → 5/12/24 |
Internet address |
|
Scopus Subject Areas
- Theoretical Computer Science
- General Computer Science
User-Defined Keywords
- Data-driven Methods
- Masked Autoencoder
- Traffic Forecasting