{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/equivariant-neural-simulators-for-stochastic-1","title":"Equivariant Neural Simulators for Stochastic Spatiotemporal Dynamics","arxiv_id":"2305.14286","date":"2023-05-23","proceeding":"NeurIPS 2023 11","authors":["Koen Minartz","Yoeri Poels","Simon Koop","Vlado Menkovski"],"abstract":"Neural networks are emerging as a tool for scalable data-driven simulation of high-dimensional dynamical systems, especially in settings where numerical methods are infeasible or computationally expensive. Notably, it has been shown that incorporating domain symmetries in deterministic neural simulators can substantially improve their accuracy, sample efficiency, and parameter efficiency. However, to incorporate symmetries in probabilistic neural simulators that can simulate stochastic phenomena, we need a model that produces equivariant distributions over trajectories, rather than equivariant function approximations. In this paper, we propose Equivariant Probabilistic Neural Simulation (EPNS), a framework for autoregressive probabilistic modeling of equivariant distributions over system evolutions. We use EPNS to design models for a stochastic n-body system and stochastic cellular dynamics. Our results show that EPNS considerably outperforms existing neural network-based methods for probabilistic simulation. More specifically, we demonstrate that incorporating equivariance in EPNS improves simulation quality, data efficiency, rollout stability, and uncertainty quantification. We conclude that EPNS is a promising method for efficient and effective data-driven probabilistic simulation in a diverse range of domains.","url_abs":"https://arxiv.org/abs/2305.14286v3","url_pdf":"https://arxiv.org/pdf/2305.14286v3.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"equivariant-neural-simulators-for-stochastic-1","repo_url":"https://github.com/kminartz/epns","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"uncertainty-quantification","task_name":"Uncertainty Quantification"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2305.14286","atlas_url":"https://app.syntology.ai/?focus=2305.14286","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2305.14286"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"deterministic:regex_extraction","url":"https://github.com/kminartz/EPNS","reach":{"status":"ok","spdx":"MIT"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/kminartz/epns","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"unverified":5},"by_repo_kind":{"official":{"samples":5,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"87cc1a538ca9f863","entry":"arr_to_im","repo":"kminartz/EPNS","repo_kind":"official","path":"utils.py","file_url":"https://github.com/kminartz/EPNS/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"87cc1a538ca9f863"}},{"code_sha256_prefix":"87bab19f49c868f7","entry":"get_data_loaders","repo":"kminartz/EPNS","repo_kind":"official","path":"datasets/Cell_Combined_Dataset.py","file_url":"https://github.com/kminartz/EPNS/blob/HEAD/datasets/Cell_Combined_Dataset.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"87bab19f49c868f7"}},{"code_sha256_prefix":"a77c39a49bdc63f5","entry":"get_data_loaders","repo":"kminartz/EPNS","repo_kind":"official","path":"datasets/N_Body_Dataset.py","file_url":"https://github.com/kminartz/EPNS/blob/HEAD/datasets/N_Body_Dataset.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a77c39a49bdc63f5"}},{"code_sha256_prefix":"ee1f8489cb2b747b","entry":"plot_NS_image","repo":"kminartz/EPNS","repo_kind":"official","path":"utils.py","file_url":"https://github.com/kminartz/EPNS/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"ee1f8489cb2b747b"}},{"code_sha256_prefix":"4a00a6ed6a166e25","entry":"plot_cell_image","repo":"kminartz/EPNS","repo_kind":"official","path":"utils.py","file_url":"https://github.com/kminartz/EPNS/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4a00a6ed6a166e25"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}