mlsanityData to 14 Oct 2026

Dimensional Balance Improves Large Scale Spatiotemporal Prediction Performance

arXiv 2605.18793 ↗First posted 11 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Accurate spatiotemporal pattern analysis is critical in fields such as urban traffic, meteorology, and public health monitoring. However, existing methods face performance bottlenecks, typically yielding only incremental gains and often exhibiting limited cross-domain transferability. We analyze this bottleneck through spatial and temporal entropy measures, which are used as diagnostic indicators of spatiotemporal complexity mismatch rather than as guarantees that entropy alignment alone yields better forecasting. Empirically, larger mismatch is often accompanied by higher prediction uncertainty, especially under a fixed model-capacity budget. Guided by this diagnostic, we propose a scalable, adaptive framework that harmonizes spatial and temporal feature representations. Spatial dimensionality is compressed via low-rank matrix embedding to preserve essential structure, while an extended temporal horizon captures long-range dependencies and mitigates cumulative errors arising from temporal heterogeneity. Extensive experiments on urban traffic, meteorological, and epidemic datasets demonstrate substantial accuracy gains and broad applicability across the evaluated domains, suggesting that the framework is promising for a wide range of spatiotemporal tasks beyond the current study. The code is available on GitHub at https://github.com/ST-Balance/ST-Balance.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 50 references that are in the map. First posted on arXiv 11 May 2026.

Builds on

23 of 50 references.

References · 23

  1. 2014Sequence to Sequence Learning with Neural Networkscited by 2,986
  2. 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,169
  3. 2016node2vec: Scalable Feature Learning for Networkscited by 1,918
  4. 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,676
  5. 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 829
  6. 2023One Fits All:Power General Time Series Analysis by Pretrained LMcited by 632
  7. 2020Adaptive Graph Convolutional Recurrent Network for Traffic Forecastingcited by 530
  8. 2021FNet: Mixing Tokens with Fourier Transformscited by 347
  9. 2024TimeMixer: Decomposable Multiscale Mixing for Time Series Forecastingcited by 346
  10. 2021Spectral Temporal Graph Neural Network for Multivariate Time-series Forecastingcited by 299
  11. 2021Spatial-Temporal Graph ODE Networks for Traffic Flow Forecastingcited by 210
  12. 2020Spatial-Temporal Transformer Networks for Traffic Flow Forecastingcited by 180
11 more
  1. 2021Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solutioncited by 157
  2. 2022Spatial-Temporal Identity: A Simple yet Effective Baseline for Multivariate Time Series Forecastingcited by 151
  3. 2023A Survey on Graph Neural Networks for Time Series: Forecasting, Classification, Imputation, and Anomaly Detectioncited by 145
  4. 2022Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecastingcited by 123
  5. 2023LargeST: A Benchmark Dataset for Large-Scale Traffic Forecastingcited by 118
  6. 2022Pre-training Enhanced Spatial-temporal Graph Neural Network for Multivariate Time Series Forecastingcited by 117
  7. 2023Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecastingcited by 27
  8. 2024FlashST: A Simple and Universal Prompt-Tuning Framework for Traffic Predictioncited by 18
  9. 2022EpiGNN: Exploring Spatial Transmission with Graph Neural Network for Regional Epidemic Forecastingcited by 17
  10. 2025Dynamic Trend Fusion Module for Traffic Flow Predictioncited by 5
  11. 2021Multiwave COVID-19 Prediction from Social Awareness using Web Search and Mobility Datacited by 3

Built on by

Not cited yet.