{"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/simulate-time-integrated-coarse-grained","title":"Simulate Time-integrated Coarse-grained Molecular Dynamics with Multi-Scale Graph Networks","arxiv_id":"2204.10348","date":"2022-04-21","proceeding":null,"authors":["Xiang Fu","Tian Xie","Nathan J. Rebello","Bradley D. Olsen","Tommi Jaakkola"],"abstract":"Molecular dynamics (MD) simulation is essential for various scientific domains but computationally expensive. Learning-based force fields have made significant progress in accelerating ab-initio MD simulation but are not fast enough for many real-world applications due to slow inference for large systems and small time steps (femtosecond-level). We aim to address these challenges by learning a multi-scale graph neural network that directly simulates coarse-grained MD with a very large time step (nanosecond-level) and a novel refinement module based on diffusion models to mitigate simulation instability. The effectiveness of our method is demonstrated in two complex systems: single-chain coarse-grained polymers and multi-component Li-ion polymer electrolytes. For evaluation, we simulate trajectories much longer than the training trajectories for systems with different chemical compositions that the model is not trained on. Structural and dynamical properties can be accurately recovered at several orders of magnitude higher speed than classical force fields by getting out of the femtosecond regime.","url_abs":"https://arxiv.org/abs/2204.10348v3","url_pdf":"https://arxiv.org/pdf/2204.10348v3.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":"simulate-time-integrated-coarse-grained","repo_url":"https://github.com/kyonofx/mlcgmd","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"machine-learning","task_name":"BIG-bench Machine Learning"},{"task_slug":"graph-clustering","task_name":"Graph Clustering"},{"task_slug":"graph-neural-network","task_name":"Graph Neural Network"}],"methods":[{"method_slug":"graph-neural-network","method_name":"Graph Neural Network"},{"method_slug":"speed","method_name":"SPEED"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2204.10348","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2204.10348"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+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":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/kyonofx/mlcgmd","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"unverified":10},"by_repo_kind":{"official":{"samples":10,"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":"afc340d2ad491785","entry":"build_mlp","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/model/networks.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/model/networks.py","link_basis":"harvester_set","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":"afc340d2ad491785"}},{"code_sha256_prefix":"dd895221e353d11a","entry":"compute_weighted_rgs","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/model/utils.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/model/utils.py","link_basis":"harvester_set","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":"dd895221e353d11a"}},{"code_sha256_prefix":"32638560701fb713","entry":"get_diffusivity","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/preprocess/battery.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/preprocess/battery.py","link_basis":"harvester_set","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":"32638560701fb713"}},{"code_sha256_prefix":"25a60b695464e797","entry":"get_env","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/common/env_init.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/common/env_init.py","link_basis":"harvester_set","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":"25a60b695464e797"}},{"code_sha256_prefix":"75ee0ec28e00f3c7","entry":"get_position_noise","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/model/utils.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/model/utils.py","link_basis":"harvester_set","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":"75ee0ec28e00f3c7"}},{"code_sha256_prefix":"3622c4ba3fcb2e13","entry":"get_position_noise_with_velocity","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/model/utils.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/model/utils.py","link_basis":"harvester_set","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":"3622c4ba3fcb2e13"}},{"code_sha256_prefix":"15ab93e2f8189754","entry":"get_scaler_from_data_list","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/common/modules.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/common/modules.py","link_basis":"harvester_set","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":"15ab93e2f8189754"}},{"code_sha256_prefix":"477ded4048ac5f14","entry":"load_bond_info","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/preprocess/battery.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/preprocess/battery.py","link_basis":"harvester_set","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":"477ded4048ac5f14"}},{"code_sha256_prefix":"c42cd744570e4fa9","entry":"read_rg","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/preprocess/chain.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/preprocess/chain.py","link_basis":"harvester_set","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":"c42cd744570e4fa9"}},{"code_sha256_prefix":"206c54c7ec936d17","entry":"reducer","repo":"kyonofx/mlcgmd","repo_kind":"official","path":"graphwm/model/networks.py","file_url":"https://github.com/kyonofx/mlcgmd/blob/HEAD/graphwm/model/networks.py","link_basis":"harvester_set","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":"206c54c7ec936d17"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}