arXiv:2606.06682cs.LG2026-06中稿 · ICML

用图结构先验改进时空数据补全,速度更快更准。

Spatiotemporal Imputation with Graph-Informed Flow Matching

论文配图:Spatiotemporal Imputation with Graph-Informed Flow Matching
图 1 · 摘自论文原文
  • 用时空滤波构建图结构先验,替代传统高斯先验
  • 在真实与合成数据集上均优于当前最佳方法
  • 适合处理交通、空气质量等时空缺失数据问题

时空系统中的缺失数据是常见挑战,如空气质量监测和城市交通管理。传统机器学习方法(如循环神经网络和图神经网络)依赖迭代传播,易积累时空误差。近期基于扩散的方法虽缓解了误差传播,但需迭代采样且依赖通用高斯先验,限制了效率与效果。为此,我们提出GiFlow:一种图结构引导的流匹配框架用于时空补全。该方法以可观测信号的时空滤波结果构建图结构先验,更贴近目标分布,简化生成轨迹。流场由融合空间注意力、时间注意力和时空传播的混合向量场建模,实现时空依赖的联合捕捉。在合成与真实数据集上的大量实验表明,GiFlow显著优于现有先进方法。代码已开源:https://github.com/zepengzhang/GiFlow。

原文摘要 · Abstract (English)

Missing data is a common challenge in spatiotemporal systems, arising in applications such as air quality monitoring and urban traffic management. Traditional machine learning approaches, like recurrent and graph neural networks, rely on iterative propagation, which tends to accumulate errors over time and space. Recent diffusion-based methods mitigate error propagation but require iterative sampling and often depend on problem-agnostic Gaussian priors, limiting both efficiency and effectiveness. To address these limitations, we propose GiFlow, a Graph-Informed Flow Matching framework for spatiotemporal imputation. GiFlow replaces the typical Gaussian prior with a graph-informed prior constructed via spatiotemporal filtering of observable signals, which better aligns the source distribution to the target and thereby simplifies the generation trajectory. The flow field is parameterized by a hybrid vector field model that integrates spatial attention, temporal attention, and spatiotemporal propagation, enabling joint modeling of spatial and temporal dependencies. Extensive experiments on both synthetic and real-world datasets demonstrate that the proposed GiFlow outperforms the state-of-the-art approaches in spatiotemporal imputation. The code is available at https://github.com/zepengzhang/GiFlow.

时空补全流匹配图神经网络

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