English

ST-LoRA: Low-rank Adaptation for Spatio-Temporal Forecasting

Machine Learning 2025-07-08 v2 Artificial Intelligence

Abstract

Spatio-temporal forecasting is essential for understanding future dynamics within real-world systems by leveraging historical data from multiple locations. Existing methods often prioritize the development of intricate neural networks to capture the complex dependencies of the data. These methods neglect node-level heterogeneity and face over-parameterization when attempting to model node-specific characteristics. In this paper, we present a novel low-rank adaptation framework for existing spatio-temporal prediction models, termed \model, which alleviates the aforementioned problems through node-level adjustments. Specifically, we introduce the node-adaptive low-rank layer and node-specific predictor, capturing the complex functional characteristics of nodes while maintaining computational efficiency. Extensive experiments on multiple real-world datasets demonstrate that our method consistently achieves superior performance across various forecasting models with minimal computational overhead, improving performance by 7% with only 1% additional parameter cost. The source code is available at https://github.com/RWLinno/ST-LoRA.

Keywords

Cite

@article{arxiv.2404.07919,
  title  = {ST-LoRA: Low-rank Adaptation for Spatio-Temporal Forecasting},
  author = {Weilin Ruan and Wei Chen and Xilin Dang and Jianxiang Zhou and Weichuang Li and Xu Liu and Yuxuan Liang},
  journal= {arXiv preprint arXiv:2404.07919},
  year   = {2025}
}

Comments

Published at ECML-PKDD 2025

R2 v1 2026-06-28T15:51:32.795Z