{"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/electra-a-symmetry-breaking-cartesian-network","title":"ELECTRA: A Cartesian Network for 3D Charge Density Prediction with Floating Orbitals","arxiv_id":"2503.08305","date":"2025-03-11","proceeding":null,"authors":["Jonas Elsborg","Luca Thiede","Alán Aspuru-Guzik","Tejs Vegge","Arghya Bhowmik"],"abstract":"We present the Electronic Tensor Reconstruction Algorithm (ELECTRA) - an equivariant model for predicting electronic charge densities using floating orbitals. Floating orbitals are a long-standing concept in the quantum chemistry community that promises more compact and accurate representations by placing orbitals freely in space, as opposed to centering all orbitals at the position of atoms. Finding the ideal placement of these orbitals requires extensive domain knowledge, though, which thus far has prevented widespread adoption. We solve this in a data-driven manner by training a Cartesian tensor network to predict the orbital positions along with orbital coefficients. This is made possible through a symmetry-breaking mechanism that is used to learn position displacements with lower symmetry than the input molecule while preserving the rotation equivariance of the charge density itself. Inspired by recent successes of Gaussian Splatting in representing densities in space, we are using Gaussian orbitals and predicting their weights and covariance matrices. Our method achieves a state-of-the-art balance between computational efficiency and predictive accuracy on established benchmarks.","url_abs":"https://arxiv.org/abs/2503.08305v2","url_pdf":"https://arxiv.org/pdf/2503.08305v2.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":[],"tasks":[{"task_slug":"computational-efficiency","task_name":"Computational Efficiency"},{"task_slug":null,"task_name":"Position"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2503.08305","atlas_url":"https://app.syntology.ai/?focus=2503.08305","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2503.08305"}},"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/Jotels/ELECTRA_2025","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran":7,"ran_honours":1,"unverified":1},"by_repo_kind":{"found_in_text":{"samples":9,"ran":8,"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":"831592d0c13c678c","entry":"add_scaling","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/utils.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/utils.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":"831592d0c13c678c"}},{"code_sha256_prefix":"f55c31d782e48ffb","entry":"collate_fn","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/ELECTRA.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/ELECTRA.py","link_basis":"harvester_set","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f55c31d782e48ffb"}},{"code_sha256_prefix":"7709a70ba89c3325","entry":"compute_pairwise_distances","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/model/miao.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/model/miao.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":"7709a70ba89c3325"}},{"code_sha256_prefix":"1f5212ae50a8dffb","entry":"expand_to","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/utils.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/utils.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":"1f5212ae50a8dffb"}},{"code_sha256_prefix":"e97b464287f8ebf2","entry":"minimum_image","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/ELECTRA.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/ELECTRA.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":"e97b464287f8ebf2"}},{"code_sha256_prefix":"bfc3b1d4bf7ca344","entry":"minimum_image_vector","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/model/miao.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/model/miao.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":"bfc3b1d4bf7ca344"}},{"code_sha256_prefix":"4808d09815f66d0d","entry":"multi_outer_product","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/utils.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/utils.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":"4808d09815f66d0d"}},{"code_sha256_prefix":"498f2364b7c8ec0e","entry":"wrap_to_cell","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/ELECTRA.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/ELECTRA.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":"498f2364b7c8ec0e"}},{"code_sha256_prefix":"4411e6f9461a97db","entry":"compile_with_dynamic_shapes","repo":"Jotels/ELECTRA_2025","repo_kind":"found_in_text","path":"models/electra_hotpp/layer/equivalent.py","file_url":"https://github.com/Jotels/ELECTRA_2025/blob/HEAD/models/electra_hotpp/layer/equivalent.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":"4411e6f9461a97db"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}