Search papers, labs, and topics across Lattice.
The Graph Mamba Operator (GraMO) integrates state-space models with graph-based interaction learning to effectively model interacting dynamical systems by coupling spatial and temporal dynamics within a single recurrence. This approach overcomes the limitations of traditional methods that treat spatial and temporal updates separately, which often leads to error accumulation over long horizons. Evaluations on N-body systems, motion capture, and robotics datasets demonstrate that GraMO achieves the lowest prediction error and significantly improves long-horizon forecasting capabilities.
GraMO achieves unprecedented accuracy in long-horizon predictions by seamlessly integrating spatial and temporal dynamics in a single model.
Modeling interacting dynamical systems requires capturing spatial interactions alongside long-range temporal dependencies. Graph neural networks (GNNs) provide a natural representation but typically rely on autoregressive rollouts and treat spatial and temporal dynamics separately, leading to error accumulation over long horizons. Existing approaches also focus on local interactions and short temporal contexts, limiting their ability to capture multi-hop dependencies and global structure. We introduce the Graph Mamba Operator (GraMO), a latent-space simulator that integrates state-space models with graph-based interaction learning. In contrast to prior work that sequences nodes or applies spatial and temporal updates in separate stages, GraMO couples graph-based interactions and temporal state updates within a single recurrence. The update is linear in the latent state, with input-dependent coefficients that adapt across regimes. We evaluate GraMO on N-body systems, motion capture, and robotics datasets, achieving the lowest error across benchmarks and the largest gains in long-horizon prediction.
