English

BiDepth: A Bidirectional-Depth Neural Network for Spatio-Temporal Prediction

Machine Learning 2025-07-15 v3 Artificial Intelligence Computer Vision and Pattern Recognition Applications

Abstract

Accurate spatial-temporal (ST) prediction for dynamic systems, such as urban mobility and weather patterns, is crucial but hindered by complex ST correlations and the challenge of concurrently modeling long-term trends with short-term fluctuations. Existing methods often falter in these areas. This paper proposes the BiDepth Multimodal Neural Network (BDMNN), which integrates two key innovations: 1) a bidirectional depth modulation mechanism that dynamically adjusts network depth to comprehensively capture both long-term seasonality and immediate short-term events; and 2) a novel convolutional self-attention cell (CSAC). Critically, unlike many attention mechanisms that can lose spatial acuity, our CSAC is specifically designed to preserve crucial spatial relationships throughout the network, akin to standard convolutional layers, while simultaneously capturing temporal dependencies. Evaluated on real-world urban traffic and precipitation datasets, BDMNN demonstrates significant accuracy improvements, achieving a 12% Mean Squared Error (MSE) reduction in urban traffic prediction and a 15% improvement in precipitation forecasting over leading deep learning benchmarks like ConvLSTM, using comparable computational resources. These advancements offer robust ST forecasting for smart city management, disaster prevention, and resource optimization.

Keywords

Cite

@article{arxiv.2501.08411,
  title  = {BiDepth: A Bidirectional-Depth Neural Network for Spatio-Temporal Prediction},
  author = {Sina Ehsani and Fenglian Pan and Qingpei Hu and Jian Liu},
  journal= {arXiv preprint arXiv:2501.08411},
  year   = {2025}
}

Comments

21 pages, 6 figures. Submitted to ACM TKDD

R2 v1 2026-06-28T21:06:30.056Z