{"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/higher-order-equivariant-neural-networks-for","title":"Higher-Order Equivariant Neural Networks for Charge Density Prediction in Materials","arxiv_id":"2312.05388","date":"2023-12-08","proceeding":null,"authors":["Teddy Koker","Keegan Quigley","Eric Taw","Kevin Tibbetts","Lin Li"],"abstract":"The calculation of electron density distribution using density functional theory (DFT) in materials and molecules is central to the study of their quantum and macro-scale properties, yet accurate and efficient calculation remains a long-standing challenge. We introduce ChargE3Net, an E(3)-equivariant graph neural network for predicting electron density in atomic systems. ChargE3Net enables the learning of higher-order equivariant feature to achieve high predictive accuracy and model expressivity. We show that ChargE3Net exceeds the performance of prior work on diverse sets of molecules and materials. When trained on the massive dataset of over 100K materials in the Materials Project database, our model is able to capture the complexity and variability in the data, leading to a significant 26.7% reduction in self-consistent iterations when used to initialize DFT calculations on unseen materials. Furthermore, we show that non-self-consistent DFT calculations using our predicted charge densities yield near-DFT performance on electronic and thermodynamic property prediction at a fraction of the computational cost. Further analysis attributes the greater predictive accuracy to improved modeling of systems with high angular variations. These results illuminate a pathway towards a machine learning-accelerated ab initio calculations for materials discovery.","url_abs":"https://arxiv.org/abs/2312.05388v2","url_pdf":"https://arxiv.org/pdf/2312.05388v2.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":"higher-order-equivariant-neural-networks-for","repo_url":"https://github.com/AIforGreatGood/charge3net","is_official":1,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}},{"paper_slug":"higher-order-equivariant-neural-networks-for","repo_url":"https://github.com/teddykoker/e3nn.c","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"graph-neural-network","task_name":"Graph Neural Network"},{"task_slug":"property-prediction","task_name":"Property Prediction"}],"methods":[{"method_slug":"graph-neural-network","method_name":"Graph Neural Network"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2312.05388","atlas_url":"https://app.syntology.ai/?focus=2312.05388","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2312.05388"}},"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":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/teddykoker/e3nn.c","reach":{"status":"ok","spdx":"MIT"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/AIforGreatGood/charge3net","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran":7,"unverified":2},"by_repo_kind":{"official":{"samples":8,"ran":7,"repositories":1},"listed":{"samples":1,"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":"c71c9fc82e35967e","entry":"approximate_gradient","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/utils/data.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/utils/data.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c71c9fc82e35967e"}},{"code_sha256_prefix":"9d9514f8ed34f3dd","entry":"calculate_grid_pos","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/utils/data.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/utils/data.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9d9514f8ed34f3dd"}},{"code_sha256_prefix":"3a3e772e2ac9e9d7","entry":"compute_nmape","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/utils/predictions.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/utils/predictions.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"3a3e772e2ac9e9d7"}},{"code_sha256_prefix":"cc233c340c3c4104","entry":"compute_nmape_components","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/utils/predictions.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/utils/predictions.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"cc233c340c3c4104"}},{"code_sha256_prefix":"e6e1aab7b6b1b46e","entry":"compute_zeta","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/utils/data.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/utils/data.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e6e1aab7b6b1b46e"}},{"code_sha256_prefix":"907034711fd04cf3","entry":"get_irreps","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/charge3net/models/e3.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/charge3net/models/e3.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"907034711fd04cf3"}},{"code_sha256_prefix":"ddea4708a433a69f","entry":"scatter","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/charge3net/models/e3.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/charge3net/models/e3.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"ddea4708a433a69f"}},{"code_sha256_prefix":"1fbb35c3cdd6e9db","entry":"benchmark","repo":"teddykoker/e3nn.c","repo_kind":"listed","path":"extra/benchmark_python.py","file_url":"https://github.com/teddykoker/e3nn.c/blob/HEAD/extra/benchmark_python.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":"1fbb35c3cdd6e9db"}},{"code_sha256_prefix":"bcdef13a27ddc2d5","entry":"tp_path_exists","repo":"AIforGreatGood/charge3net","repo_kind":"official","path":"src/charge3net/models/e3.py","file_url":"https://github.com/AIforGreatGood/charge3net/blob/HEAD/src/charge3net/models/e3.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":"bcdef13a27ddc2d5"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}