Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning to Factorize and Adapt: A Versatile Approach Toward Universal Spatio-Temporal Foundation Models

About

Spatio-Temporal (ST) Foundation Models (STFMs) promise cross-dataset generalization, yet joint ST pretraining is computationally expensive and grapples with the heterogeneity of domain-specific spatial patterns. Substantially extending our preliminary conference version, we present FactoST-v2, an enhanced factorized framework redesigned for full weight transfer and arbitrary-length generalization. FactoST-v2 decouples universal temporal learning from domain-specific spatial adaptation. The first stage pretrains a minimalist encoder-only backbone using randomized sequence masking to capture invariant temporal dynamics, enabling probabilistic quantile prediction across variable horizons. The second stage employs a streamlined adapter to rapidly inject spatial awareness via meta adaptive learning and prompting. Comprehensive evaluations across diverse domains demonstrate that FactoST-v2 achieves state-of-the-art accuracy with linear efficiency - significantly outperforming existing foundation models in zero-shot and few-shot scenarios while rivaling domain-specific expert baselines. This factorized paradigm offers a practical, scalable path toward truly universal STFMs. Code is available at https://github.com/CityMind-Lab/FactoST.

Siru Zhong, Junjie Qiu, Yangyu Wu, Yiqiu Liu, Yuanpeng He, Zhongwen Rao, Bin Yang, Chenjuan Guo, Hao Xu, Yuxuan Liang• 2026

Related benchmarks

TaskDatasetResultRank
Traffic ForecastingPEMS-03 Long-term (24 -> 24avg)
MAE28.62
30
Traffic ForecastingPEMS-03 Short-term (12 -> 12avg)
MAE21.08
30
Grid-based Spatio-Temporal ForecastingTraffic-SH (Short-term)
MAE0.4
19
Grid-based Spatio-Temporal ForecastingTraffic-SH long-term (24 -> 24avg) (test)
MAE0.58
19
Grid-based Spatio-Temporal ForecastingBike-NYC (Short-term)
MAE3.77
5
Traffic ForecastingPEMS-07 Long-term (24 -> 24avg)
MAE39.99
5
Showing 6 of 6 rows

Other info

Follow for update