Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces ST-Sheaf GNN, a novel spatio-temporal graph neural network that models information flow over locally structured spaces using sheaf-theoretic vector spaces and learned linear restriction maps. By learning dynamic restriction maps that adapt to local spatio-temporal patterns, the model mitigates oversmoothing and enables more expressive interactions compared to traditional message passing approaches. Experiments on real-world datasets demonstrate state-of-the-art performance, highlighting the benefits of sheaf-theoretic topological representations for spatio-temporal forecasting.
Ditch the global alignment: ST-Sheaf GNN learns dynamic, localized information flow for spatio-temporal forecasting, outperforming state-of-the-art methods by explicitly modeling latent local structure.
Spatio-temporal systems often exhibit highly heterogeneous and non-intuitive responses to localized disruptions, limiting the effectiveness of conventional message passing approaches in modeling higher-order interactions under local heterogeneity. This paper reformulates spatio-temporal forecasting as the problem of learning information flow over locally structured spaces, rather than propagating globally aligned node representations. We introduce a spatio-temporal sheaf diffusion graph neural network (ST-Sheaf GNN) that embeds graph topology into sheaf-theoretic vector spaces connected by learned linear restriction maps. Unlike prior work that relies on static or globally shared transformations, our model learns dynamic restriction maps that evolve over time and adapt to local spatio-temporal patterns to enable substantially more expressive interactions. By explicitly modeling latent local structure, the proposed framework efficiently mitigates the oversmoothing phenomenon in deep GNN architectures. We evaluate our framework on a diverse set of real-world spatio-temporal forecasting benchmarks spanning multiple domains. Experimental results demonstrate state-of-the-art performance, highlighting the effectiveness of sheaf-theoretic topological representations as a powerful foundation for spatio-temporal graph learning. The code is available at: https://anonymous.4open.science/r/ST-SheafGNN-6523/.