Structural Inference of Dynamical Systems with Conjoined State Space Models
Aoran Wang, Jun Pang
Abstract
This paper introduces SICSM, a novel structural inference framework that integrates Selective State Space Models (selective SSMs) with Generative Flow Networks (GFNs) to handle the challenges posed by dynamical systems with irregularly sampled trajectories and partial observations. By utilizing the robust temporal modeling capabilities of selective SSMs, our approach learns input-dependent transition functions that adapt to non-uniform time intervals, thereby enhancing the accuracy of structural inference. By aggregating dynamics across diverse temporal dependencies and channeling them into the GFN, the SICSM adeptly approximates the posterior distribution of the system’s structure. This process not only enables precise inference of complex interactions within partially observed systems but also ensures the seamless integration of prior knowledge, enhancing the model’s accuracy and robustness. Extensive evaluations on sixteen diverse datasets demonstrate that SICSM outperforms existing methods, particularly in scenarios characterized by irregular sampling and incomplete observations, which highlight its potential as a reliable tool for scientific discovery and system diagnostics in disciplines that demand precise modeling of complex interactions.
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 papers3
- From Uniform to Learned Graph Priors: Diffusion for Structure DiscoveryQi Shao, Hao Guo, Jiawen Chen, Duxin Chen et al.KDD 2026 · 2 citations
- IPSI: Enhancing Structural Inference with Automatically Learned Structural PriorsZhongben Gong, Xiaoqun Wu, Mingyang ZhouNeurIPS 2025 · 1 citation
- Guided Structural Inference: Leveraging Priors with Soft Gating MechanismsAoran Wang, Xinnan Dai, Jun PangICML 2025
Builds on22
- Efficiently Modeling Long Sequences with Structured State SpacesAlbert Gu, Karan Goel, Christopher RéICLR 2022 · 3,482 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
- GraphSAINT: Graph Sampling Based Inductive Learning MethodHanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava, Rajgopal Kannan et al.ICLR 2020 · 1,155 citations
- HiPPO: Recurrent Memory with Optimal Polynomial ProjectionsAlbert Gu, Tri Dao, Stefano Ermon, Atri Rudra et al.NeurIPS 2020 · 1,100 citations
- Flow Network based Generative Models for Non-Iterative Diverse Candidate GenerationEmmanuel Bengio, Moksh Jain, Maksym Korablyov, Doina Precup et al.NeurIPS 2021 · 565 citations
Related papers
- Free-Form Variational Inference for Gaussian Process State-Space ModelsXuhui Fan, Edwin V. Bonilla, Terence J. O'Kane, Scott A. SissonICML 2023 · 13 citations
- DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNetsLazar Atanackovic, Alexander Tong, Bo Wang, Leo J. Lee et al.NeurIPS 2023 · 37 citations
- Learning Continuous System Dynamics from Irregularly-Sampled Partial ObservationsZijie Huang, Yizhou Sun, Wei WangNeurIPS 2020 · 103 citations
- Neural structure learning with stochastic differential equationsBenjie Wang, Joel Jennings, Wenbo GongICLR 2024 · 10 citations
- Sequential Monte Carlo Learning for Time Series Structure DiscoveryFeras Saad, Brian Patton, Matthew Douglas Hoffman, Rif A. Saurous et al.ICML 2023 · 14 citations
