PGODE: Towards High-quality System Dynamics Modeling
Not on the map. Why?
Abstract (arXiv)
This paper studies the problem of modeling multi-agent dynamical systems, where agents could interact mutually to influence their behaviors. Recent research predominantly uses geometric graphs to depict these mutual interactions, which are then captured by powerful graph neural networks (GNNs). However, predicting interacting dynamics in challenging scenarios such as out-of-distribution shift and complicated underlying rules remains unsolved. In this paper, we propose a new approach named Prototypical Graph ODE (PGODE) to address the problem. The core of PGODE is to incorporate prototype decomposition from contextual knowledge into a continuous graph ODE framework. Specifically, PGODE employs representation disentanglement and system parameters to extract both object-level and system-level contexts from historical trajectories, which allows us to explicitly model their independent influence and thus enhances the generalization capability under system changes. Then, we integrate these disentangled latent representations into a graph ODE model, which determines a combination of various interacting prototypes for enhanced model expressivity. The entire model is optimized using an end-to-end variational inference framework to maximize the likelihood. Extensive experiments in both in-distribution and out-of-distribution settings validate the superiority of PGODE compared to various baselines.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 73 of its 105 references that are in the map. First posted on arXiv 11 Nov 2023.
Builds on
73 of 105 references.
References · 73
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,684
- 2019An Introduction to Variational Autoencoderscited by 997
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
38 more
- 2020Learning Mesh-Based Simulation with Graph Networkscited by 840
- 2022Mixture-of-Experts with Expert Choice Routingcited by 611
- 2020Adaptive Universal Generalized PageRank Graph Neural Networkcited by 592
- 2019Augmented Neural ODEscited by 396
- 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 365
- 2018Neural Relational Inference for Interacting Systemscited by 339
- 2021AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecastingcited by 327
- 2020Inductive Biases for Deep Learning of Higher-Level Cognitioncited by 273
- 2020Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Predictioncited by 218
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 205
- 2022Stochastic Trajectory Prediction via Motion Indeterminacy Diffusioncited by 184
- 2021Two Sides of the Same Coin: Heterophily and Oversmoothing in Graph Convolutional Neural Networkscited by 182
- 2022Finding Global Homophily in Graph Neural Networks When Meeting Heterophilycited by 179
- 2022Graph Neural Networks for Graphs with Heterophily: A Surveycited by 175
- 2022Revisiting Heterophily For Graph Neural Networkscited by 169
- 2019Graph Neural Ordinary Differential Equationscited by 153
- 2019Continuous Graph Neural Networkscited by 114
- 2021Modeling Irregular Time Series with Continuous Recurrent Unitscited by 101
- 2023EqMotion: Equivariant Multi-agent Motion Prediction with Invariant Interaction Reasoningcited by 97
- 2021Breaking the Limit of Graph Neural Networks by Improving the Assortativity of Graphs with Local Mixing Patternscited by 95
- 2022Convolutional Neural Networks on Graphs with Chebyshev Approximation, Revisitedcited by 87
- 2020SDE-Net: Equipping Deep Neural Networks with Uncertainty Estimatescited by 75
- 2022Multivariate Time Series Forecasting with Dynamic Graph Neural ODEscited by 72
- 2023A Comprehensive Survey on Deep Graph Representation Learningcited by 72
- 2023Graph Mixture of Experts: Learning on Large-Scale Graphs with Explicit Diversity Modelingcited by 63
- 2021Encoding physics to learn reaction-diffusion processescited by 61
- 2021MG-GAN: A Multi-Generator Model Preventing Out-of-Distribution Samples in Pedestrian Trajectory Predictioncited by 61
- 2020Learning Continuous System Dynamics from Irregularly-Sampled Partial Observationscited by 60
- 2022An Efficient Domain-Incremental Learning Approach to Drive in All Weather Conditionscited by 54
- 2022Graph Pooling for Graph Neural Networks: Progress, Challenges, and Opportunitiescited by 52
- 2023From Zero to Turbulence: Generative Modeling for 3D Flow Simulationcited by 46
- 2022ADATIME: A Benchmarking Suite for Domain Adaptation on Time Series Datacited by 45
- 2021MALI: A memory efficient and reverse accurate integrator for Neural ODEscited by 42
- 2021Heavy Ball Neural Ordinary Differential Equationscited by 42
- 2022Learning Physical Dynamics with Subequivariant Graph Neural Networkscited by 38
- 2023Taming Local Effects in Graph-based Spatiotemporal Forecastingcited by 37
- 2020Evaluating Robustness to Input Perturbations for Neural Machine Translationcited by 36
- 2020"Hey, that's not an ODE": Faster ODE Adjoints via Seminormscited by 35
50 most cited of 73 All 73 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 2; 2026: 2.
7 papers.
- 2025Graph ODEs and Beyond: A Comprehensive Survey on Integrating Differential Equations with Graph Neural Networkscited by 14
- 2025STFlow: Data-Coupled Flow Matching for Geometric Trajectory Simulationcited by 7
- 2024System-Aware Neural ODE Processes for Few-Shot Bayesian Optimizationcited by 3
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamicscited by 2
- 2024Learning System Dynamics without Forgettingcited by 1
- 2026DGNet: Discrete Green Networks for Data-Efficient Learning of Spatiotemporal PDEscited by 1
- 2026Generalizing Dynamics Modeling More Easily from Representation Perspectivecited by 0
- 2026Generalizing Dynamics Modeling More Easily from Representation Perspective24 Mar 2026
- 2026DGNet: Discrete Green Networks for Data-Efficient Learning of Spatiotemporal PDEs2 Mar 2026
- 2025STFlow: Data-Coupled Flow Matching for Geometric Trajectory Simulation24 May 2025
- 2025Graph ODEs and Beyond: A Comprehensive Survey on Integrating Differential Equations with Graph Neural Networks29 Mar 2025
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics3 Nov 2024
- 2024Learning System Dynamics without Forgetting30 Jun 2024
- 2024System-Aware Neural ODE Processes for Few-Shot Bayesian Optimization4 Jun 2024