用动态拓扑结构建模时空图,提升异质性数据预测效果。
Dynamic Sheaf Diffusion Networks with Adaptive Local Structure for Heterogeneous Spatio-Temporal Graph Learning

- 将图结构嵌入分层向量空间,通过动态映射学习局部变化模式。
- 在多领域真实数据上达到当前最优性能,深度扩展下仍保持区分力。
- 适合处理具有复杂局部动态的时空预测任务,如交通、气象建模。
时空过程常对局部扰动表现出高度异质且非直观的响应,限制了传统消息传递方法对局部异质性的建模能力。本文将时空预测重新定义为在局部结构化空间中学习信息流动的问题,而非传播全局对齐的节点表示。为此,提出一种基于层积张量的时空层积图神经网络(ST-Sheaf GNN),将图拓扑嵌入由可学习线性限制映射连接的层积向量空间。与依赖静态或全局共享变换的先前方法不同,本模型学习随时间演化的动态限制映射,能自适应局部时空模式,实现更丰富的交互。所提框架在理论上保证并实证证明其扩散机制可缓解过度平滑问题,在增加扩散层数时仍能保持判别性节点表示。在多个领域的多种真实世界时空预测基准上实验表明,该方法表现优于现有技术,凸显了层积拓扑表示作为时空图学习原理基础的有效性。代码已公开:https://anonymous.4open.science/r/ST-SheafGNN-6523/。
原文摘要 · Abstract (English)
Spatio-temporal processes often exhibit highly heterogeneous and non-intuitive responses to localized disruptions, limiting the effectiveness of conventional message passing approaches in modeling local heterogeneity. We reformulate spatio-temporal forecasting as the problem of learning information flow over locally structured spaces, rather than propagating globally aligned node representations. To this end, we introduce a spatio-temporal sheaf diffusion graph neural network (ST-Sheaf GNN) that embeds graph topology into sheaf-based vector spaces connected by learned linear restriction maps. Unlike prior approaches relying on static or globally shared transformations, our model learns dynamic restriction maps that evolve over time and adapt to local spatio-temporal patterns, enabling more expressive interactions. The proposed framework both theoretically guarantees and empirically demonstrates evidence that the proposed diffusion mechanism mitigates oversmoothing, preserving discriminative node representations even with increasing diffusion layer depth. Experiments on diverse real-world spatio-temporal forecasting benchmarks across multiple domains demonstrate state-of-the-art performance, highlighting the effectiveness of sheaf topological representations as a principled foundation for spatio-temporal graph learning. The code is available at: https://anonymous.4open.science/r/ST-SheafGNN-6523/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。