mlsanityData to 13 Oct 2026

UrbanFM: Scaling Urban Spatio-Temporal Foundation Models

arXiv 2602.20677 ↗First posted 24 Feb 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Urban systems, as dynamic complex systems, continuously generate spatio-temporal data streams that encode the fundamental laws of human mobility and city evolution. While AI for Science has witnessed the transformative power of foundation models in disciplines like genomics and meteorology, urban computing remains fragmented due to "scenario-specific" models, which are overfitted to specific regions or tasks, hindering their generalizability. To bridge this gap and advance spatio-temporal foundation models for urban systems, we adopt scaling as the central perspective and systematically investigate two key questions: what to scale and how to scale. Grounded in first-principles analysis, we identify three critical dimensions: heterogeneity, correlation, and dynamics, aligning these principles with the fundamental scientific properties of urban spatio-temporal data. Specifically, to address heterogeneity through data scaling, we construct WorldST. This billion-scale corpus standardizes diverse physical signals, such as traffic flow and speed, from over 100 global cities into a unified data format. To enable computation scaling for modeling correlations, we introduce the MiniST unit, a novel split mechanism that discretizes continuous spatio-temporal fields into learnable computational units to unify representations of grid-based and sensor-based observations. Finally, addressing dynamics via architecture scaling, we propose UrbanFM, a minimalist self-attention architecture designed with limited inductive biases to autonomously learn dynamic spatio-temporal dependencies from massive data. Furthermore, we establish EvalST, the largest-scale urban spatio-temporal benchmark to date. Extensive experiments demonstrate that UrbanFM achieves remarkable zero-shot generalization across unseen cities and tasks, marking a pivotal first step toward large-scale urban spatio-temporal foundation models.

Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Builds on 52 of its 83 references that are in the map. First posted on arXiv 24 Feb 2026.

Builds on

52 of 83 references.

References · 52

  1. 2017Attention Is All You Needcited by 60,106
  2. 2020Language Models are Few-Shot Learnerscited by 32,320
  3. 2020Scaling Laws for Neural Language Modelscited by 6,494
  4. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,628
  5. 2024GPT-4o System Cardcited by 4,536
  6. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,637
  7. 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 2,016
  8. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,994
  9. 2022Are Transformers Effective for Time Series Forecasting?cited by 1,413
  10. 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 825
  11. 2024Chronos: Learning the Language of Time Seriescited by 737
  12. 2023A decoder-only foundation model for time-series forecastingcited by 674
38 more
  1. 2024Unified Training of Universal Time Series Forecasting Transformerscited by 602
  2. 2016Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Predictioncited by 333
  3. 2024Time-MoE: Billion-Scale Time Series Foundation Models with Mixture of Expertscited by 286
  4. 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 181
  5. 2019Deep Learning for Spatio-Temporal Data Mining: A Surveycited by 172
  6. 2022Spatial-Temporal Identity: A Simple yet Effective Baseline for Multivariate Time Series Forecastingcited by 149
  7. 2023A Survey on Graph Neural Networks for Time Series: Forecasting, Classification, Imputation, and Anomaly Detectioncited by 145
  8. 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Surveycited by 140
  9. 2022Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecastingcited by 122
  10. 2023Exploring Progress in Multivariate Time Series Forecasting: Comprehensive Benchmarking and Heterogeneity Analysiscited by 115
  11. 2021Filling the G_ap_s: Multivariate Time Series Imputation by Graph Neural Networkscited by 94
  12. 2022Temporal Attention Unit: Towards Efficient Spatiotemporal Predictive Learningcited by 86
  13. 2024UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Predictioncited by 86
  14. 2023ImputeFormer: Low Rankness-Induced Transformers for Generalizable Spatiotemporal Imputationcited by 50
  15. 2020Generative Adversarial Networks for Spatio-temporal Data: A Surveycited by 35
  16. 2021Spatio-Temporal Variational Gaussian Processescited by 32
  17. 2020Preserving Dynamic Attention for Long-Term Spatial-Temporal Predictioncited by 30
  18. 2023Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecastingcited by 27
  19. 2024OpenCity: Open Spatio-Temporal Foundation Models for Traffic Predictioncited by 27
  20. 2021Historical Inertia: A Neglected but Powerful Baseline for Long Sequence Time-series Forecastingcited by 26
  21. 2025Foundation Models for Spatio-Temporal Data Science: A Tutorial and Surveycited by 24
  22. 2018Machine Learning for Spatiotemporal Sequence Forecasting: A Surveycited by 19
  23. 2024Graph-based Forecasting with Missing Data through Spatiotemporal Downsamplingcited by 16
  24. 2024Future Directions in Human Mobility Sciencecited by 16
  25. 2024BIGCity: A Universal Spatiotemporal Model for Unified Trajectory and Traffic State Data Analysiscited by 16
  26. 2025Speed Always Wins: A Survey on Efficient Architectures for Large Language Modelscited by 16
  27. 2024On Catastrophic Inheritance of Large Foundation Modelscited by 15
  28. 2024Expand and Compress: Exploring Tuning Principles for Continual Spatio-Temporal Graph Forecastingcited by 10
  29. 2024Diffusion Transformers as Open-World Spatiotemporal Foundation Modelscited by 10
  30. 2025Unraveling Spatio-Temporal Foundation Models via the Pipeline Lens: A Comprehensive Reviewcited by 10
  31. 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecastingcited by 9
  32. 2024UniFlow: A Foundation Model for Unified Urban Spatio-Temporal Flow Predictioncited by 7
  33. 2024MAS4POI: a Multi-Agents Collaboration System for Next POI Recommendationcited by 6
  34. 2024FairSTG: Countering performance heterogeneity via collaborative sample-level optimizationcited by 3
  35. 2025UniSTD: Towards Unified Spatio-Temporal Learning across Diverse Disciplinescited by 3
  36. 2025Beyond Regularity: Modeling Chaotic Mobility Patterns for Next Location Predictioncited by 3
  37. 2025Spatiotemporal Forecasting as Planning: A Model-Based Reinforcement Learning Approach with Generative World Modelscited by 3
  38. 2025UrbanMind: Urban Dynamics Prediction with Multifaceted Spatial-Temporal Large Language Modelscited by 2

50 most cited of 52 All 52 (JSON)

Built on by

Cited by year (2026 to 13 Oct)

Citing papers per year: 2026: 1.

1 paper.