Global Plane Waves From Local Gaussians: Periodic Charge Densities in a Blink
Not on the map. Why?
Abstract (arXiv)
We introduce ELECTRAFI, a fast, end-to-end differentiable model for predicting periodic charge densities in crystalline materials. ELECTRAFI constructs anisotropic Gaussians in real space and exploits their closed-form Fourier transforms to analytically evaluate plane-wave coefficients via the Poisson summation formula. This formulation delegates non-local and periodic behavior to analytic transforms, enabling reconstruction of the full periodic charge density with a single inverse FFT. By avoiding explicit real-space grid probing, periodic image summation, and spherical harmonic expansions, ELECTRAFI matches or exceeds state-of-the-art accuracy across periodic benchmarks while being up to 633 × faster than the strongest competing method, reconstructing crystal charge densities in a fraction of a second. When used to initialize DFT calculations, ELECTRAFI reduces total DFT compute cost by up to ~20%, whereas slower charge density models negate savings due to high inference times. Our results show that accuracy and inference cost jointly determine end-to-end DFT speedups, and motivate our focus on efficiency.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 29 of its 80 references that are in the map. First posted on arXiv 27 Jan 2026.
Builds on
29 of 80 references.
References · 29
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 735
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 701
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 467
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2021Factorized Fourier Neural Operatorscited by 237
- 2023EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representationscited by 201
- 2023Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNscited by 113
- 2017DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamicscited by 79
- 2021Generating 3D Molecules Conditional on Receptor Binding Sites with Deep Generative Modelscited by 72
17 more
- 2016By-passing the Kohn-Sham equations with machine learningcited by 65
- 2022GemNet-OC: Developing Graph Neural Networks for Large and Diverse Molecular Simulation Datasetscited by 61
- 2023Ewald-based Long-Range Message Passing for Molecular Graphscited by 46
- 2024The Importance of Being Scalable: Improving the Speed and Accuracy of Neural Network Interatomic Potentials Across Chemical Domainscited by 40
- 2021Informing Geometric Deep Learning with Electronic Interactions to Accelerate Quantum Chemistrycited by 27
- 2021Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solidscited by 24
- 2025PET-MAD, a lightweight universal interatomic potential for advanced materials modelingcited by 23
- 2023Higher-Order Equivariant Neural Networks for Charge Density Prediction in Materialscited by 12
- 2025HIP: Hessian Interatomic Potentials without derivativescited by 11
- 2023Equivariant Neural Operator Learning with Graphon Convolutioncited by 10
- 2024Neural P3M: A Long-Range Interaction Modeling Enhancer for Geometric GNNscited by 9
- 2025ELECTRA: A Cartesian Network for 3D Charge Density Prediction with Floating Orbitalscited by 9
- 2024Gaussian Plane-Wave Neural Operator for Electron Density Estimationcited by 8
- 2024Image Super-resolution Inspired Electron Density Predictioncited by 8
- 2024A Recipe for Charge Density Predictioncited by 8
- 2023Towards Combinatorial Generalization for Catalysts: A Kohn-Sham Charge-Density Approachcited by 5
- 2025Machine-Learning Interatomic Potentials for Long-Range Systemscited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.