Invariant Structure Learning with Pre-trained Language Models for Spatio-temporal Graph
Ting Wang, Duo Zhou, Daqian Shi, Hao Tang, Hao Deng, Shengjie Zhao
Abstract
While Pre-trained Language Models (PLMs) have demonstrated remarkable potential in spatiotemporal forecasting, their deployment in critical real-world systems is hindered by a fundamental vulnerability, brittleness under distribution shifts. Existing PLM-based approaches primarily exploit statistical correlations during training, often by linearizing complex spatiotemporal graph structures into flat token sequences, but fail to distinguish stable mechanisms (e.g., road network topology) from spurious environmental features (e.g., temporary weather). To address this, we propose InvSTG-PLM, an invariance-aware spatiotemporal graph learning framework that grounds PLMs in invariant structures. We formalize an invariance principle via token re-assignment, which simulates environments in the representation space, and design a variance-regularized objective which penalizes predictors relying on variant factors. Guided by this principle, we introduce a practical realization including the InvSTG-Tokenizer and InvSTG-Adapter, which goes beyond standard serialization by acting as an invariant filter: it separates invariant structure from variant factors and aligns invariant geometric structures with the semantic space of PLMs. Theoretical analysis justifies a deterministic robustness bound, showing that mean–variance control upper bounds worst-case risk over simulated environments. Extensive experiments on multiple benchmarks demonstrate that InvSTG-PLM outperforms existing methods and exhibits improved robustness under distribution shifts.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get 41cc9052-d7ef-426b-94ea-a18f5465c91eRelated papers
- STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal ShiftsBinwu Wang, Jiaming Ma, Pengkun Wang, Xu Wang et al.KDD 2024 · 13 citations
- STD-PLM: Understanding Both Spatial and Temporal Properties of Spatial-Temporal Data with PLMYiheng Huang, Xiaowei Mao, Shengnan Guo, Yubin Chen et al.AAAI 2025 · 16 citations
- Decoupling Universal Laws and Environmental Heterogeneity: A Physics-Inspired Framework for Robust Spatio-Temporal ForecastingAoyu Liu, Liming Wei, YAYING ZHANGICML 2026
- Towards Robust and Adaptive Motion Forecasting: A Causal Representation PerspectiveYuejiang Liu, Riccardo Cadei, Jonas Schweizer, Sherwin Bahmani et al.CVPR 2022 · 44 citations
- Exploiting Pre-trained Language Model for Cross-city Urban Flow Prediction Guided by Information-theoretic AnalysisQiang Zhou, Xudong Tong, Yuting Liu, Chuanxing Liu et al.AAAI 2026
