Rethink GraphODE Generalization within Coupled Dynamical System
Guancheng Wan, Zijie Huang, Wanjia Zhao, Xiao Luo, Yizhou Sun, Wei Wang
Abstract
Coupled dynamical systems govern essential phenomena across physics, biology, and engineering, where components interact through complex dependencies. While Graph Ordinary Differential Equations (GraphODE) offer a powerful framework to model these systems, their generalization capabilities degrade severely under limited observational training data due to two fundamental flaws: (i) the entanglement of static attributes and dynamic states in the initialization process, and (ii) the reliance on context-specific coupling patterns during training, which hinders performance in unseen scenarios. In this paper, we propose a Generalizable GraphODE with disentanglement and regularization (GREAT) to address these challenges. Through systematic analysis via the Structural Causal Model, we identify backdoor paths that undermine generalization and design two key modules to mitigate their effects. The Dynamic-Static Equilibrium Decoupler (DyStaED) disentangles static and dynamic states via orthogonal subspace projections, ensuring robust initialization. Furthermore, the Causal Mediation for Coupled Dynamics (CMCD) employs variational inference to estimate latent causal factors, reducing spurious correlations and enhancing universal coupling dynamics. Extensive experiments across diverse dynamical systems demonstrate that ours outperforms state-of-the-art methods within both in-distribution and out-of-distribution.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers7
- HYPERION: Fine-Grained Hypersphere Alignment for Robust Federated Graph LearningFrank Wan, Xiaoran Shang, Yuxin Wu, Guibin Zhang et al.NeurIPS 2025 · 4 citations
- Riemannian Liquid Spatio-Temporal Graph NetworkLiangsi Lu, Jingchao Wang, Zhaorong Dai, Hanqian Liu et al.WWW 2026 · 1 citation
- EAGLES: Towards Effective, Efficient, and Economical Federated Graph Learning via Unified SparsificationZitong Shi, Guancheng Wan, Wenke Huang, Guibin Zhang et al.ICML 2025
- S2FGL: Spatial Spectral Federated Graph LearningZihan Tan, Suyuan Huang, Guancheng Wan, Wenke Huang et al.ICML 2025
- GHOST: Generalizable One-Shot Federated Graph Learning with Proxy-Based Topology Knowledge RetentionJiaru Qian, Guancheng Wan, Wenke Huang, Guibin Zhang et al.ICML 2025
Builds on21
- Graph Contrastive Learning with AugmentationsYuning You, Tianlong Chen, Yongduo Sui, Ting Chen et al.NeurIPS 2020 · 3,042 citations
- Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data ForecastingChao Song, Youfang Lin, Shengnan Guo, Huaiyu WanAAAI 2020 · 1,659 citations
- InfoGCL: Information-Aware Graph Contrastive LearningDongkuan Xu, Wei Cheng, Dongsheng Luo, Haifeng Chen et al.NeurIPS 2021 · 261 citations
- Learning Continuous System Dynamics from Irregularly-Sampled Partial ObservationsZijie Huang, Yizhou Sun, Wei WangNeurIPS 2020 · 103 citations
- Block Modeling-Guided Graph Convolutional Neural NetworksDongxiao He, Chundong Liang, Huixin Liu, Mingxiang Wen et al.AAAI 2022 · 85 citations
Related papers
- PGODE: Towards High-quality System Dynamics ModelingXiao Luo, Yiyang Gu, Huiyu Jiang, Hang Zhou et al.ICML 2024 · 11 citations
- Interpretable Neural ODEs for Gene Regulatory Network Discovery under PerturbationsZaikang Lin, Sei Chang, Aaron Zweig, Minseo Kang et al.ICML 2026 · 8 citations
- CARE: Modeling Interacting Dynamics Under Temporal Environmental VariationXiao Luo, Haixin Wang, Zijie Huang, Huiyu Jiang et al.NeurIPS 2023 · 13 citations
- Prometheus: Out-of-distribution Fluid Dynamics Modeling with Disentangled Graph ODEHao Wu, Huiyuan Wang, Kun Wang, Weiyan Wang et al.ICML 2024 · 25 citations
- Generalizing Graph ODE for Learning Complex System Dynamics across EnvironmentsZijie Huang, Yizhou Sun, Wei WangKDD 2023 · 16 citations
