arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04071cs.LGcs.AI

驯化基于图的时空预测中的局部效应

Taming Local Effects in Graph-based Spatiotemporal Forecasting

The Swiss AI Lab IDSIA USI-SUPSI, Università della Svizzera italiana(瑞士人工智能实验室 IDSIA USI-SUPSI,瑞士意大利语大学);Politecnico di Milano(米兰理工大学)

Andrea Cini, Ivan Marisca, Daniele Zambon, Cesare Alippi

AI总结:

本文研究基于图的时空预测中全局与局部效应的平衡,提出将可训练节点嵌入用于摊销专门组件学习,以结合共享层与节点特定参数并支持模型迁移,经验证明其提升预测准确性。

摘要:

时空图神经网络已被证明在时间序列预测应用中有效,在多种设置下其性能优于标准单变量预测器。这些架构利用图结构和关系归纳偏置来学习一个单一的(全局)归纳模型,以预测任意数量的输入时间序列,每个时间序列与一个图节点相关联。尽管在计算和数据效率方面相对于拟合一组局部模型有所提升,但当某些时间序列由不同的时空随机过程生成时,依赖单一全局模型可能成为局限。本文的主要目标是理解基于图的时空预测中全局性与局部性之间的相互作用,同时提出一个方法论框架,以合理化在此类架构中包含可训练节点嵌入的做法。我们将可训练节点嵌入的作用归因于摊销专门组件的学习。此外,嵌入允许1)有效结合共享消息传递层与节点特定参数的优点,以及2)将学习到的模型高效迁移到新的节点集。在强有力的经验证据支持下,我们提供了将基于图的模型专门化到每个时间序列动态的见解和指导,并展示了这一方面在获得准确预测中起到的关键作用。

英文摘要:

Spatiotemporal graph neural networks have shown to be effective in time series forecasting applications, achieving better performance than standard univariate predictors in several settings. These architectures take advantage of a graph structure and relational inductive biases to learn a single (global) inductive model to predict any number of the input time series, each associated with a graph node. Despite the gain achieved in computational and data efficiency w.r.t. fitting a set of local models, relying on a single global model can be a limitation whenever some of the time series are generated by a different spatiotemporal stochastic process. The main objective of this paper is to understand the interplay between globality and locality in graph-based spatiotemporal forecasting, while contextually proposing a methodological framework to rationalize the practice of including trainable node embeddings in such architectures. We ascribe to trainable node embeddings the role of amortizing the learning of specialized components. Moreover, embeddings allow for 1) effectively combining the advantages of shared message-passing layers with node-specific parameters and 2) efficiently transferring the learned model to new node sets. Supported by strong empirical evidence, we provide insights and guidelines for specializing graph-based models to the dynamics of each time series and show how this aspect plays a crucial role in obtaining accurate predictions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑