Invariant Structure Learning with Pre-trained Language Models for Spatio-temporal Graph
Ting Wang, Duo Zhou, Daqian Shi, Hao Tang, Hao Deng, Shengjie Zhao
摘要
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.
问问这篇 Paper
问问你的智能体。
Lune 读过与它相关的顶会 Paper,每个回答都会注明依据哪几篇。
相关 Paper
- STONE: A Spatio-temporal OOD Learning Framework Kills Both Spatial and Temporal ShiftsBinwu Wang, Jiaming Ma, Pengkun Wang, Xu Wang 等KDD 2024 · 被引用 13 次
- STD-PLM: Understanding Both Spatial and Temporal Properties of Spatial-Temporal Data with PLMYiheng Huang, Xiaowei Mao, Shengnan Guo, Yubin Chen 等AAAI 2025 · 被引用 16 次
- 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 等CVPR 2022 · 被引用 44 次
- Exploiting Pre-trained Language Model for Cross-city Urban Flow Prediction Guided by Information-theoretic AnalysisQiang Zhou, Xudong Tong, Yuting Liu, Chuanxing Liu 等AAAI 2026
