• Home
  • Search
  • Revealing the Power of Masked Autoencoders in Traffic Forecasting
  • Cite Icon6
  • https://doi.org/10.1145/3627673.3679989Copy DOI Icon

Revealing the Power of Masked Autoencoders in Traffic Forecasting

  • Oct 21, 2024
  • Jiarui Sun +4 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Traffic forecasting, crucial for urban planning, requires accurate predictions of spatial-temporal traffic patterns across urban areas. Existing research mainly focuses on designing complex spatial-temporal models to capture these dependencies. However, this field faces challenges related to data scarcity and model stability, which results in limited performance improvement. To address these issues, we propose Spatial-Temporal Masked AutoEncoders (STMAE), a plug-and-play framework designed to enhance existing spatial-temporal models on traffic prediction. STMAE operates in two stages. In the pretraining stage, an encoder processes partially visible traffic data produced by a dual-masking strategy, including biased random walk-based spatial masking and patch-based temporal masking. Subsequently, two decoders aim to reconstruct the masked counterparts from both spatial and temporal perspectives. The fine-tuning stage retains the pretrained encoder and integrates it with decoders from existing backbones to improve traffic forecasting accuracy. Our results on traffic benchmarks show that STMAE can largely enhance the forecasting capabilities of various spatial-temporal models.

Similar Papers
  • PDF
  • Research Article
  • Citations22

An Overview Based on the Overall Architecture of Traffic Forecasting

  • Mar 23, 2024
  • Data Science and Engineering
  • Lilan Peng +4
  • Book Chapter
  • Citations14

A Survey of Traffic Prediction Based on Deep Neural Network: Data, Methods and Challenges

  • Jan 01, 2022
  • Lecture notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering
  • Pengfei Cao +4
  • Research Article
  • Citations37

Incorporating multimodal context information into traffic speed forecasting through graph deep learning

  • Jul 18, 2023
  • International Journal of Geographical Information Science
  • Yatao Zhang +3
  • Research Article

Graph Neural Networks for Urban Traffic Flow Forecasting: A Comprehensive Review and Future Perspectives

  • Oct 23, 2025
  • Theoretical and Natural Science
  • Yuchen Zhu
  • Conference Article
  • Citations11

Traffic Flow Forecasting in Intelligent Transportation Systems Prediction Using Machine Learning

  • Nov 25, 2022
  • Mohammad Naveed Hossain +2
  • Research Article
  • Citations8

Traffic Data Collection and Visualization Tool for Knowledge Discovery Using Google Maps

  • Jan 14, 2022
  • International Journal of Software Innovation
  • Iftekhar Hossain +1
  • Research Article
  • Citations82

RT-GCN: Gaussian-based spatiotemporal graph convolutional network for robust traffic prediction

  • Oct 14, 2023
  • Information Fusion
  • Yutian Liu +4
  • Research Article
  • Citations22

Challenges and opportunities in traffic flow prediction: review of machine learning and deep learning perspectives

  • Jan 01, 2024
  • Data and Metadata
  • Syed Aleem Uddin Gilani +2
  • Research Article
  • Citations26

Traffic demand prediction based on spatial-temporal guided multi graph Sandwich-Transformer

  • Jun 01, 2023
  • Information Sciences
  • Yanjie Wen +3
  • Research Article
  • Citations17

Multi-Slice Privacy-Aware Traffic Forecasting at RAN Level: A Scalable Federated-Learning Approach

  • Dec 01, 2023
  • IEEE Transactions on Network and Service Management
  • Hnin Pann Phyu +2
  • Research Article
  • Citations67

A Combined Deep Learning Method with Attention-Based LSTM Model for Short-Term Traffic Speed Forecasting

  • Nov 23, 2020
  • Journal of Advanced Transportation
  • Pan Wu +5
  • Research Article
  • Citations27

Real-time traffic congestion prediction using big data and machine learning techniques

  • Oct 24, 2022
  • World Journal of Engineering
  • Priyanka Chawla +7
  • Conference Article
  • Citations64

Improving Urban Traffic Speed Prediction Using Data Source Fusion and Deep Learning

  • Feb 01, 2019
  • Aniekan Essien +3
  • PDF
  • Research Article
  • Citations31

Assessment of short-term heat effects on cardiovascular mortality and vulnerability factors using small area data in Europe

  • Aug 16, 2023
  • Environment International
  • Siqi Zhang +15
  • Conference Article
  • Citations7

Basis-Prediction Method for Short-Term Traffic Flow Prediction and Its Application

  • Nov 01, 2019
  • Zhiyang Gu +1
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.