用图卡尔曼网络建模图上时序数据,实现高效预测与补全。
GKNet: Graph Kalman Filtering and Model Inference via Model-based Deep Learning
- 基于图结构设计状态空间模型,融合图滤波与随机微分方程。
- 在部分观测下可实现高精度预测与缺失值补全,支持多跳邻域影响。
- 适合城市供水、神经网络等复杂图时序数据建模任务。
面向城市供水网络、经济学和脑网络等场景中的图上时序推理任务,本文提出一种图感知的状态空间模型。该模型通过参数化的图诱导状态方程和观测方程,联合捕捉图-时序模式。状态方程采用图边驱动的随机偏微分方程,引入噪声以建模边不确定性并增加自由度;噪声分布受图结构调节,使状态可在局部区域偏离标准过程。观测模型为状态的采样与图滤波版本,捕获多跳邻居影响。目标是从部分观测数据中学习两模型参数,用于预测与数据补全。初始通过最大似然法推断,具理论可分析性但表达能力与扩展性有限。为此,基于状态空间形式构建了类卡尔曼神经网络的深度学习架构,实现参数与状态的端到端联合学习。
原文摘要 · Abstract (English)
Inference tasks with time series over graphs are of importance in applications such as urban water networks, economics, and networked neuroscience. Addressing these tasks typically relies on identifying a computationally affordable model that jointly captures the graph-temporal patterns of the data. In this work, we propose a graph-aware state space model for graph time series, where both the latent state and the observation equation are parametric graph-induced models with a limited number of parameters that need to be learned. More specifically, we consider the state equation to follow a stochastic partial differential equation driven by noise over the graphs edges accounting not only for potential edge uncertainties but also for increasing the degrees of freedom in the latter in a tractable manner. The graph structure conditioning of the noise dispersion allows the state variable to deviate from the stochastic process in certain neighborhoods. The observation model is a sampled and graph-filtered version of the state capturing multi-hop neighboring influence. The goal is to learn the parameters in both state and observation models from the partially observed data for downstream tasks such as prediction and imputation. The model is inferred first through a maximum likelihood approach that provides theoretical tractability but is limited in expressivity and scalability. To improve on the latter, we use the state-space formulation to build a principled deep learning architecture that jointly learns the parameters and tracks the state in an end-to-end manner in the spirit of Kalman neural networks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。