arXiv:2602.04153cs.LGcs.AI2026-02

通过剪枝非关键图结构,提升模型在数据少和跨域场景下的预测能力。

Pruning for Generalization: A Transfer-Oriented Spatiotemporal Graph Framework

  • 基于信息论与相关性筛选关键子图和特征,构建紧凑语义表示。
  • 在低数据转移场景下,相比基线模型性能显著提升。
  • 适合交通预测等图结构时间序列任务,尤其在数据稀缺时有效。

图结构域中的多变量时间序列预测对现实应用至关重要,但现有时空模型在数据稀疏和跨域分布偏移下常出现性能下降。本文从结构感知上下文选择视角出发,提出一种面向迁移的时空框架TL-GPSTGN,通过有选择地剪枝非优化图上下文,提升样本效率和分布外泛化能力。具体而言,该方法采用信息论与相关性准则提取结构上有意义的子图与特征,生成紧凑且语义明确的表示,并将其融入时空卷积架构以捕捉复杂的多变量动态。在大规模交通基准上的实验表明,TL-GPSTGN在低数据迁移场景中持续优于多个基线模型。研究结果表明,显式上下文剪枝可作为增强图基预测模型鲁棒性的有力归纳偏置。

原文摘要 · Abstract (English)

Multivariate time series forecasting in graph-structured domains is critical for real-world applications, yet existing spatiotemporal models often suffer from performance degradation under data scarcity and cross-domain shifts. We address these challenges through the lens of structure-aware context selection. We propose TL-GPSTGN, a transfer-oriented spatiotemporal framework that enhances sample efficiency and out-of-distribution generalization by selectively pruning non-optimized graph context. Specifically, our method employs information-theoretic and correlation-based criteria to extract structurally informative subgraphs and features, resulting in a compact, semantically grounded representation. This optimized context is subsequently integrated into a spatiotemporal convolutional architecture to capture complex multivariate dynamics. Evaluations on large-scale traffic benchmarks demonstrate that TL-GPSTGN consistently outperforms baselines in low-data transfer scenarios. Our findings suggest that explicit context pruning serves as a powerful inductive bias for improving the robustness of graph-based forecasting models.

图神经网络时间序列模型剪枝迁移学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。