Neural graphical modelling in continuous-time: consistency guarantees and algorithms
Alexis Bellot, Kim Branson, Mihaela van der Schaar
Abstract
The discovery of structure from time series data is a key problem in fields of study working with complex systems. Most identifiability results and learning algorithms assume the underlying dynamics to be discrete in time. Comparatively few, in contrast, explicitly define dependencies in infinitesimal intervals of time, independently of the scale of observation and of the regularity of sampling. In this paper, we consider score-based structure learning for the study of dynamical systems. We prove that for vector fields parameterized in a large class of neural networks, least squares optimization with adaptive regularization schemes consistently recovers directed graphs of local independencies in systems of stochastic differential equations. Using this insight, we propose a score-based learning algorithm based on penalized Neural Ordinary Differential Equations (modelling the mean process) that we show to be applicable to the general setting of irregularly-sampled multivariate time series and to outperform the state of the art across a range of dynamical systems.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1a760e01-cd21-446a-9bc4-e58d95e69779Cited by top-tier papers20
- CUTS+: High-Dimensional Causal Discovery from Irregular Time-SeriesYuxiao Cheng, Lianglong Li, Tingxiong Xiao, Zongren Li et al.AAAI 2024 · 58 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
- CausalTime: Realistically Generated Time-series for Benchmarking of Causal DiscoveryYuxiao Cheng, Ziqian Wang, Tingxiong Xiao, Qin Zhong et al.ICLR 2024 · 35 citations
- CausalStock: Deep End-to-end Causal Discovery for News-driven Multi-stock Movement PredictionShuqi Li, Yuebo Sun, Yuxin Lin, Xin Gao et al.NeurIPS 2024 · 25 citations
- Jacobian Regularizer-based Neural Granger CausalityWanqi Zhou, Shuanghao Bai, Shujian Yu, Qibin Zhao et al.ICML 2024 · 22 citations
Builds on6
- Economy Statistical Recurrent Units For Inferring Nonlinear Granger CausalitySaurabh Khanna, Vincent Y. F. TanICLR 2020 · 93 citations
- Interpretable Models for Granger Causality Using Self-explaining Neural NetworksRicards Marcinkevics, Julia E. VogtICLR 2021 · 84 citations
- Consistent feature selection for analytic deep neural networksVu C. Dinh, Lam Si Tung HoNeurIPS 2020 · 66 citations
- ODIN: ODE-Informed Regression for Parameter and State Inference in Time-Continuous Dynamical SystemsPhilippe Wenk, Gabriele Abbati, Michael A. Osborne, Bernhard Schölkopf et al.AAAI 2020 · 33 citations
- Learning Neural Event Functions for Ordinary Differential EquationsRicky T. Q. Chen, Brandon Amos, Maximilian NickelICLR 2021 · 24 citations
Related papers
- Neural structure learning with stochastic differential equationsBenjie Wang, Joel Jennings, Wenbo GongICLR 2024 · 10 citations
- Graph Neural Flows for Unveiling Systemic Interactions Among Irregularly Sampled Time SeriesGiangiacomo Mercatali, André Freitas, Jie ChenNeurIPS 2024 · 17 citations
- Learning continuous-time PDEs from sparse data with graph neural networksValerii Iakovlev, Markus Heinonen, Harri LähdesmäkiICLR 2021 · 81 citations
- DyCAST: Learning Dynamic Causal Structure from Time SeriesYue Cheng, Bochen Lyu, Weiwei Xing, Zhanxing ZhuICLR 2025
- Anamnesic Neural Differential Equations with Orthogonal Polynomial ProjectionsEdward De Brouwer, Rahul G. KrishnanICLR 2023 · 2 citations
